class RandomWords::Config

def all_parts_of_speech?(dir, lang = nil)
  lang ||= @lang
  dir ||= @source_dir
  exists = true
  SPEECH_PARTS.each do |part|
    next if File.exist?(File.join(dir, "#{part}.txt"))
    warn "Missing #{File.join(dir, "#{part}.txt")} for #{lang}"
    exists = false
    break
  end
  CONFIG_FILES.each do |file|
    unless File.exist?(File.join(dir, "#{file}.yml"))
      warn "Missing #{File.join(dir, "#{file}.yml")} for #{lang}"
      exists = false
    end
  end
  exists
end

def base_config
  config_file = File.join(config_dir, 'config.yml')
  if File.exist?(config_file)
    config = YAML.load_file(config_file).symbolize_keys
    return handle_config(config)
  end
  # If the config file doesn't exist, create it
  # and return the default configuration
  create_base_config(config_file)
  config = YAML.load_file(config_file).symbolize_keys
  handle_config(config)
end

def builtin_lang_dir(lang = nil, basedir: nil)
  lang ||= @lang
  basedir ||= __dir__
  dir = File.join(basedir, 'words', lang)
  return dir if File.directory?(dir)
  warn "No dictionary found for #{lang}"
  nil
end

def config
  @config ||= base_config
end

def config_dir
  @config_dir ||= File.expand_path(File.join(Dir.home, '.config', 'random-words'))
end

def config_dir=(dir)
  @config_dir = File.expand_path(dir)
end

def config_file
  config_file = File.join(config_dir, 'config.yml')
  return config_file if File.exist?(config_file)
  create_base_config(config_file)
  config_file
end

def create_base_config(config_file)
  FileUtils.mkdir_p(config_dir) unless File.directory?(config_dir)
  config = {
    'source' => 'latin',
    'length' => 'medium',
    'paragraph_length' => 5,
    'extended_punctuation' => false
  }
  File.write(config_file, config.to_yaml)
  warn "Created #{config_file}"
  config_file
end

def create_user_dictionary(lang = nil)
  return lang.to_sym if File.directory?(File.join(config_dir, 'words',
                                                  lang)) && all_parts_of_speech?(File.join(config_dir, 'words', lang), lang)
  lang_dir = File.join(config_dir, 'words', lang)
  FileUtils.mkdir_p(lang_dir) unless File.directory?(lang_dir)
  SPEECH_PARTS.each do |part|
    source_file = File.join(builtin_lang_dir('english'), "#{part}.txt")
    target_file = File.join(lang_dir, "#{part}.txt")
    unless File.exist?(target_file)
      FileUtils.cp(source_file, target_file)
      warn "Created #{part}.txt"
    end
  end
  # Copy numbers.yml file from the builtin directory
  source_file = File.join(builtin_lang_dir('english'), 'numbers.yml')
  target_file = File.join(lang_dir, 'numbers.yml')
  unless File.exist?(target_file)
    FileUtils.cp(source_file, target_file)
    warn 'Created numbers.yml'
  end
  # Create the config.yml file if it doesn't exist
  target_file = File.join(lang_dir, 'config.yml')
  unless File.exist?(target_file)
    config = {
      'name' => lang,
      'triggers' => [lang],
      'description' => "User dictionary for #{lang}"
    }
    File.write(target_file, config.to_yaml)
    warn "Created #{target_file}"
  end
  unless all_parts_of_speech?(lang_dir, lang) || (RandomWords.testing && !RandomWords.tested.include?('create_user_dictionary'))
    return
  end
  RandomWords.tested << 'create_user_dictionary'
  warn "Created #{lang} in #{lang_dir}"
  lang.to_sym
end

def handle_config(configuration)
  ext_punc = configuration[:extended_punctuation]
  {
    source: configuration[:source].to_source || :latin,
    sentence_length: configuration[:length].to_length || :medium,
    paragraph_length: configuration[:paragraph_length].to_i || 5,
    use_extended_punctuation: (ext_punc && ext_punc.trueish?) || false
  }
end

def initialize(lang)
  @lang = lang.to_s
  FileUtils.mkdir_p(config_dir) unless File.directory?(config_dir)
  @source_dir = user_dictionary_exist? ? user_lang_dir : builtin_lang_dir
  raise "No dictionary found for #{@lang}" unless @source_dir
  rw_source = RandomWords::Source.new(@lang, @source_dir)
  @dictionary = rw_source.dictionary
end

def sources
  return @sources if @sources
  @sources = {}
  Dir[File.join(__dir__, 'words', '*')].each do |dir|
    next unless File.directory?(dir)
    name = File.basename(dir)
    @sources[name] = RandomWords::Source.new(name, dir)
  end
  Dir[File.join(config_dir, 'words', '*')].each do |dir|
    next unless File.directory?(dir)
    name = File.basename(dir)
    @sources[name] = RandomWords::Source.new(name, dir)
  end
  @sources
end

def user_dictionary_exist?
  if user_lang_dir
    raise "User dictionary for #{@lang} is incomplete. Please run create_user_dictionary." unless all_parts_of_speech?(
      user_lang_dir, @lang
    )
    true
  else
    false
  end
end

def user_lang_dir
  File.join(config_dir, 'words', @lang) if File.exist?(File.join(config_dir, 'words', @lang))
end