getch 0.1.0 → 0.1.6

data/lib/getch/gentoo/sources.rb

@@ -3,31 +3,37 @@ module Getch
     class Sources
       def initialize
         @lsmod = `lsmod`.chomp
-        @filesystem = Getch.class_fs::Deps.new()
+        @class_fs = Getch::select_fs
+        @filesystem = @class_fs::Deps.new
       end
 
       def build_others
+        cryptsetup
         virtualbox_guest
         qemu_guest
         install_wifi
         install_audio
+        flash_mod
       end
 
       def build_kspp
         puts "Adding KSPP to the kernel source"
-        garden("-b -a systemd")
+        bask("-b -a systemd")
       end
 
       def make
-        if DEFAULT_OPTIONS[:fs] == 'lvm' || DEFAULT_OPTIONS[:fs] == 'zfs' || DEFAULT_OPTIONS[:encrypt]
+        if Getch::OPTIONS[:fs] == 'lvm' ||
+            Getch::OPTIONS[:fs] == 'zfs' ||
+            Getch::OPTIONS[:encrypt]
           @filesystem.make
         else
           make_kernel
         end
       end
 
-      def init_config
-        Getch::Make.new("make localyesconfig").run!
+      def firewall
+        bask("-a iptables")
+        Getch::Emerge.new("net-firewall/iptables").pkg!
       end
 
       private
@@ -40,40 +46,76 @@ module Getch
         raise "No kernel installed, compiling source fail..." if is_kernel == []
       end
 
+      def cryptsetup
+        return unless Getch::OPTIONS[:encrypt]
+        make_conf = "#{MOUNTPOINT}/etc/portage/make.conf"
+
+        puts "Adding support for cryptsetup."
+        bask("-a cryptsetup")
+        Getch::Chroot.new("euse -E cryptsetup").run! unless Helpers::grep?(make_conf, /cryptsetup/)
+        Getch::Emerge.new("sys-fs/cryptsetup").pkg!
+      end
+
       def virtualbox_guest
         systemd=`systemd-detect-virt`.chomp
         return if ! ismatch?('vmwgfx') || systemd.match(/none/)
