require_relative 'getch/options'
require_relative 'getch/states'
require_relative 'getch/gentoo'
require_relative 'getch/filesystem'
require_relative 'getch/command'
require_relative 'getch/helpers'
require_relative 'getch/log'

module Getch

  DEFAULT_OPTIONS = {
    language: 'en_US',
    zoneinfo: 'US/Eastern',
    keyboard: 'us',
    disk: 'sda',
    fs: 'ext4',
    username: nil,
    encrypt: false,
    verbose: false
  }

  STATES = {
    :partition => false,
    :format => false,
    :mount => false,
    :gentoo_base => false,
    :gentoo_config => false,
    :gentoo_update => false,
    :gentoo_kernel => false
  }

  MOUNTPOINT = "/mnt/gentoo".freeze
  OPTIONS_FS = {
    'ext4' => DEFAULT_OPTIONS[:encrypt] ? Getch::FileSystem::Ext4::Encrypt : Getch::FileSystem::Ext4,
    'lvm' => DEFAULT_OPTIONS[:encrypt] ? Getch::FileSystem::Lvm::Encrypt : Getch::FileSystem::Lvm
  }.freeze

  def self.resume_options(opts)
    puts "\nBuild Gentoo with the following args:\n"
    puts "lang: #{DEFAULT_OPTIONS[:language]}"
    puts "zoneinfo: #{DEFAULT_OPTIONS[:zoneinfo]}"
    puts "keyboard: #{DEFAULT_OPTIONS[:keyboard]}"
    puts "disk: #{DEFAULT_OPTIONS[:disk]}"
    puts "fs: #{DEFAULT_OPTIONS[:fs]}"
    puts "username: #{DEFAULT_OPTIONS[:username]}"
    puts "encrypt: #{DEFAULT_OPTIONS[:encrypt]}"
    puts
    print "Continue? (n,y) "
    case gets.chomp
    when /^y|^Y/
      return
    else
      exit 1
    end
  end

  def self.format(disk, fs, user)
    return if STATES[:format] and STATES[:partition]
    log = Log.new
    puts
    print "Partition and format disk #{disk}, this will erase all data, continue? (n,y) "
    case gets.chomp
    when /^y|^Y/
      log.info("Partition start")
      OPTIONS_FS[fs]::Partition.new
      OPTIONS_FS[fs]::Format.new
    else
      exit 1
    end
  end

  def self.init_gentoo(options)
    gentoo = Getch::Gentoo
    gentoo.stage3
    gentoo.config(options)
    gentoo.chroot
    gentoo.kernel
    gentoo.boot(options)
  end

  def self.main(argv)
    options = Options.new(argv)
    DEFAULT_OPTIONS.freeze
    resume_options(options)
    Getch::States.new # Update States
    format(options.disk, options.fs, options.username)
    OPTIONS_FS[DEFAULT_OPTIONS[:fs]]::Mount.new.run
    init_gentoo(options)
  end
end