require 'thor'
require 'gogetit'
require 'gogetit/util'

module Gogetit

  class CLI < Thor
    include Gogetit::Util
    package_name 'Gogetit'

    @result = nil
    class << self
      attr_accessor :result
    end

    attr_reader :config, :logger, :lxd, :libvirt, :providers

    def initialize(*args)
      super
      @config = Gogetit.config
      @logger = Gogetit.logger
      @lxd = Gogetit.lxd
      @libvirt = Gogetit.libvirt
      @providers = {
        lxd: lxd,
        libvirt: libvirt
      }
    end

    desc 'list', 'List containers and instances, running currently.'
    method_option :out, :aliases => '-o', :type => :string, \
      :default => '', :desc => 'to list from all remotes'
    def list
      case options[:out]
      when 'custom'
        Gogetit.list_all_types
      when 'all'
        config[:lxd][:nodes].each do |node|
          puts "Listing LXD containers on #{node[:url]}.."
          system("lxc list #{node[:name]}:")
        end
        puts "Listing KVM domains on #{config[:libvirt][:nodes][0][:url]}.."
        system("virsh -c #{config[:libvirt][:nodes][0][:url]} list --all")
      when ''
        puts "Listing LXD containers on #{config[:lxd][:nodes][0][:url]}.."
        system("lxc list #{config[:lxd][:nodes][0][:name]}:")
        puts ''
        puts "Listing KVM domains on #{config[:libvirt][:nodes][0][:url]}.."
        system("virsh -c #{config[:libvirt][:nodes][0][:url]} list --all")
      else
        puts "Invalid option or command"
      end
    end

    desc 'create NAME', 'Create either a container or KVM domain.'
    method_option :provider, :aliases => '-p', :type => :string, \
      :default => 'lxd', :desc => 'A provider such as lxd and libvirt'
    method_option :alias, :aliases => '-a', :type => :string, \
      :desc => 'An alias name for a lxd image'
    method_option :distro, :aliases => '-d', :type => :string, \
      :desc => 'A distro name with its series for libvirt provider'
    method_option :chef, :aliases => '-c', :type => :boolean, \
      :default => false, :desc => 'Chef awareness'
    method_option :zero, :aliases => '-z', :type => :boolean, \
      :default => false, :desc => 'Chef Zero awareness'
    method_option :vlans, :aliases => '-v', :type => :array, \
      :desc => 'A list of VLAN IDs to connect to'
    method_option :ipaddresses, :aliases => '-i', :type => :array, \
      :desc => 'A list of static IPs to assign'
    method_option :"no-maas", :type => :boolean, \
      :desc => 'Without MAAS awareness(only for LXD provider)'
    method_option :"maas-on-lxc", :type => :boolean, \
      :desc => 'To install MAAS on a LXC enabling necessary user config'\
      '(only for LXD provider with no-maas enabled)'
    method_option :"file", :aliases => '-f', :type => :string, \
      :desc => 'File location(only for LXD provider)'
    def create(name)
      abort("'vlans' and 'ipaddresses' can not be set together.") \
        if options['vlans'] and options['ipaddresses']
      abort("'chef' and 'zero' can not be set together.") \
        if options['chef'] and options['zero']
      abort("when 'no-maas', the network configuration have to be set by 'file'.") \
        if options['no-maas'] and (options['vlans'] or options['ipaddresses'])
      abort("'no-maas' and 'file' have to be set together.") \
        if options['no-maas'] ^ !!options['file']
      abort("'maas-on-lxc' has to be set with 'no-maas'.") \
        if options['maas-on-lxc'] and !options['no-maas']
      abort("'distro' has to be set with libvirt provider.") \
        if options['distro'] and options['provider'] == 'lxd'
      abort("'alias' has to be set with lxd provider.") \
        if options['alias'] and options['provider'] == 'libvirt'

      case options['provider']
      when 'lxd'
        Gogetit::CLI.result = lxd.create(name, options.to_hash)
      when 'libvirt'
        Gogetit::CLI.result = libvirt.create(name, options.to_hash)
      else
        abort('Invalid argument entered.')
      end

      # post-tasks
      if options['chef']
        knife_bootstrap_chef(name, options[:provider], config)
        update_databags(config)
      elsif options['zero']
        knife_bootstrap_zero(name, options[:provider], config)
      end
    end

    desc 'destroy NAME', 'Destroy either a container or KVM instance.'
    method_option :chef, :aliases => '-c', :type => :boolean, \
      :default => false, :desc => 'Chef awareness'
    method_option :zero, :aliases => '-z', :type => :boolean, \
      :default => false, :desc => 'Chef Zero awareness'
    def destroy(name)
      abort("'chef' and 'zero' can not be set together.") \
        if options['chef'] and options['zero']

      provider = get_provider_of(name, providers)
      if provider
        case provider
        when 'lxd'
          Gogetit::CLI.result = lxd.destroy(name)
        when 'libvirt'
          Gogetit::CLI.result = libvirt.destroy(name)
        else
          abort('Invalid argument entered.')
        end
      end
      # post-tasks
      if options['chef']
        knife_remove(name, options)
        update_databags(config)
      elsif options['zero']
        knife_remove(name, options)
      end
    end

    desc 'deploy NAME', 'Deploy a node existing in MAAS.'
    method_option :distro, :aliases => '-d', :type => :string, \
      :desc => 'A distro name with its series for libvirt provider'
    method_option :chef, :aliases => '-c', :type => :boolean, \
      :default => false, :desc => 'Chef awareness'
    method_option :zero, :aliases => '-z', :type => :boolean, \
      :default => false, :desc => 'Chef Zero awareness'
    def deploy(name)
      abort("'chef' and 'zero' can not be set together.") \
        if options['chef'] and options['zero']

      Gogetit::CLI.result = libvirt.deploy(name, options.to_hash)

      # post-tasks
      if options['chef']
        knife_bootstrap(name, options[:provider], config)
        update_databags(config)
      elsif options['zero']
        knife_bootstrap_zero(name, options[:provider], config)
      end
    end

    desc 'release NAME', 'Release a node in MAAS'
    method_option :chef, :type => :boolean, :desc => "Enable chef awareness."
    method_option :chef, :aliases => '-c', :type => :boolean, \
      :default => false, :desc => 'Chef awareness'
    method_option :zero, :aliases => '-z', :type => :boolean, \
      :default => false, :desc => 'Chef Zero awareness'
    def release(name)
      abort("'chef' and 'zero' can not be set together.") \
        if options['chef'] and options['zero']

      Gogetit::CLI.result = libvirt.release(name)

      # post-tasks
      if options['chef']
        knife_remove(name, options)
        update_databags(config)
      elsif options['zero']
        knife_remove(name, options)
      end
    end

    desc 'rebuild NAME', 'Destroy(or release) and create(or deploy)'\
    ' either a container or a node(machine) in MAAS again.'
    method_option :chef, :aliases => '-c', :type => :boolean, \
      :default => false, :desc => 'Chef awareness'
    method_option :zero, :aliases => '-z', :type => :boolean, \
      :default => false, :desc => 'Chef Zero awareness'
    def rebuild(name)
      abort("'chef' and 'zero' can not be set together.") \
        if options['chef'] and options['zero']

      provider = get_provider_of(name, providers)
      if provider
        case provider
        when 'lxd'
          1.upto(100) { print '_' }; puts
          puts "Destroying #{name}.."
          invoke :destroy, [name]
          alias_name = YAML.load(
            Gogetit::CLI.result[:info][:config][:"user.user-data"]
          )['source_image_alias']
          1.upto(100) { print '_' }; puts
          puts "Creating #{name}.."
          invoke :create, [name], :alias => alias_name
        when 'libvirt'
          invoke :release, [name]
          distro_name = Gogetit::CLI.result[:info][:machine]['distro_series']
          invoke :deploy, [name], :distro => distro_name
        else
          abort('Invalid argument entered.')
        end
      end
      # post-tasks
      if options['chef']
        knife_remove(name) if options[:chef]
        update_databags(config)
      end
    end
  end
end