-        garden("-a virtualbox-guest")
+        bask("-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')
+        bask("-a kvm-host") if ismatch?('kvm')
+        bask("-a kvm-guest") if ismatch?('virtio')
       end
 
       def ismatch?(arg)
         @lsmod.match?(/#{arg}/)
       end
 
-      def garden(cmd)
-        Getch::Garden.new(cmd).run!
+      def bask(cmd)
+        Getch::Bask.new(cmd).run!
       end
 
       def install_wifi
         return if ! ismatch?('cfg80211')
-        garden("-a wifi")
+        bask("-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")
+        bask("-a sound")
       end
 
       def wifi_drivers
-        garden("-a ath9k-driver") if ismatch?('ath9k')
+        conf = "#{MOUNTPOINT}/etc/modules-load.d/wifi.conf"
+        File.delete(conf) if File.exists? conf
+
+        if ismatch?('ath9k')
+          bask("-a ath9k-driver")
+        end
+
+        module_load("iwlmvm", conf)
+        module_load("ath9k", conf)
+      end
+
+      def flash_mod
+        conf = "#{MOUNTPOINT}/etc/modules-load.d/usb.conf"
+        File.delete(conf) if File.exists? conf
+
+        module_load("ehci_pci", conf)
+        module_load("rtsx_pci_sdmmc", conf)
+        module_load("sdhci_pci", conf)
+        module_load("uas", conf)
+        module_load("uhci_hcd", conf)
+        module_load("xhci_pci", conf)
+      end
+
+      def module_load(name, file)
+        return unless name
+        return unless ismatch?(name)
+        File.write(file, "#{name}\n", mode: 'a')
       end
     end
   end

data/lib/getch/gentoo/stage.rb

@@ -4,7 +4,6 @@ require 'open3'
 module Getch
   module Gentoo
     class Stage
-
       def initialize
         @mirror = "https://mirrors.soeasyto.com/distfiles.gentoo.org"
         @release = release

data/lib/getch/gentoo/use.rb

@@ -0,0 +1,43 @@
+module Getch
+  module Gentoo
+    class Use
+      def initialize(pkg = nil)
+        @use_dir = "#{MOUNTPOINT}/etc/portage/package.use"
+        @pkg = pkg
+        @file = @pkg ? @pkg.match(/[\w]+$/) : nil
+        @make = "#{MOUNTPOINT}/etc/portage/make.conf"
+      end
+
+      def add(*flags)
+        @flags = flags.join(' ')
+        write
+      end
+
+      def add_global(*flags)
+        @flags = flags
+        write_global
+      end
+
+      private
+
+      def write
+        content = "#{@pkg} #{@flags}\n"
+        File.write("#{@use_dir}/#{@file}", content, mode: 'w')
+      end
+
+      def write_global
+        list = []
+
+        @flags.each { |f|
+          unless Helpers::grep?(@make, /#{f}/)
+            list << f
+          end
+        }
+
+        use = list.join(' ')
+        line = "USE=\"${USE} #{use}\"\n"
+        File.write(@make, line, mode: 'a')
+      end
+    end
+  end
+end

data/lib/getch/gentoo/use_flag.rb

@@ -0,0 +1,63 @@
+# lib/use_flag.rb
+
+module Getch::Gentoo
+  class UseFlag
+    def initialize
+      @efi = Helpers::efi?
+    end
+
+    def apply
+      systemd
+      kmod
+      grub
+      zfs
+      lvm
+      cryptsetup
+    end
+
+    private
+
+    def systemd
+      flags = []
+      use = Getch::Gentoo::Use.new('sys-apps/systemd')
+      flags << 'dns-over-tls'
+      flags << 'gnuefi' if @efi
+      use.add(flags)
+    end
+
+    def kmod
+      use = Getch::Gentoo::Use.new('sys-apps/kmod')
+      use.add('zstd', 'lzma')
+    end
+
+    def grub
+      return if @efi
+      flags = []
+      use = Getch::Gentoo::Use.new('sys-boot/grub')
+      flags << '-grub_platforms_efi-64'
+      flags << 'libzfs' if Getch::OPTIONS[:fs] == 'zfs'
+      flags << 'device-mapper' if Getch::OPTIONS[:fs] == 'lvm'
+      use.add(flags)
+    end
+
+    def zfs
+      return unless Getch::OPTIONS[:fs] == 'zfs'
+      use = Getch::Gentoo::Use.new('sys-fs/zfs-kmod')
+      use.add('rootfs')
+      use = Getch::Gentoo::Use.new('sys-fs/zfs')
+      use.add('rootfs')
+    end
+
+    def lvm
+      return unless Getch::OPTIONS[:fs] == 'lvm'
+      use = Getch::Gentoo::Use.new
+      use.add_global('lvm', 'device-mapper')
+    end
+
+    def cryptsetup
+      return unless Getch::OPTIONS[:encrypt]
+      use = Getch::Gentoo::Use.new
+      use.add_global('cryptsetup')
+    end
+  end
+end

data/lib/getch/guard.rb

@@ -0,0 +1,64 @@
+class InvalidDisk < StandardError
+end
+
+class InvalidFormat < StandardError
+end
+
+class InvalidZone < StandardError
+end
+
+class InvalidKeymap < StandardError
+end
+
+module Getch::Guard
+  def self.disk(name)
+    raise InvalidDisk, "No disk." unless name
+    raise InvalidDisk, "Bad device name #{name}." unless name.match(/^sd[a-z]{1}$/)
+    raise InvalidDisk, "Disk /dev/#{name} no found." unless File.exist? "/dev/#{name}"
+    name
+  rescue InvalidDisk => e
+    puts "#{e.class} => #{e}"
+    exit 1
+  end
+
+  def self.format(name)
+    raise InvalidFormat, "No format specified." unless name
+    raise InvalidFormat, "Format #{name} not yet available." if name.match(/btrfs/)
+    raise InvalidFormat, "Format #{name} not supported." unless name.match(/zfs|lvm|ext4/)
+    name
+  rescue InvalidFormat => e
+    puts "#{e.class} => #{e}"
+    exit 1
+  end
+
+  def self.zone(name)
+    raise InvalidZone, "No zoneinfo specified." unless name
+    raise InvalidZone, "Directory /usr/share/zoneinfo/ no found on this system..." unless Dir.exist? "/usr/share/zoneinfo/"
+    raise InvalidZone, "Zoneinfo #{name} is no found in /usr/share/zoneinfo/." unless File.exist? "/usr/share/zoneinfo/#{name}"
+    name
+  rescue InvalidZone => e
+    puts "#{e.class} => #{e}"
+    exit 1
+  end
+
+  def self.keymap(name)
+    raise InvalidKeymap, "No keymap specified." unless name
+    key = []
+
+    if Dir.exist? '/usr/share/keymaps'
+      key = Dir.glob("/usr/share/keymaps/**/#{name}.map.gz")
+    elsif Dir.exist? '/usr/share/kbd/keymaps'
+      key = Dir.glob("/usr/share/kbd/keymaps/**/#{name}.map.gz")
+    else
+      raise InvalidKeymap, "No directory found for keymap."
+    end
+
+    if key == []
+      raise InvalidKeymap, "Keymap #{name} no found."
+    end
+    name
+  rescue InvalidKeymap => e
+    puts "#{e.class} => #{e}"
+    exit 1
+  end
+end

data/lib/getch/helpers.rb

@@ -38,6 +38,11 @@ module Helpers
     File.write file, '' if ! File.exist? file
   end
 
+  def self.cp(src, dest)
+    raise "Src file #{src} no found" unless File.exist? src
+    FileUtils.cp(src, dest)
+  end
+
   def self.grep?(file, regex)
     is_found = false
     return is_found if ! File.exist? file
@@ -55,4 +60,132 @@ module Helpers
       raise "Error with #{cmd}"
     end
   end
+
+  def self.partuuid(dev)
+    `lsblk -o PARTUUID #{dev}`.match(/[\w]+-[\w]+-[\w]+-[\w]+-[\w]+/)
+  end
+
+  def self.uuid(dev)
+    Dir.glob("/dev/disk/by-uuid/*").each { |f|
+      if File.readlink(f).match(/#{dev}/)
+        return f.delete_prefix("/dev/disk/by-uuid/")
+      end
+    }
+  end
+
+  # Used with ZFS for the pool name
+  def self.pool_id(dev)
+    if dev.match(/[0-9]/)
+      sleep 1
+      `lsblk -o PARTUUID #{dev}`.delete("\n").delete("PARTUUID").match(/[\w]{5}/)
+    else
+      puts "Please, enter a pool name"
+      while true
+        print "\n> "
+        value = gets
+        if value.match(/[a-z]{4,20}/)
+          return value
+        end
+        puts "Bad name, you enter: #{value}"
+        puts "Valid pool name use character only, between 4-20."
+      end
+    end
+  end
+
+  module Void
+    def command(args)
+      print " => Exec: #{args}..."
+      cmd = "chroot #{Getch::MOUNTPOINT} /bin/bash -c \"#{args}\""
+      _, stderr, status = Open3.capture3(cmd)
+      if status.success? then
+        puts "\s[OK]"
+        return
+      end
+      raise "\n[-] Fail cmd #{args} - #{stderr}."
+    end
+
+    def command_output(args)
+      print " => Exec: #{args}..."
+      cmd = "chroot #{Getch::MOUNTPOINT} /bin/bash -c \"#{args}\""
+      Open3.popen2e(cmd) do |stdin, stdout_err, wait_thr|
+        puts
+        while line = stdout_err.gets
+          puts line
+        end
+
+        exit_status = wait_thr.value
+        unless exit_status.success?
+          raise "\n[-] Fail cmd #{args} - #{stdout_err}."
+        end
+      end
+    end
+
+    def add_line(file, line)
+      raise "No file #{file} found !" unless File.exist? file
+      File.write(file, "#{line}\n", mode: 'a')
+    end
+
+    def search(file, text)
+      File.open(file).each { |line|
+        return true if line.match(/#{text}/)
+      }
+      return false
+    end
+
+    # Used only when need password
+    def chroot(cmd)
+      if !system("chroot", Getch::MOUNTPOINT, "/bin/bash", "-c", cmd)
+        raise "[-] Error with: #{cmd}"
+      end
+    end
+
+    def s_uuid(dev)
+      device = dev.delete_prefix("/dev/")
+      Dir.glob("/dev/disk/by-partuuid/*").each { |f|
+        link = File.readlink(f)
+        return f.delete_prefix("/dev/disk/by-partuuid/") if link.match(/#{device}$/)
+      }
+    end
+
+    def line_fstab(dev, rest)
+      conf = "#{Getch::MOUNTPOINT}/etc/fstab"
+      device = s_uuid(dev)
+      raise "No partuuid for #{dev} #{device}" if !device
+      raise "Bad partuuid for #{dev} #{device}" if device.kind_of? Array
+      add_line(conf, "PARTUUID=#{device} #{rest}")
+    end
+
+    def grub_cmdline(*args)
+      conf = "#{Getch::MOUNTPOINT}/etc/default/grub"
+      list = args.join(" ")
+      secs = "GRUB_CMDLINE_LINUX=\"#{list} init_on_alloc=1 init_on_free=1"
+      secs += " slab_nomerge pti=on slub_debug=ZF vsyscall=none\""
+      raise "No default/grub found" unless File.exist? conf
+      unless search(conf, "GRUB_CMDLINE_LINUX=")
+        File.write(conf, "#{secs}\n", mode: 'a')
+      end
+    end
+  end
+
+  module Cryptsetup
+    def encrypt(dev)
+      raise "No device #{dev}" unless File.exist? dev
+      puts " => Encrypting device #{dev}..."
+      if Helpers::efi? && Getch::OPTIONS[:os] == 'gentoo'
+        Helpers::sys("cryptsetup luksFormat --type luks #{dev}")
+      else
+        Helpers::sys("cryptsetup luksFormat --type luks1 #{dev}")
+      end
+    end
+
+    def open_crypt(dev, map_name)
+      raise "No device #{dev}" unless File.exist? dev
+      puts " => Opening encrypted device #{dev}..."
+      if Helpers::efi? && Getch::OPTIONS[:os] == 'gentoo'
+        Helpers::sys("cryptsetup open --type luks #{dev} #{map_name}")
+      else
+        Helpers::sys("cryptsetup open --type luks1 #{dev} #{map_name}")
+      end
+    end
+  end
 end