module Scaffolding::FileManipulator

def self.add_line_to_yml_file(file, content, location_array)
  # First check that the given location array actually exists in the yml file:
  yml = YAML.safe_load_file(file)
  location_array.map!(&:to_s)
  # TODO: Raise an error if we're returning nil.
  return nil if yml.dig(*location_array).nil?
  content += "\n" unless content[-1] == "\n"
  # Find the location in the file where the location_array is
  lines = File.readlines(file)
  current_needle = location_array.shift.to_s
  current_space = ""
  insert_after = 1
  lines.each_with_index do |line, index|
    break if current_needle.nil?
    if line.strip == current_needle + ":"
      current_needle = location_array.shift.to_s
      insert_after = index
      current_space = line.match(/\s+/).to_s
    end
  end
  new_lines = []
  current_space += "  "
  lines.each_with_index do |line, index|
    new_lines << line
    new_lines << current_space + content if index == insert_after
  end
  File.write(file, new_lines.join)
end

def self.find(lines, needle, within = 0)
  lines_within(lines, within).each_with_index do |line, line_number|
    return (within + (within ? 1 : 0) + line_number) if line.match?(needle)
  end
  nil
end

def self.lines_within(lines, within)
  return lines unless within
  lines[(within + 1)..(Scaffolding::BlockManipulator.find_block_end(starting_from: within, lines: lines) + 1)]
end

def self.replace_line_in_file(file, content, in_place_of, options = {})
  begin
    target_file_content = File.read(file)
  rescue Errno::ENOENT => _
    puts "Couldn't find '#{file}'".red unless options[:suppress_could_not_find]
    return false
  end
  if options[:content_replacement_transformed] && target_file_content.include?(content)
    puts "No need to update '#{file}'. It already has '#{content}'."
  else
    puts "Updating '#{file}'." unless silence_logs?
    target_file_content.gsub!(in_place_of, content)
    File.write(file, target_file_content)
  end
end

def self.write(file_name, lines, strip: true)
  puts "Updating '#{file_name}'." unless silence_logs?
  if strip
    File.open(file_name, "w+") do |file|
      file.puts(lines.join.strip + "\n")
    end
  else
    File.write(file_name, lines.join)
  end
end