getch 0.1.9 → 0.3.0

data/lib/getch/gentoo/tarball.rb

@@ -0,0 +1,91 @@
+# frozen_string_literal: true
+
+require 'open-uri'
+require 'open3'
+
+module Getch
+  module Gentoo
+    class Tarball
+      def initialize
+        @log = Log.new
+        @mirror = 'https://mirror.rackspace.com/gentoo'
+        @release = release
+        @stage_file = OPTIONS[:musl] ?
+          "stage3-amd64-musl-#{@release}.tar.xz" :
+          "stage3-amd64-systemd-#{@release}.tar.xz"
+      end
+
+      def x
+        get_stage3
+        control_files
+        checksum
+        install
+      end
+
+      protected
+
+      def stage3
+        OPTIONS[:musl] ?
+          @mirror + '/releases/amd64/autobuilds/latest-stage3-amd64-musl.txt' :
+          @mirror + '/releases/amd64/autobuilds/latest-stage3-amd64-systemd.txt'
+      end
+
+      def release
+        URI.open(stage3) do |file|
+          file.read.match(/^[[:alnum:]]+/)
+        end
+      rescue Net::OpenTimeout => e
+        @log.fatal "Problem with DNS? #{e}"
+      end
+
+      def file
+        "#{@release}/#{@stage_file}"
+      end
+
+      def get_stage3
+        Dir.chdir OPTIONS[:mountpoint]
+        return if File.exist? @stage_file
+
+        @log.info "wget #{@stage_file}, please wait...\n"
+        Helpers.get_file_online(@mirror + '/releases/amd64/autobuilds/' + file, @stage_file)
+      end
+
+      def control_files
+        @log.info "Download other files..."
+        ['DIGESTS', 'asc', 'CONTENTS.gz'].each do |f|
+          Helpers.get_file_online("#{@mirror}/releases/amd64/autobuilds/#{file}.#{f}", "#{@stage_file}.#{f}")
+        end
+        @log.result_ok
+      end
+
+      def checksum
+        @log.info 'Checking SHA512 checksum...'
+        command = "awk '/SHA512 HASH/{getline;print}' #{@stage_file}.DIGESTS | sha512sum --check"
+        _, stderr, status = Open3.capture3(command)
+        if status.success? then
+          @log.result_ok
+        else
+          cleaning
+          @log.fatal "Problem with the checksum, stderr\n#{stderr}"
+        end
+      end
+
+      def install
+        decompress
+        cleaning
+      end
+
+      private
+
+      # https://wiki.gentoo.org/wiki/Handbook:AMD64/Installation/Stage
+      def decompress
+        cmd = "tar xpf #{@stage_file} --xattrs-include=\'*.*\' --numeric-owner"
+        Getch::Command.new(cmd)
+      end
+
+      def cleaning
+        Dir.glob('stage3-amd64-*').each { |f| File.delete(f) }
+      end
+    end
+  end
+end

data/lib/getch/gentoo/terraform.rb

@@ -0,0 +1,34 @@
+module Getch
+  module Gentoo
+    class Terraform
+      def initialize
+        x
+      end
+
+      protected
+
+      def x
+        Gentoo::Sources.new
+        install_pkgs
+        emerge_deep
+      end
+
+      def install_pkgs
+        @pkgs = 'app-portage/gentoolkit'
+        @pkgs << ' app-admin/sudo'
+        @pkgs << ' app-editors/vim'
+        @pkgs << ' net-firewall/iptables'
+        @pkgs << ' net-wireless/iwd'
+        @pkgs << ' net-misc/dhcpcd' unless Helpers.systemd?
+        @pkgs << ' sys-kernel/linux-firmware'
+        @pkgs << ' sys-firmware/intel-microcode'
+        @pkgs << ' sys-fs/dosfstools' if Helpers.efi?
+        Install.new(@pkgs)
+      end
+
+      def emerge_deep
+        ChrootOutput.new('emerge --deep --newuse @world')
+      end
+    end
+  end
+end

data/lib/getch/gentoo/update.rb

