class ChefSpec::Coverage::ResourceWrapper

def initialize(resource = nil)
  @resource = resource
end

def shortname(file)
  if file.include?(Dir.pwd)
    file.split(Dir.pwd, 2).last
  elsif file.include?("cookbooks")
    file.split("cookbooks/", 2).last
  else
    file
  end
end

def source_file
  @source_file ||= if @resource.source_line
                     shortname(@resource.source_line.split(":").first)
                   else
                     "Unknown"
                   end
end

def source_line
  @source_line ||= if @resource.source_line
                     @resource.source_line.split(":", 2).last.to_i
                   else
                     "Unknown"
                   end
end

def to_json
  {
    "source_file" => source_file,
    "source_line" => source_line,
    "touched" => touched?,
    "resource" => to_s,
  }.to_json
end

def to_s
  @resource.to_s
end

def touch!
  @touched = true
end

def touched?
  !!@touched
end