require 'hyperkit'
require 'gogetit/util'
require 'yaml'
require 'hashie'
require 'table_print'

module Gogetit
  class GogetLXD
    include Gogetit::Util

    attr_reader :config, :logger, :conn, :maas

    def initialize(conf, maas, logger)
      @config = conf
      @conn = Hyperkit::Client.new(
          api_endpoint: config[:lxd][:nodes][0][:url],
          verify_ssl: false
      )
      @maas = maas
      @logger = logger
    end

    def list
      logger.info("Calling <#{__method__.to_s}>")
      conn.containers
    end

    def container_exists?(name)
      logger.info("Calling <#{__method__.to_s}> for #{name}")
      list.each do |c|
        return true if c == name
      end
      false
    end

    def get_state(name)
      logger.info("Calling <#{__method__.to_s}>")
      conn.container(name)[:status]
    end

    def wait_until_state(name, state)
      logger.info("Calling <#{__method__.to_s}> for being #{state}..")
      until get_state(name) == state
        sleep 3
      end
    end

    # to generate 'user.user-data'
    def generate_user_data(lxd_params, options)
      logger.info("Calling <#{__method__.to_s}>")

      lxd_params[:config] = {}

      if options[:'no-maas']
        lxd_params[:config][:"user.user-data"] = {}
      else
        sshkeys = maas.get_sshkeys
        pkg_repos = maas.get_package_repos

        lxd_params[:config][:'user.user-data'] = { 'ssh_authorized_keys' => [] }

        sshkeys.each do |key|
          lxd_params[:config][:'user.user-data']['ssh_authorized_keys'].push(key['key'])
        end

        pkg_repos.each do |repo|
          if repo['name'] == 'main_archive'
            lxd_params[:config][:'user.user-data']['apt_mirror'] = repo['url']
          end
        end

        lxd_params[:config][:"user.user-data"]['source_image_alias'] = lxd_params[:alias]
        lxd_params[:config][:"user.user-data"]['maas'] = true
      end

      if options[:'maas-on-lxc']
        lxd_params[:config][:"security.privileged"] = "true"
      end

      if options[:'lxd-in-lxd']
        lxd_params[:config][:"security.nesting"] = "true"
      end

      lxd_params[:config][:"user.user-data"]['gogetit'] = true

      # To disable to update apt database on first boot
      # so chef client can keep doing its job.
      lxd_params[:config][:'user.user-data']['package_update'] = false
      lxd_params[:config][:'user.user-data']['package_upgrade'] = false

      lxd_params[:config][:'user.user-data'] = generate_cloud_init_config(
        options,
        config,
        lxd_params[:config][:'user.user-data']
      )

      lxd_params[:config][:"user.user-data"] = \
        "#cloud-config\n" + YAML.dump(lxd_params[:config][:"user.user-data"])[4..-1]

      return lxd_params
    end

    def generate_network_config(lxd_params, options)
      logger.info("Calling <#{__method__.to_s}>")

      if options[:'no-maas']
        lxd_params[:config][:'user.network-config'] = \
          YAML.load_file(options[:file])['network']

        # physical device will be the gate device
        lxd_params[:config][:"user.network-config"]['config'].each do |iface|
          if iface['type'] == 'physical'
            config[:ip_to_access] = iface['subnets'][0]['address'].split('/')[0]
          end
        end

        lxd_params[:config][:"user.network-config"] = \
          YAML.dump(lxd_params[:config][:"user.network-config"])[4..-1]

      elsif options[:ipaddresses]
        config[:ifaces] = check_ip_available(options[:ipaddresses], maas)
        abort("There is no dns server specified for the gateway network.") \
          unless config[:ifaces][0]['dns_servers'][0]
        abort("There is no gateway specified for the gateway network.") \
          unless config[:ifaces][0]['gateway_ip']

        lxd_params[:config][:'user.network-config'] = {
          'version' => 1,
          'config' => [
            {
              'type' => 'nameserver',
              'address' => config[:ifaces][0]['dns_servers'][0],
              'search' => maas.get_domain,
            }
          ]
        }

        # to generate configuration for [:config][:'user.network-config']['config']
        config[:ifaces].each_with_index do |iface,index|
          if index == 0
            iface_conf = {
              'type' => 'physical',
              'name' => "eth#{index}",
              'subnets' => [
                {
                  'type' => 'static',
                  'ipv4' => true,
                  'address' => iface['ip'] + '/' + iface['cidr'].split('/')[1],
                  'gateway' => iface['gateway_ip'],
                  'mtu' => iface['vlan']['mtu'],
                  'control' => 'auto'
                }
              ]
            }
          elsif index > 0
            if config[:ifaces][0]['vlan']['name'] != 'untagged'
              iface_conf = {
                'type' => 'physical',
                'name' => "eth#{index}",
                'subnets' => [
                  {
                    'type' => 'static',
                    'ipv4' => true,
                    'address' => iface['ip'] + '/' + iface['cidr'].split('/')[1],
                    'mtu' => iface['vlan']['mtu'],
                    'control' => 'auto'
                  }
                ]
              }
            elsif config[:ifaces][0]['vlan']['name'] == 'untagged'
              iface_conf = {
                'type' => 'vlan',
                'name' => "eth0.#{iface['vlan']['vid'].to_s}",
                'vlan_id' => iface['vlan']['vid'].to_s,
                'vlan_link' => 'eth0',
                'subnets' => [
                  {
                    'type' => 'static',
                    'ipv4' => true,
                    'address' => iface['ip'] + '/' + iface['cidr'].split('/')[1],
                    'mtu' => iface['vlan']['mtu'],
                    'control' => 'auto'
                  }
                ]
              }
            end
          end

          lxd_params[:config][:'user.network-config']['config'].push(iface_conf)
        end

        lxd_params[:config][:"user.network-config"] = \
          YAML.dump(lxd_params[:config][:"user.network-config"])[4..-1]
      end

      return lxd_params
    end

    # To configure devices
    def generate_devices(lxd_params, options)
      logger.info("Calling <#{__method__.to_s}>")
      lxd_params[:devices] = {}

      if options[:'no-maas']
        lxd_params[:devices] = YAML.load_file(options[:file])['devices']

        # Now, LXD API can handle integer as a value of a map
        lxd_params[:devices].each do |k, v|
          v.each do |kk, vv|
            if vv.is_a? Integer
              v[kk] = vv.to_s
            end
          end
        end

        lxd_params[:devices] = (Hashie.symbolize_keys lxd_params[:devices])

      elsif options[:ipaddresses]
        config[:ifaces].each_with_index do |iface,index|
          if index == 0
            if iface['vlan']['name'] == 'untagged' # or vid == 0
              lxd_params[:devices][:"eth#{index}"] = {
                mtu: iface['vlan']['mtu'].to_s,   #This must be string
                name: "eth#{index}",
                nictype: 'bridged',
                parent: config[:default][:root_bridge],
                type: 'nic'
              }
            elsif iface['vlan']['name'] != 'untagged' # or vid != 0
              lxd_params[:devices][:"eth#{index}"] = {
                mtu: iface['vlan']['mtu'].to_s,   #This must be string
                name: "eth#{index}",
                nictype: 'bridged',
                parent: config[:default][:root_bridge] + "-" + iface['vlan']['vid'].to_s,
                type: 'nic'
              }
            end
          # When config[:ifaces][0]['vlan']['name'] == 'untagged' and index > 0,
          # it does not need to generate more devices 
          # since it will configure the IPs with tagged VLANs.
          elsif config[:ifaces][0]['vlan']['name'] != 'untagged'
            lxd_params[:devices][:"eth#{index}"] = {
              mtu: iface['vlan']['mtu'].to_s,   #This must be string
              name: "eth#{index}",
              nictype: 'bridged',
              parent: config[:default][:root_bridge] + "-" + iface['vlan']['vid'].to_s,
              type: 'nic'
            }
          end
        end
      else
        abort("root_bridge #{config[:default][:root_bridge]} does not exist.") \
           unless conn.networks.include? config[:default][:root_bridge]

        root_bridge_mtu = nil
        # It assumes you only use one fabric as of now,
        # since there might be more fabrics with each untagged vlans on them,
        # which might make finding exact mtu fail as following process.
        default_fabric = 'fabric-0'

        maas.get_subnets.each do |subnet|
          if subnet['vlan']['name'] == 'untagged' and \
              subnet['vlan']['fabric'] == default_fabric
            root_bridge_mtu = subnet['vlan']['mtu']
            break
          end
        end

        lxd_params[:devices] = {}
        lxd_params[:devices][:"eth0"] = {
          mtu: root_bridge_mtu.to_s,   #This must be string
          name: 'eth0',
          nictype: 'bridged',
          parent: config[:default][:root_bridge],
          type: 'nic'
        }
      end

      if options[:'maas-on-lxc']
        # https://docs.maas.io/2.4/en/installconfig-lxd-install
        for i in 0..7
          i = i.to_s
          lxd_params[:devices]["loop" + i] = {}
          lxd_params[:devices]["loop" + i]["path"] = "/dev/loop" + i
          lxd_params[:devices]["loop" + i]["type"] = "unix-block"
        end
      end

      return lxd_params
    end

    def reserve_ips(name, options, container)
      logger.info("Calling <#{__method__.to_s}>")
      # Generate params to reserve IPs
      config[:ifaces].each_with_index do |iface,index|
        if index == 0
          params = {
            'subnet' => iface['cidr'],
            'ip' => iface['ip'],
            'hostname' => name,
            'mac' => container[:expanded_config][:"volatile.eth#{index}.hwaddr"]
          }
        elsif index > 0
          # if dot, '.', is used as a conjunction instead of '-',
          # it fails ocuring '404 not found'.
          # if under score, '_', is used as a conjunction instead of '-',
          # it breaks MAAS DNS somehow..
          if config[:ifaces][0]['vlan']['name'] == 'untagged'
            params = {
              'subnet' => iface['cidr'],
              'ip' => iface['ip'],
              'hostname' => 'eth0' + '-' + iface['vlan']['vid'].to_s  + '-' + name,
              'mac' => container[:expanded_config][:"volatile.eth0.hwaddr"]
            }
          elsif config[:ifaces][0]['vlan']['name'] != 'untagged'
            params = {
              'subnet' => iface['cidr'],
              'ip' => iface['ip'],
              'hostname' => "eth#{index}" + '-' + name,
              'mac' => container[:expanded_config][:"volatile.eth#{index}.hwaddr"]
            }
          end
        end
        maas.ipaddresses('reserve', params)
      end
    end

    def create(name, options = {})
      logger.info("Calling <#{__method__.to_s}>")

      abort("Container #{name} already exists!") \
        if container_exists?(name)

      abort("Domain #{name}.#{maas.get_domain} already exists!") \
        if maas.domain_name_exists?(name) unless options[:'no-maas']

      lxd_params = {}

      if options[:alias].nil? or options[:alias].empty?
        lxd_params[:alias] = config[:lxd][:default_alias]
      else
        lxd_params[:alias] = options[:alias]
      end

      lxd_params = generate_user_data(lxd_params, options)
      lxd_params = generate_network_config(lxd_params, options)
      lxd_params = generate_devices(lxd_params, options)

      lxd_params[:sync] ||= true

      conn.create_container(name, lxd_params)
      container = conn.container(name)

      container.devices = lxd_params[:devices].merge!(container.devices.to_hash)

      # https://github.com/jeffshantz/hyperkit/blob/master/lib/hyperkit/client/containers.rb#L240
      # Adding configurations that are necessary for shipping MAAS on lxc
      if options[:'maas-on-lxc']
        container.config = container.config.to_hash
        # https://docs.maas.io/2.4/en/installconfig-lxd-install
        container.config[:"raw.lxc"] = "\
lxc.cgroup.devices.allow = c 10:237 rwm\n\
lxc.aa_profile = unconfined\n\
lxc.cgroup.devices.allow = b 7:* rwm"
      end

      conn.update_container(name, container)
      # Fetch container object again
      container = conn.container(name)

      reserve_ips(name, options, container) \
        if options[:vlans] or options[:ipaddresses] \
          unless options[:'no-maas']

      conn.start_container(name, :sync=>"true")

      if options[:'no-maas']
        ip_or_fqdn = config[:ip_to_access]
      else
        ip_or_fqdn = name + '.' + maas.get_domain
      end

      if config[:default][:user] == config[:cloud_init][:users][0]['name']
          default_user = config[:default][:user]
      else
        if conn.execute_command(name, "ls /etc/lsb-release")[:metadata][:return] == 0
          default_user = 'ubuntu'
        elsif conn.execute_command(name, "ls /etc/redhat-release")[:metadata][:return] == 0
          default_user = 'centos'
        else
          default_user = config[:default][:user]
        end
      end

      lxd_params[:default_user] = default_user

      wait_until_available(ip_or_fqdn, default_user)
      logger.info("#{name} has been created.")

      if options[:'no-maas']
        puts "ssh #{default_user}@#{config[:ip_to_access]}"
      else
        puts "ssh #{default_user}@#{name}"
      end

      { result: true, info: lxd_params }
    end

    def destroy(name, lxd_params = {})
      logger.info("Calling <#{__method__.to_s}>")

      container = conn.container(name)
      lxd_params[:sync] ||= true

      info = container.to_hash

      if get_state(name) == 'Running'
        conn.stop_container(name, lxd_params)
      end

      wait_until_state(name, 'Stopped')

      if YAML.load(container[:config][:"user.user-data"])['maas']
        logger.info("This is a MAAS enabled container.")
        if container[:config][:"user.network-config"]
          net_conf = YAML.load(
            container[:config][:"user.network-config"]
          )['config']
          # To remove DNS configuration
          net_conf.shift

          net_conf.each do |nic|
            if nic['subnets'][0]['type'] == 'static'
              # It assumes we only assign a single subnet on a VLAN.
              # Subnets in a VLAN, VLANs in a Fabric
              ip = nic['subnets'][0]['address'].split('/')[0]

              if maas.ipaddresses_reserved?(ip)
                maas.ipaddresses('release', { 'ip' => ip })
              end
            end
          end
        end

        maas.delete_dns_record(name)
      end if container[:config][:"user.user-data"]

      conn.delete_container(name, lxd_params)

      # When multiple static IPs were reserved, it will not delete anything
      # since they are deleted when releasing the IPs above.
      logger.info("#{name} has been destroyed.")

      { result: true, info: info }
    end
  end
end