@@ -0,0 +1,54 @@
+# frozen_string_literal: true
+
+require 'nito'
+
+module Getch
+  module Gentoo
+    class Update
+      include NiTo
+
+      def initialize
+        @log = Log.new
+        x
+      end
+
+      protected
+
+      def x
+        sync
+        add_musl_repo if OPTIONS[:musl]
+        update
+      end
+
+      private
+
+      def sync
+        gentoo_conf = "#{OPTIONS[:mountpoint]}/etc/portage/repos.conf/gentoo.conf"
+        @log.info "Synchronize index, please waiting...\n"
+        ChrootOutput.new('emaint sync --auto')
+        sed gentoo_conf, /^sync-type/, 'sync-type = rsync'
+      end
+
+      def add_musl_repo
+        Install.new('dev-vcs/git')
+
+        file = "#{OPTIONS[:mountpoint]}/etc/portage/repos.conf/musl.conf"
+        content = <<~CONF
+        [musl]
+        location = /var/db/repos/musl
+        sync-type = git
+        sync-uri = https://github.com/gentoo/musl.git
+        auto-sync = Yes
+        CONF
+        File.write file, "#{content}\n"
+
+        ChrootOutput.new('emaint sync -r musl')
+      end
+
+      def update
+        cmd = 'emerge --update --deep --newuse @world'
+        ChrootOutput.new(cmd)
+      end
+    end
+  end
+end

data/lib/getch/gentoo/use.rb

@@ -1,13 +1,17 @@
 # frozen_string_literal: true
 
+require 'nito'
+
 module Getch
   module Gentoo
     class Use
+      include NiTo
+
       def initialize(pkg = nil)
-        @use_dir = "#{MOUNTPOINT}/etc/portage/package.use"
+        @use_dir = "#{OPTIONS[:mountpoint]}/etc/portage/package.use"
         @pkg = pkg
         @file = @pkg ? @pkg.match(/[\w]+$/) : nil
-        @make = "#{MOUNTPOINT}/etc/portage/make.conf"
+        @make = "#{OPTIONS[:mountpoint]}/etc/portage/make.conf"
       end
 
       def add(*flags)
@@ -24,15 +28,14 @@ module Getch
 
       def write
         content = "#{@pkg} #{@flags}\n"
-        File.write("#{@use_dir}/#{@file}", content, mode: 'w')
+        echo "#{@use_dir}/#{@file}", content
       end
 
       def write_global
         list = []
