class Packnga::DocumentTask

def base_dir=(dir)
  dir = Pathname.new(dir)
  @yard_task.base_dir = dir
  @reference_task.base_dir = dir
end

def c_source_file?(file)
  file.start_with?("ext/") and file.end_with?(".c")
end

def define
  set_default_values
  define_tasks
end

def define_tasks
  @yard_task.define
  @reference_task.define
end

def initialize(spec)
  @spec = spec
  @yard_task = YARDTask.new(@spec)
  @reference_task = ReferenceTask.new(@spec)
  self.base_dir = "doc"
  yield(self) if block_given?
  define
end

def original_language=(language)
  @reference_task.original_language = language
end

def reference
  yield(@reference_task)
end

def ruby_source_file?(file)
  file.start_with?("lib/") and file.end_with?(".rb")
end

def set_default_readme
  readme = @spec.files.find do |file|
    file.include?("README")
  end
  return if readme.nil?
  @yard_task.readme = readme
  @reference_task.readme = readme
end

def set_default_source_files
  source_files = @spec.files.find_all do |file|
    ruby_source_file?(file) or
      c_source_file?(file)
  end
  @yard_task.source_files = source_files
  @reference_task.source_files = source_files
end

def set_default_text_files
  text_dir = @yard_task.base_dir + "text"
  text_files = @spec.files.find_all do |file|
    file.start_with?(text_dir.to_s)
  end
  @yard_task.text_files = text_files
  @reference_task.text_files = text_files
end

def set_default_values
  set_default_readme
  set_default_source_files
  set_default_text_files
end

def translate_language=(language)
  self.translate_languages = [language]
end

def translate_languages=(languages)
  @reference_task.translate_languages = languages
end

def yard
  yield(@yard_task)
end