class Middleman::CoreExtensions::FrontMatter::FrontmatterManager

def clear_data(file)
  # Copied from Sitemap::Store#file_to_path, but without
  # removing the file extension
  file = File.expand_path(file, @app.root)
  prefix = @app.source_dir.sub(/\/$/, "") + "/"
  return unless file.include?(prefix)
  path = file.sub(prefix, "")
  @cache.delete(path)
end

def data(path)
  p = normalize_path(path)
  @cache[p] ||= frontmatter_and_content(p)
end

def frontmatter_and_content(path)
  full_path = File.expand_path(path, @app.source_dir)
  content = File.read(full_path)
  if result = parse_yaml_front_matter(content)
    data, content = result
  elsif result = parse_json_front_matter(content)
    data, content = result
  else
    data = {}
  end
  [::Middleman::Util.recursively_enhance(data).freeze, content]
end

def initialize(app)
  @app = app
  @cache = {}
end

def manipulate_resource_list(resources)
  resources.each do |r|
    if !r.proxy? && r.data["ignored"] == true
      r.frontmatter_ignored = true
    end
  end
  
  resources
end

def normalize_path(path)
  path.sub(@app.source_dir, "").sub(/^\//, "")
end

def parse_json_front_matter(content)
  json_regex = /^(\{\{\{\s*\n.*?\n?)^(\}\}\}\s*$\n?)/m
  
  if content =~ json_regex
    content = content.sub(json_regex, "")
    begin
      json = ($1+$2).sub("{{{", "{").sub("}}}", "}")
      data = ActiveSupport::JSON.decode(json)
    rescue => e
      puts "JSON Exception: #{e.message}"
      return false
    end
  else
    return false
  end
  [data, content]
rescue
  [{}, content]
end

def parse_yaml_front_matter(content)
  yaml_regex = /^(---\s*\n.*?\n?)^(---\s*$\n?)/m
  if content =~ yaml_regex
    content = content.sub(yaml_regex, "")
    begin
      data = YAML.load($1)
    rescue => e
      puts "YAML Exception: #{e.message}"
      return false
    end
  else
    return false
  end
  [data, content]
rescue
  [{}, content]
end