update_index_attributes.rb

#!/usr/bin/env ruby

# Function to update front matter of `index.md` files in directories.
# Uses the output file from `extract_index_attributes.rb`
#
# Usage:
# ruby update_index_attributes.rb

input_file = 'update_attributes_here.txt'

def update_front_matter(input_file)
  directory_groups = parse_input_file(input_file)

  directory_groups.each do |new|
    path = new[0]

    current = current_front_matter(path)
    old = extract_front_matter(current).split("\n")

    all_properties = {
      path:  new[0],
      old_title: parse_properties(old[0]),
      new_title: parse_properties(new[1]),
      old_label: parse_properties(old[1]),
      new_label: parse_properties(new[2]),
      old_position: parse_properties(old[2]),
      new_position: parse_properties(new[3]),
    }

    updated_front_matter = replace_properties(current, all_properties)
    write_front_matter(path, updated_front_matter)
  end
end

def extract_front_matter(input)

  # Extract front matter using regex
  front_matter_match = input.match(/^---\n(.*?)\n---/m)

  if front_matter_match
    front_matter = front_matter_match[1]

    # Extract specific properties, preserving original quotes
    title_match = front_matter.match(/title:\s*(["']?)([^"'\n]+)\1/)
    sidebar_position_match = front_matter.match(/sidebar_position:\s*(\d+)/)
    sidebar_label_match = front_matter.match(/sidebar_label:\s*(["']?)([^"'\n]+)\1/)

    # Prepare results
    title = title_match ? title_match[2].strip : 'N/A'
    sidebar_position = sidebar_position_match ? sidebar_position_match[1] : 'N/A'
    sidebar_label = sidebar_label_match ? sidebar_label_match[2].strip : 'N/A'

    # Determine if quotes were present in the original
    title_quotes = title_match ? title_match[1] : ''
    sidebar_label_quotes = sidebar_label_match ? sidebar_label_match[1] : ''

    # Build and return the results as a string
    <<~RESULTS.chomp
    - title: #{title_quotes}#{title}#{title_quotes}
    - sidebar_label: #{sidebar_label_quotes}#{sidebar_label}#{sidebar_label_quotes}
    - sidebar_position: #{sidebar_position}
    RESULTS
  else
    "Error: No front matter found"
  end
end

def parse_input_file(file_path)
  lines = File.readlines(file_path).map(&:strip)
  directory_groups = []
  current_group = []

  lines.each do |line|
    if line.match?(/^docs\//)
      directory_groups << current_group if current_group.any?
      current_group = [line]
    elsif line.start_with?('- ')
      current_group << line
    end
  end

  directory_groups << current_group if current_group.any?
  directory_groups
end

def parse_properties(line)
  match = line.match(/^- (title:|sidebar_label:|sidebar_position:).*?\ *[\"\']?(.*)\b[\"\']?/m)
  if match
    match[2].strip
  else
    ""
  end
end

def current_front_matter(path)
  index_file = File.join(path, 'index.md')
  File.exist?(path) ? path : nil

  # Check if index.md exists
  unless File.exist?(index_file)
    puts "File not found: #{index_file}"
  end

  begin
    file_contents = File.read(index_file)
  rescue Errno::ENOENT
    return "Error: File not found at #{index_file}"
  end

  frontmatter_match = file_contents.match(/---\n(.*?)\n---/m)

  unless frontmatter_match
    puts "No front matter found in: #{index_file}"
  end
  frontmatter_match[0]
end

def replace_properties(current, all)
  if all[:old_title] != all[:new_title]
    current.gsub!(/title: .*/, "title: \"#{all[:new_title]}\"")
  end

  if all[:old_label] == "N/A" && all[:new_label] != "N/A"
    current.gsub!(/(\n---\s*)$/, "\nsidebar_label: \"#{all[:new_label]}\"\\1")
  elsif all[:old_label] != all[:new_label]
    current.gsub!(/sidebar_label: .*/, "sidebar_label: \"#{all[:new_label]}\"")
  end

  if all[:old_position] != all[:new_position]
    current.gsub!(/sidebar_position: .*/, "sidebar_position: #{all[:new_position]}")
  end

  current
end

def write_front_matter(path, front_matter)
  index_file = File.join(path, 'index.md')
  content = File.read(index_file)

  existing = content.match(/---\n(.*?)\n---/m)

  if existing[0] == front_matter
    puts "No changes, skipping: #{index_file}"
  else
    content.sub!(/---\n.*?\n---/m, "#{front_matter}")
    File.write(index_file, content)
    puts "Updated file: #{index_file}"
  end
end

update_front_matter(input_file)