module Getch
  module Gentoo
    class Sources
      def initialize
        @lsmod = `lsmod`.chomp
        @filesystem = OPTIONS_FS[DEFAULT_OPTIONS[:fs]]::Deps.new()
      end

      def build_others
        install_zfs if ismatch?('zfs')
        virtualbox_guest
        qemu_guest
        install_wifi
        install_audio
      end

      def build_kspp
        puts "Adding KSPP to the kernel source"
        garden("-b -a systemd")
      end

      def make
        if DEFAULT_OPTIONS[:fs] == 'lvm'
          @filesystem.make
        else
          just_make
        end
      end

      def init_config
        Getch::Make.new("make localyesconfig").run!
      end

      private

      def only_make
        Getch::Make.new("make -j$(nproc)").run!
      end

      def just_make
        puts "Compiling kernel sources"
        cmd = "make -j$(nproc) && make modules_install && make install"
        Getch::Make.new(cmd).run!
        is_kernel = Dir.glob("#{MOUNTPOINT}/boot/vmlinuz-*")
        raise "No kernel installed, compiling source fail..." if is_kernel == []
      end

      def virtualbox_guest
        return if ! ismatch?('vmwgfx')
        garden("-a virtualbox-guest")
        Getch::Emerge.new("app-emulation/virtualbox-guest-additions").pkg!
      end

      def qemu_guest
        garden("-a kvm-guest") if ismatch?('virtio')
        garden("-a kvm") if ismatch?('kvm')
      end

      def ismatch?(arg)
        @lsmod.match?(/#{arg}/)
      end

      def garden(cmd)
        Getch::Garden.new(cmd).run!
      end

      def install_wifi
        return if ! ismatch?('cfg80211')
        garden("-a wifi")
        wifi_drivers
        Getch::Emerge.new("net-wireless/iw wpa_supplicant net-wireless/iwd").pkg!
      end

      def install_audio
        return if ! ismatch?('snd_pcm')
        garden("-a sound")
      end

      def install_zfs
        garden("-a zfs")
        only_make # a first 'make' is necessary before emerge zfs
        Getch::Emerge.new("sys-fs/zfs").pkg!
      end

      def wifi_drivers
        garden("-a ath9k-driver") if ismatch?('ath9k')
      end
    end
  end
end