-        @flags.each { |f| list << f unless Helpers.grep?(@make, /#{f}/) }
+        @flags.each { |f| list << f unless grep?(@make, f) }
         use = list.join(' ')
-        line = "USE=\"${USE} #{use}\"\n"
-        File.write(@make, line, mode: 'a')
+        echo_a @make, "USE=\"${USE} #{use}\""
       end
     end
   end

data/lib/getch/gentoo/use_flag.rb

@@ -4,13 +4,15 @@ module Getch
   module Gentoo
     class UseFlag
       def initialize
-        @efi = Helpers.efi?
+        x
       end
 
-      def apply
+      protected
+
+      def x
+        dist_kernel
         systemd
         pam
-        kernel
         kmod
         grub
         zfs
@@ -20,41 +22,40 @@ module Getch
 
       private
 
+      # https://wiki.gentoo.org/wiki/Project:Distribution_Kernel#Trying_it_out
+      def dist_kernel
+        use = Getch::Gentoo::Use.new
+        use.add_global('dist-kernel')
+      end
+
       def systemd
+        return unless Helpers.systemd?
+
         flags = []
         use = Getch::Gentoo::Use.new('sys-apps/systemd')
         flags << 'dns-over-tls'
-        flags << 'gnuefi' if @efi
+        flags << 'gnuefi' if Helpers.efi?
         use.add(flags)
       end
 
       def pam
         flags = []
         use = Getch::Gentoo::Use.new('sys-auth/pambase')
-        flags << '-passwdqc'
-        flags << 'pwquality'
         flags << 'sha512'
         use.add(flags)
       end
 
-      def kernel
-        use = Getch::Gentoo::Use.new('sys-kernel/gentoo-kernel')
-        use.add('hardened')
-      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'
+        flags << '-grub_platforms_efi-64' unless Helpers.efi?
+        flags << 'libzfs' if OPTIONS[:fs] == 'zfs'
+        flags << 'device-mapper' if OPTIONS[:fs] == 'lvm' or OPTIONS[:encrypt]
         use.add(flags)
       end
 
@@ -65,10 +66,6 @@ module Getch
         use.add('rootfs')
         use = Getch::Gentoo::Use.new('sys-fs/zfs')
         use.add('rootfs')
-
-        # https://wiki.gentoo.org/wiki/Project:Distribution_Kernel
-        use = Getch::Gentoo::Use.new
-        use.add_global('dist-kernel')
       end
 
       def lvm

data/lib/getch/gentoo.rb

@@ -1,85 +1,19 @@
 # frozen_string_literal: true
 
-require_relative 'gentoo/stage'
-require_relative 'gentoo/config'
-require_relative 'gentoo/chroot'
-require_relative 'gentoo/bootloader'
 require_relative 'gentoo/sources'
-require_relative 'gentoo/boot'
 require_relative 'gentoo/use'
 require_relative 'gentoo/use_flag'
 
 module Getch
   module Gentoo
-    class Main
-      def initialize
-        @state = Getch::States.new
-      end
-
-      def stage3
-        return if STATES[:gentoo_base]
-
-        stage = Getch::Gentoo::Stage.new
-        stage.get_stage3
-        stage.control_files
-        stage.checksum
-        @state.stage3
-      end
-
-      def config
-        return if STATES[:gentoo_config]
-
-        config = Getch::Gentoo::Config.new
-        config.portage
-        config.portage_fs
-        config.portage_bashrc
-        config.repo
-        config.network
-        config.systemd
-        config.hostname
-        @state.config
-      end
-
-      def chroot
-        return if STATES[:gentoo_update]
-
-        chroot = Getch::Gentoo::Chroot.new
-        chroot.update
-        chroot.cpuflags
-        chroot.systemd
-
-        flags = Getch::Gentoo::UseFlag.new
-        flags.apply
-
-        chroot.world
-        chroot.kernel_license
-        chroot.install_pkgs
-        @state.update
-      end
-
-      def bootloader
-        return if STATES[:gentoo_bootloader]
-
-        bootloader = Getch::Gentoo::Bootloader.new
-        bootloader.start
-        @state.bootloader
-      end
-
-      def kernel
-        return if STATES[:gentoo_kernel]
-
-        source = Getch::Gentoo::Sources.new
-        source.bask
-        source.configs
-        source.make
-        source.load_modules
-        @state.kernel
-      end
-
-      def boot
-        boot = Getch::Gentoo::Boot.new
-        boot.start
-      end
-    end
   end
 end
+
+require_relative 'gentoo/tarball'
+require_relative 'gentoo/pre_config'
+require_relative 'gentoo/update'
+require_relative 'gentoo/post_config'
+require_relative 'gentoo/terraform'
+require_relative 'gentoo/services'
+require_relative 'gentoo/bootloader'
+require_relative 'gentoo/finalize'

data/lib/getch/guard.rb

@@ -27,8 +27,8 @@ module Getch
 
     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/)
+      raise InvalidFormat, "Format #{name} not yet available." if name.match(/btrfs|xfs/)
+      raise InvalidFormat, "Format #{name} not supported." unless name.match(/zfs|ext4/)
 
       name
     rescue InvalidFormat => e

data/lib/getch/helpers.rb

@@ -3,6 +3,7 @@
 require 'open-uri'
 require 'open3'
 require 'fileutils'
+require 'nito'
 
 module Getch
   module Helpers
@@ -10,63 +11,40 @@ module Getch
       Dir.exist? '/sys/firmware/efi/efivars'
     end
 
-    def self.get_file_online(url, dest)
-      URI.open(url) do |l|
-        File.open(dest, "wb") { |f| f.write(l.read) }
-      end
-    end
-
-    def self.exec_or_die(cmd)
-      _, stderr, status = Open3.capture3(cmd)
-      unless status.success?
-        raise "Problem running #{cmd}, stderr was:\n#{stderr}"
-      end
-    end
-
-    def self.create_dir(path, perm = 0755)
-      FileUtils.mkdir_p path, mode: perm unless Dir.exist? path
-    end
-
-    def self.add_file(path, content = '')
-      File.write path, content unless File.exist? path
-    end
-
-    def self.mkdir(dir)
-      FileUtils.mkdir_p dir unless Dir.exist? dir
+    def self.systemd?
+      Dir.exist? "#{OPTIONS[:mountpoint]}/etc/systemd"
     end
 
-    def self.touch(file)
-      File.write file, '' unless File.exist? file
+    def self.openrc?
+      File.exist? "#{OPTIONS[:mountpoint]}/etc/conf.d/keymaps"
     end
 
-    def self.echo(src, content = '')
-      File.write(src, "#{content}\n", mode: 'w')
+    def self.runit?
+      Dir.exist? "#{OPTIONS[:mountpoint]}/etc/runit"
     end
 
-    def self.echo_a(src, content = '')
-      raise "No file #{src} found !" unless File.exist? src
-
-      File.write(src, "#{content}\n", mode: 'a') unless self.grep?(src, content)
+    def self.grub?
+      File.exist? "#{OPTIONS[:mountpoint]}/etc/default/grub"
     end
 
-    def self.cp(src, dest)
-      raise "Src file #{src} no found" unless File.exist? src
-
-      FileUtils.cp(src, dest)
+    def self.get_file_online(url, dest)
+      URI.open(url) do |l|
+        File.open(dest, "wb") { |f| f.write(l.read) }
+      end
+    rescue Net::OpenTimeout => e
+      abort "DNS error #{e}"
     end
 
-    def self.grep?(file, regex)
-      is_found = false
-      return is_found unless File.exist? file
-      File.open(file) do |f|
-        f.each { |l| is_found = true if l.match(regex) }
+    def self.exec_or_die(cmd)
+      _, stderr, status = Open3.capture3(cmd)
+      unless status.success?
+        abort "Problem running #{cmd}, stderr was:\n#{stderr}"
       end
-      is_found
     end
 
     def self.sys(cmd)
       system(cmd)
-      raise "Error with #{cmd}" unless $?.success?
+      $?.success? || abort("Error with #{cmd}")
     end
 
     def self.partuuid(dev)
@@ -79,39 +57,58 @@ module Getch
           return f.delete_prefix('/dev/disk/by-uuid/')
         end
       end
+      Log.new.fatal("UUID on #{dev} is no found")
     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.'
+    def self.get_dm(name)
+      Dir.glob('/dev/mapper/*').each do |f|
+        if f =~ /#{name}/ && f != '/dev/mapper/control'
+          return File.readlink(f).tr('../', '')
         end
       end
+      Log.new.fatal("Dm for #{name} is no found")
     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?
-          puts "\s[OK]"
-          return
+    # Used by ZFS for the pool creation
+    # sleep is necessary here at least the first time
+    def self.get_id(dev)
+      sleep 3
+      Dir.glob('/dev/disk/by-id/*').each do |f|
+        if File.readlink(f).match(/#{dev}/)
+          return f.delete_prefix('/dev/disk/by-id/')
         end
-        raise "\n[-] Fail cmd #{args} - #{stderr}."
       end
+      Log.new.fatal("ID on #{dev} is no found")
+    end
+
+    # https://wiki.gentoo.org/wiki/Handbook:AMD64/Installation/Base
+    def self.mount_all
+      dest = OPTIONS[:mountpoint]
+      NiTo.mount '--types proc /proc', "#{dest}/proc"
+      ['dev', 'sys', 'run'].each do |d|
+        NiTo.mount '--rbind', "/#{d}", "#{dest}/#{d}"
+        NiTo.mount '--make-rslave', "#{dest}/#{d}"
+      end
+    end
+
+    def self.get_memory
+      mem = nil
+      File.open('/proc/meminfo').each do |l|
+        t = l.split(' ') if l =~ /memtotal/i
+        t && mem = t[1]
+      end
+      mem || Log.new.fatal('get_memory - failed to get memory')
+
+      mem += 'K'
+    end
 
+    # get the sector size of a disk
+    def self.get_bs(path)
+      cmd = Getch::Command.new('blockdev', '--getpbsz', path)
+      cmd.res
+    end
+
+    module Void
       def command_output(args)
         print " => Exec: #{args}..."
         cmd = "chroot #{Getch::MOUNTPOINT} /bin/bash -c \"#{args}\""
@@ -126,19 +123,6 @@ module Getch
         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 do |line|
-          return true if line.match(/#{text}/)
-        end
-        false
-      end
-
       # Used only when need password
       def chroot(cmd)
         unless system('chroot', Getch::MOUNTPOINT, '/bin/bash', '-c', cmd)
@@ -162,42 +146,6 @@ module Getch
 
         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
 end