getch 0.1.9 → 0.3.3

data/lib/getch/assembly.rb

@@ -0,0 +1,150 @@
+# frozen_string_literal: true
+
+require 'clean'
+require 'nito'
+require 'cryptsetup'
+
+module Getch
+  class Assembly
+    include NiTo
+
+    def initialize
+      @os = Tree::Os.new.select
+      @fs = Tree::FS.new.select
+      @state = Getch::States.new
+      Getch::Device.new
+      init_devs
+    end
+
+    def init_devs
+      DEVS[:root] && return
+
+      @fs::Device.new
+      DEVS[:root] || Log.new.fatal('No root, device prob !')
+    end
+
+    def clean
+      return if STATES[:partition]
+
+      print "\nPartition and format disk #{OPTIONS[:disk]}, this will erase all data, continue? (y,N) "
+      case gets.chomp
+      when /^y|^Y/
+      else
+        exit
+      end
+
+      Clean.new(OPTIONS).x
+    end
+
+    def partition
+      return if STATES[:partition]
+
+      @fs::Partition.new
+      @state.partition
+    end
+
+    def format
+      return if STATES[:format]
+
+      @fs::Format.new
+      @state.format
+    end
+
+    def mount
+      return if STATES[:mount]
+
+      @fs::Mount.new
+      @state.mount
+    end
+
+    def tarball
+      return if STATES[:tarball]
+
+      @os::Tarball.new.x
+      @state.tarball
+    end
+
+    # pre_config
+    # Pre configuration before updates and install packages
+    # Can contain config for a repository, CPU compilation flags, etc...
+    def pre_config
+      return if STATES[:pre_config]
+
+      @os::PreConfig.new
+      @state.pre_config
+    end
+
+    # update
+    # Synchronise and Update the new system
+    def update
+      return if STATES[:update]
+
+      Helpers.mount_all
+      @os::Update.new
+      @state.update
+    end
+
+    def post_config
+      return if STATES[:post_config]
+
+      @os::PostConfig.new
+      @state.post_config
+    end
+
+    # terraform
+    # Install all the required packages
+    # Also add services
+    def terraform
+      return if STATES[:terraform]
+
+      #@fs::PreDeps.new
+      @os::Terraform.new
+      @fs::Deps.new
+      @state.terraform
+    end
+
+    def services
+      return if STATES[:services]
+
+      @os::Services.new
+      @state.services
+    end
+
+    # Luks_keys
+    # Install external keys to avoid enter password multiple times
+    def luks_keys
+      return if not OPTIONS[:encrypt] or OPTIONS[:fs] == 'zfs'
+
+      return if STATES[:luks_keys]
+
+      CryptSetup.new(DEVS, OPTIONS).keys
+      @state.luks_keys
+    end
+
+    # bootloader
+    # Install and configure Grub2 or Systemd-boot with Dracut
+    # Adding keys for Luks
+    def bootloader
+      return if STATES[:bootloader]
+
+      bootloader = @os::Bootloader.new
+      bootloader.dependencies
+      @fs::Config.new
+      bootloader.install
+      @state.bootloader
+    end
+
+    # finalize
+    # Password for root, etc
+    def finalize
+      return if STATES[:finalize]
+
+      @os::Finalize.new
+      puts
+      puts '[*!*] Installation finished [*!*]'
+      puts
+      @fs.end
+      @state.finalize
+    end
+  end
+end

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