getch 0.1.9 → 0.3.0

data/lib/getch/command.rb

@@ -1,161 +1,71 @@
 # frozen_string_literal: true
 
 require 'open3'
+require 'nito'
 
 module Getch
   class Command
-    def initialize(cmd)
-      @cmd = cmd
-      @block_size = 1024
+    attr_reader :res
+
+    def initialize(*args)
+      @cmd = args.join(' ')
       @log = Getch::Log.new
+      x
     end
 
-    def run!
-      @log.info 'Running command: ' + @cmd.gsub(/\"/, '')
-
-      Open3.popen3(@cmd) do |stdin, stdout, stderr, wait_thr|
-        stdin.close_write
-        code = wait_thr.value
-
-        # only stderr
-        begin
-          @log.debug stderr.readline until stderr.eof.nil?
-        rescue
-        end
+    def to_s
+      @res
+    end
 
-        begin
-          files = [stdout, stderr]
+    protected
 
-          until all_eof(files) do
-            ready = IO.select(files)
+    def x
+      @log.info 'Exec: ' + @cmd
+      cmd = build_cmd
 
-            if ready
-              readable = ready[0]
-              # writable = ready[1]
-              # exceptions = ready[2]
+      Open3.popen3(cmd) do |stdin, stdout, stderr, wait_thr|
+        stdin.close_write
+        code = wait_thr.value
 
-              display_lines(readable)
-            end
+        unless code.success?
+          begin
+            @log.debug stderr.readline until stderr.eof.nil?
+          rescue EOFError
+            print
           end
-        rescue IOError => e
-          puts "IOError: #{e}"
         end
 
-        unless code.success?
-          @log.fatal "Running #{@cmd}"
-          exit 1
+        if code.success?
+          @log.result_ok
+          @res = stdout.read.chomp
+          return
         end
 
-        @log.debug "Done - #{@cmd} - #{code}"
-      end
-    end
-
-    private
-
-    # Returns true if all files are EOF
-    def all_eof(files)
-      files.find { |f| !f.eof }.nil?
-    end
-
-    def display_lines(block)
-      block.each do |f|
-        begin
-          data = f.read_nonblock(@block_size)
-          puts data if OPTIONS[:verbose]
-        rescue EOFError
-          puts
-        rescue => e
-          puts "Fatal - #{e}"
-        end
-      end
-    end
-  end
-
-  # Use system, the only ruby method to display stdout with colors !
-  class Emerge
-    def initialize(cmd)
-      @gentoo = MOUNTPOINT
-      @cmd = cmd
-      @log = Getch::Log.new
-    end
-
-    def run!
-      @log.info "Running emerge: #{@cmd}"
-      system('chroot', @gentoo, '/bin/bash', '-c', "source /etc/profile && #{@cmd}")
-      read_exit
-    end
-
-    def pkg!
-      @log.info "Running emerge pkg: #{@cmd}"
-      system('chroot', @gentoo, '/bin/bash', '-c', "source /etc/profile && emerge --changed-use #{@cmd}")
-      read_exit
-    end
-
-    private
-
-    def read_exit
-      if $?.exitstatus > 0
+        puts
+        @log.error "#{@cmd} - #{code}"
         @log.fatal "Running #{@cmd}"
-      else
-        @log.info "Done #{@cmd}"
       end
     end
-  end
 
-  class Make
-    def initialize(cmd)
-      @gentoo = MOUNTPOINT
-      @cmd = cmd
-      @log = Getch::Log.new
-    end
+    private
 
-    def run!
-      @log.info "Running Make: #{@cmd}"
-      cmd = "chroot #{@gentoo} /bin/bash -c \"source /etc/profile \
-        && env-update \
-        && cd /usr/src/linux \
-        && #{@cmd}\""
-      Open3.popen2e(cmd) do |_, stdout_err, wait_thr|
-        stdout_err.each { |l| puts l }
-
-        exit_status = wait_thr.value
-        unless exit_status.success?
-          @log.fatal "Running #{cmd}"
-          exit 1
-        end
-      end
+    def build_cmd
+      @cmd
     end
   end
 
   class Bask
     def initialize(cmd)
       @cmd = cmd
-      @log = Getch::Log.new
+      @log = Log.new
       @version = '0.6'
       @config = "#{MOUNTPOINT}/etc/kernel/config.d"
       download_bask unless Dir.exist? "#{MOUNTPOINT}/root/bask-#{@version}"
     end
 
-    def run!
-      @log.info "Running Bask: #{@cmd}"
-      cmd = "chroot #{MOUNTPOINT} /bin/bash -c \"source /etc/profile \
-        && env-update \
-        && cd /root/bask-#{@version} \
-        && ./bask.sh #{@cmd} -k /usr/src/linux\""
-      Open3.popen2e(cmd) do |_, stdout_err, wait_thr|
-        stdout_err.each { |l| puts l }
-
-        exit_status = wait_thr.value
-        unless exit_status.success?
-          @log.fatal "Running #{cmd}"
-          exit 1
-        end
-      end
-    end
-
     def cp
-      Helpers.mkdir @config
-      Helpers.cp(
+      NiTo.mkdir @config
+      NiTo.cp(
         "#{MOUNTPOINT}/root/bask-#{@version}/config.d/#{@cmd}",
         "#{@config}/#{@cmd}"
       )
@@ -168,20 +78,70 @@ module Getch
     private
 
     def download_bask
-      @log.info 'Installing Bask...'
+      @log.info "Installing Bask...\n"
       url = "https://github.com/szorfein/bask/archive/refs/tags/#{@version}.tar.gz"
       file = "bask-#{@version}.tar.gz"
 
       Dir.chdir("#{MOUNTPOINT}/root")
       Helpers.get_file_online(url, file)
-      Getch::Command.new("tar xzf #{file}").run!
+      Getch::Command.new("tar xzf #{file}")
     end
   end
 
   class Chroot < Command
-    def initialize(cmd)
-      super
-      @cmd = "chroot #{MOUNTPOINT} /bin/bash -c \"source /etc/profile; #{cmd}\""
+    def build_cmd
+      dest = OPTIONS[:mountpoint]
+      case OPTIONS[:os]
+      when 'gentoo'
+        "chroot #{dest} /bin/bash -c \"source /etc/profile; #{@cmd}\""
+      when 'void'
+        "chroot #{dest} /bin/bash -c \"#{@cmd}\""
+      end
+    end
+  end
+
+  class ChrootOutput
+    def initialize(*args)
+      @cmd = args.join(' ')
+      @log = Log.new
+      x
+    end
+
+    private
+
+    def x
+      msg
+      system('chroot', OPTIONS[:mountpoint], '/bin/bash', '-c', other_args)
+      $?.success? && return
+
+      @log.fatal "Running #{@cmd}"
+    end
+
+    def msg
+      @log.info "Exec: #{@cmd}...\n"
+    end
+
+    def other_args
+      case OPTIONS[:os]
+      when 'gentoo' then "source /etc/profile && #{@cmd}"
+      when 'void' then @cmd
+      end
+    end
+  end
+
+  # Install
+  # use system() to install packages
+  # Usage: Install.new(pkg_name)
+  class Install < ChrootOutput
+    def msg
+      @log.info "Installing #{@cmd}...\n"
+    end
+
+    def other_args
+      case OPTIONS[:os]
+      when 'gentoo' then "source /etc/profile && emerge --changed-use #{@cmd}"
+      when 'void' then "/usr/bin/xbps-install -y #{@cmd}"
+      end
     end
   end
 end

data/lib/getch/config/account.rb

@@ -0,0 +1,39 @@
+# frozen_string_literal: true
+
+module Getch
+  module Config
+    class Account
+      def initialize
+        @user = OPTIONS[:username] ||= nil
+        @log = Log.new
+      end
+
+      def root
+        @log.info "Add a password for root.\n"
+        ChrootOutput.new('passwd')
+      end
+
+      def new_user
+        return unless @user
+
+        create_user
+        @log.info "Add a password for #{@user}.\n"
+        ChrootOutput.new("passwd #{@user}")
+        fix_perm
+      end
+
+      private
+
+      def create_user
+        @log.info "Creating a new user #{@user}..."
+        Getch::Chroot.new("useradd -m -G users,wheel,audio,video #{@user}")
+        @log.result_ok
+      end
+
+      def fix_perm
+        Getch::Chroot.new("chown -R #{@user}:#{@user} /home/#{@user}")
+        Getch::Chroot.new("chmod 700 -R /home/#{@user}")
+      end
+    end
+  end
+end

data/lib/getch/config/dhcp.rb

@@ -0,0 +1,104 @@
+# frozen_string_literal: true
+
+require 'nito'
+
+module Getch
+  module Config
+    class Dhcp
+      include NiTo
+
+      def initialize
+        x
+      end
+
+      protected
+
+      def x
+        runit
+        systemd
+        openrc
+      end
+
+      private
+
+      # https://docs.voidlinux.org/config/network/iwd.html
+      def runit
+        Helpers.runit? || return
+
+        echo_a "#{OPTIONS[:mountpoint]}/etc/dhcpcd.conf", 'nohook resolv.conf'
+        resolv_conf
+        service = '/etc/runit/runsvdir/default/'
+        Chroot.new("ln -fs /etc/sv/dhcpcd #{service}")
+      end
+
+      def systemd
+        Helpers.systemd? || return
+
+        systemd_ethernet
+        systemd_wireless
+        systemd_resolve
+        Chroot.new('systemctl enable systemd-networkd')
+        Chroot.new('systemctl enable systemd-resolved')
+      end
+
+      def openrc
+        Helpers.openrc? || return
+
+        echo_a "#{OPTIONS[:mountpoint]}/etc/dhcpcd.conf", 'nohook resolv.conf'
+        resolv_conf
+        Chroot.new('rc-update add dhcpcd default')
+      end
+
+      # https://www.dnsknowledge.com/tutorials/how-to-setup-quad9-dns-on-a-linux/
+      def resolv_conf
+        conf = "#{OPTIONS[:mountpoint]}/etc/resolv.conf"
+        content = <<~CONF
+        nameserver 9.9.9.9
+        nameserver 2620:fe::fe
+        options rotate
+        CONF
+        File.write conf, "#{content}\n"
+      end
+
+      def systemd_ethernet
+        conf = "#{OPTIONS[:mountpoint]}/etc/systemd/network/20-ethernet.network"
+        content = <<~NETWORK
+        [Match]
+        Name=en*
+        Name=eth*
+        [Network]
+        DHCP=yes
+        IPv6PrivacyExtensions=yes
+        [DHCP]
+        RouteMetric=512
+        NETWORK
+        File.write(conf, "#{content}\n")
+      end
+
+      def systemd_wireless
+        conf = "#{OPTIONS[:mountpoint]}/etc/systemd/network/20-wireless.network"
+        content = <<~NETWORK
+        [Match]
+        Name=wl*
+        [Network]
+        DHCP=yes
+        IPv6PrivacyExtensions=yes
+        [DHCP]
+        RouteMetric=1024
+        NETWORK
+        File.write conf, "#{content}\n"
+      end
+
+      def systemd_resolve
+        mkdir "#{OPTIONS[:mountpoint]}/etc/systemd/resolved.conf.d"
+        conf = "#{OPTIONS[:mountpoint]}/etc/systemd/resolved.conf.d/dns_tls.conf"
+        content = <<~CONF
+        [Resolve]
+        DNS=9.9.9.9#dns.quad9.net
+        DNSOverTLS=yes
+        CONF
+        File.write conf, "#{content}\n"
+      end
+    end
+  end
+end

data/lib/getch/config/grub.rb

@@ -0,0 +1,42 @@
+# frozen_string_literal: true
+
+require 'nito'
+
+module Getch
+  module Config
+    class Grub
+      def initialize
+        @log = Log.new
+        @disk = OPTIONS[:boot_disk] ||= OPTIONS[:disk]
+        # https://wiki.archlinux.org/title/Install_Arch_Linux_on_ZFS
+        @prefix = OPTIONS[:fs] == 'zfs' ? 'ZPOOL_VDEV_NAME_PATH=1' : ''
+        @os_name = OPTIONS[:os].capitalize
+        x
+      end
+
+      def x
+        @log.info "Installing Grub on #{@disk}...\n"
+        Helpers.efi? ? grub_efi : grub_bios
+      end 
+
+      private
+
+      def grub_efi
+        mount_efivars
+        cmd = "#{@prefix} grub-install --target=x86_64-efi --efi-directory=/efi --bootloader-id=\"#{@os_name}\""
+        ChrootOutput.new(cmd)
+      end
+
+      def grub_bios
+        cmd = "#{@prefix} grub-install /dev/#{@disk}"
+        ChrootOutput.new(cmd)
+      end
+
+      # In case where efivars is not mounted
+      # avoid error with grub
+      def mount_efivars
+        NiTo.mount '-t efivarfs', 'efivarfs', '/sys/firmware/efi/efivars'
+      end
+    end
+  end
+end

data/lib/getch/config/iwd.rb

@@ -0,0 +1,60 @@
+module Getch
+  module Config
+    class Iwd
+      include NiTo
+
+      def initialize
+        x
+      end
+
+      protected
+
+      def x
+        runit
+        systemd
+        openrc
+      end
+
+      private
+
+      # https://docs.voidlinux.org/config/network/iwd.html
+      def runit
+        Helpers.runit? || return
+
+        iwd_conf
+        service = '/etc/runit/runsvdir/default/'
+        Chroot.new("ln -fs /etc/sv/dbus #{service}")
+        Chroot.new("ln -fs /etc/sv/iwd #{service}")
+      end
+
+      def systemd
+        Helpers.systemd? || return
+
+        iwd_conf
+        Chroot.new('systemctl enable iwd')
+      end
+
+      def openrc
+        Helpers.openrc? || return
+
+        iwd_conf
+        Chroot.new('rc-update add iwd default')
+      end
+
+      # https://docs.voidlinux.org/config/network/iwd.html#troubleshooting
+      def iwd_conf
+        conf = "#{OPTIONS[:mountpoint]}/etc/iwd/main.conf"
+        content = "[General]\n"
+        content << "UseDefaultInterface=true\n"
+        content << "[Network]\n"
+        Helpers.systemd? ?
+          content << "NameResolvingService=systemd\n" :
+          content << "NameResolvingService=resolvconf\n"
+        content << "[Scan]\n"
+        content << "DisablePeriodicScan=true\n"
+        mkdir "#{OPTIONS[:mountpoint]}/etc/iwd"
+        echo conf, "#{content}\n"
+      end
+    end
+  end
+end

data/lib/getch/config/keymap.rb

@@ -0,0 +1,78 @@
+# frozen_string_literal: true
+
+require 'nito'
+
+module Getch
+  module Config
+    # Search and configure the keymap (man loadkeys)
+    class Keymap
+      include NiTo
+
+      def initialize
+        @log = Log.new
+        @rc_conf = "#{OPTIONS[:mountpoint]}/etc/rc.conf"
+        @vconsole_conf = "#{OPTIONS[:mountpoint]}/etc/vconsole.conf"
+        @conf_d = "#{OPTIONS[:mountpoint]}/etc/conf.d/keymaps"
+        @keymaps_dir = nil
+        @keymap = nil
+        x
+      end
+
+      protected
+
+      def x
+        @log.info "Configuring keymap...\n"
+        search_keymap
+        apply_conf
+      end
+
+      def search_keymap
+        search_dir
+        path = "#{OPTIONS[:mountpoint]}#{@keymaps_dir}/**/#{OPTIONS[:keymap]}.map.gz"
+        Dir.glob(path) { |f| @keymap = OPTIONS[:keymap] if f }
+
+        @keymap || @log.fatal("No keymap found for #{OPTIONS[:keymap]}.")
+      end
+
+      def apply_conf
+        @log.info "Setting keymap to \"#{@keymap}\"..."
+        writing_rc_conf
+        writing_vconsole_conf
+        writing_conf_d_keymaps
+        @log.result_ok
+      end
+
+      def writing_rc_conf
+        return unless File.exist? @rc_conf
+
+        echo_a @rc_conf, "KEYMAP=\"#{@keymap}\""
+      end
+
+      def writing_vconsole_conf
+        return unless Helpers.systemd?
+
+        echo_a @vconsole_conf, "KEYMAP=\"#{@keymap}\""
+      end
+
+      def writing_conf_d_keymaps
+        return unless File.exist? @conf_d
+
+        sed @conf_d, /^keymap=/, "keymap=\"#{@keymap}\""
+      end
+
+      private
+
+      def search_dir
+        case OPTIONS[:os]
+        when 'gentoo' then @keymaps_dir = '/usr/share/keymaps'
+        when 'void' then @keymaps_dir = '/usr/share/kbd/keymaps'
+        else
+          @log.fatal('OPTIONS[:os] not supported yet.')
+        end
+
+        File.exist? "#{OPTIONS[:mountpoint]}#{@keymaps_dir}" ||
+          @log.fatal("No dir keymaps #{@keymaps_dir} found.")
+      end
+    end
+  end
+end

data/lib/getch/config/locale.rb

@@ -0,0 +1,96 @@
+# frozen_string_literal: true
+
+require 'nito'
+
+# Gentoo: https://wiki.gentoo.org/wiki/Handbook:AMD64/Installation/Base#Locale_generation
+# Void: https://docs.voidlinux.org/config/locales.html#locales-and-translations
+module Getch
+  module Config
+    class Locale
+      include NiTo
+
+      # Gentoo use i18n_supported
+      # Void use libc_locale
+      def initialize
+        @log = Log.new
+        @i18n_supported = "#{OPTIONS[:mountpoint]}/usr/share/i18n/SUPPORTED"
+        @libc_locales = "#{OPTIONS[:mountpoint]}/etc/default/libc-locales"
+        @locale_conf = "#{OPTIONS[:mountpoint]}/etc/locale.conf"
+        @i18n = nil
+        @lang = nil
+        x
+      end
+
+      def x
+        @log.info "Configuring locales...\n"
+        search_locale
+        apply_conf
+      end
+
+      protected
+
+      def search_locale
+        search_i18n
+        search_libc
+        lang
+      end
+
+      def apply_conf
+        return if OPTIONS[:musl]
+
+        File.exist?("#{OPTIONS[:mountpoint]}/etc/locale.gen") && write_locale_gen
+        File.exist?(@libc_locales) && write_libc_locales
+      end
+
+      def write_locale_gen
+        @log.fatal("No UTF8 locale found for #{OPTIONS[:language]}") unless @i18n
+
+        @log.info "Using locale #{@i18n}...\n"
+        echo "#{OPTIONS[:mountpoint]}/etc/locale.gen", @i18n
+        locale_conf
+        Getch::Chroot.new('locale-gen')
+      end
+
+      def write_libc_locales
+        @log.fatal("No UTF8 locale found for #{OPTIONS[:language]}") unless @i18n
+
+        @log.info "Using locale #{@i18n}...\n"
+        echo @libc_locales, @i18n
+        locale_conf
+        Getch::Chroot.new('xbps-reconfigure -f glibc-locales')
+      end
+
+      private
+
+      def search_i18n
+        return unless File.exist? @i18n_supported
+
+        File.open(@i18n_supported).each do |l|
+          @i18n = l.chomp if l =~ /#{OPTIONS[:language]}.*UTF-8$/
+        end
+      end
+
+      def search_libc
+        return unless File.exist? @libc_locales
+
+        File.open(@libc_locales).each do |l|
+          @i18n = l.tr('#', '').chomp if l =~ /\#?#{OPTIONS[:language]}.*UTF-8/
+        end
+      end
+
+      def lang
+        return unless @i18n
+
+        lang = @i18n.split(' ')
+        @lang = lang[0]
+      end
+
+      def locale_conf
+        return unless Helpers.systemd?
+
+        echo @locale_conf, "LANG=#{@lang}"
+        echo_a @locale_conf, 'LC_COLLATE=C.UTF-8'
+      end
+    end
+  end
+end