getch 0.1.0 → 0.1.6

data/lib/getch/filesystem/lvm/void.rb

@@ -0,0 +1,45 @@
+require_relative '../../helpers'
+
+module Getch
+  module FileSystem
+    module Lvm
+      class Void < Device
+        include Helpers::Void
+        attr_reader :boot_disk
+
+        def fstab
+          conf = "#{MOUNTPOINT}/etc/fstab"
+          File.write(conf, "\n", mode: 'w', chmod: 0644)
+          line_fstab(@dev_esp, "/efi vfat noauto,rw,relatime 0 0") if @dev_esp
+          line_fstab(@dev_boot, "/boot ext4 noauto,rw,relatime 0 0") if @dev_boot
+          add_line(conf, "#{@lv_swap} swap swap rw,noatime,discard 0 0") if @lv_swap
+          add_line(conf, "#{@lv_home} /home/#{@user} ext4 rw,noatime,discard 0 2") if @lv_home
+          add_line(conf, "#{@lv_root} / ext4 rw,relatime 0 1")
+          add_line(conf, "tmpfs /tmp tmpfs defaults,nosuid,nodev 0 0")
+        end
+
+        def config_dracut
+          conf = "#{MOUNTPOINT}/etc/dracut.conf.d/lvm.conf"
+          # dracut: value+= should be surrounding by white space
+          content = [
+            "hostonly=\"yes\"",
+            "omit_dracutmodules+=\" btrfs \"",
+            ""
+          ]
+          File.write(conf, content.join("\n"), mode: 'w', chmod: 0644)
+        end
+
+        def kernel_cmdline_dracut
+          conf = "#{MOUNTPOINT}/etc/dracut.conf.d/cmdline.conf"
+          c="kernel_cmdline=\"rd.lvm.vg=#{@vg} rootflags=rw,relatime\""
+          File.write(conf, "#{c}\n", mode: 'w', chmod: 0644)
+        end
+
+        def finish
+          puts "+ Enter in your system: chroot /mnt /bin/bash"
+          puts "+ Reboot with: shutdown -r now"
+        end
+      end
+    end
+  end
+end

data/lib/getch/filesystem/mount.rb

@@ -0,0 +1,56 @@
+require 'fileutils'
+
+module Getch
+  module FileSystem
+    class Mount
+      def initialize
+        @root_dir = MOUNTPOINT
+        @boot_dir = "#{@root_dir}/boot"
+        @boot_efi_dir = "#{@root_dir}/efi"
+        @home_dir = "#{@root_dir}/home"
+        @state = Getch::States.new()
+        @log = Getch::Log.new
+      end
+
+      def swap(dev)
+        return if ! dev
+        if Helpers::grep?('/proc/swaps', /^\/dev/)
+          exec("swapoff #{dev}")
+        end
+
+        exec("swapon #{dev}")
+      end
+
+      def root(dev)
+        return if ! dev
+        Helpers::mkdir(@root_dir)
+        exec("mount #{dev} #{@root_dir}")
+      end
+
+      def esp(dev)
+        return if ! dev
+        Helpers::mkdir(@boot_efi_dir)
+        exec("mount #{dev} #{@boot_efi_dir}")
+      end
+
+      def boot(dev)
+        return if ! dev
+        Helpers::mkdir(@boot_dir)
+        exec("mount #{dev} #{@boot_dir}")
+      end
+
+      def home(dev)
+        return if ! dev
+        Helpers::mkdir(@home_dir)
+        exec("mount #{dev} #{@home_dir}")
+      end
+
+      private
+
+      def exec(cmd)
+        @log.info("==> #{cmd}")
+        Helpers::sys(cmd)
+      end
+    end
+  end
+end

data/lib/getch/filesystem/partition.rb

@@ -0,0 +1,77 @@
+module Getch
+  module FileSystem
+    class Partition
+      def initialize
+        @log = Getch::Log.new
+      end
+
+      def gpt(dev)
+        return if ! dev
+        disk = disk_name(dev)
+        part = dev.match(/[0-9]/)
+        exec("sgdisk -n#{part}:1MiB:+1MiB -t#{part}:EF02 #{disk}")
+      end
+
+      def boot(dev)
+        return if ! dev
+        disk = disk_name(dev)
+        part = dev.match(/[0-9]/)
+        if Getch::OPTIONS[:fs] == "zfs"
+          exec("sgdisk -n#{part}:0:+2G -t#{part}:BE00 #{disk}")
+        else
+          exec("sgdisk -n#{part}:0:+128MiB -t#{part}:8300 #{disk}")
+        end
+      end
+
+      def efi(dev)
+        return if ! dev
+        disk = disk_name(dev)
+        part = dev.match(/[0-9]/)
+        exec("sgdisk -n#{part}:1M:+260M -t#{part}:EF00 #{disk}")
+      end
+
+      def swap(dev)
+        return if ! dev
+        disk = disk_name(dev)
+        part = dev.match(/[0-9]/)
+        if Getch::OPTIONS[:cache_disk]
+          exec("sgdisk -n#{part}:0:0 -t#{part}:8200 #{disk}")
+        else
+          mem=`awk '/MemTotal/ {print $2}' /proc/meminfo`.chomp + 'K'
+          exec("sgdisk -n#{part}:0:+#{mem} -t#{part}:8200 #{disk}")
+        end
+      end
+
+      def root(dev, code)
+        return if ! dev
+        disk = disk_name(dev)
+        part = dev.match(/[0-9]/)
+        exec("sgdisk -n#{part}:0:0 -t#{part}:#{code} #{disk}")
+      end
+
+      def home(dev, code)
+        return if ! dev
+        disk = disk_name(dev)
+        part = dev.match(/[0-9]/)
+        if Getch::OPTIONS[:home_disk]
+          exec("sgdisk -n#{part}:0:0 -t#{part}:#{code} #{disk}")
+        end
+      end
+
+      private
+
+      def disk_name(dev)
+        dev.match(/[^0-9]+/)
+      end
+
+      def exec(cmd)
+        @log.debug "Partition disk with #{cmd}"
+        if Getch::OPTIONS[:encrypt]
+          Helpers::sys(cmd)
+        else
+          Getch::Command.new(cmd).run!
+        end
+      end
+    end
+  end
+end

data/lib/getch/filesystem/zfs.rb

@@ -11,4 +11,5 @@ require_relative 'zfs/format'
 require_relative 'zfs/mount'
 require_relative 'zfs/config'
 require_relative 'zfs/deps'
+require_relative 'zfs/void'
 require_relative 'zfs/encrypt'

data/lib/getch/filesystem/zfs/config.rb

@@ -1,7 +1,7 @@
 module Getch
   module FileSystem
     module Zfs
-      class Config < Getch::FileSystem::Zfs::Device
+      class Config < Device
         def initialize
           super
           gen_uuid
@@ -17,13 +17,13 @@ module Getch
 
         def systemd_boot
           return if ! Helpers::efi? 
-          esp = '/boot/efi'
+          esp = '/efi'
           dir = "#{@root_dir}/#{esp}/loader/entries/"
           datas_gentoo = [
             'title Gentoo Linux',
             'linux /vmlinuz',
             'initrd /initramfs',
-            "options resume=UUID=#{@uuid_swap} root=ZFS=#{@pool_name}/ROOT/gentoo init=#{@init} dozfs"
+            "options resume=UUID=#{@uuid_swap} root=ZFS=#{@pool_name}/ROOT/#{@n} init=#{@init} dozfs"
           ]
           File.write("#{dir}/gentoo.conf", datas_gentoo.join("\n"))
         end
@@ -33,7 +33,7 @@ module Getch
           return if Helpers::efi?
           file = "#{@root_dir}/etc/default/grub"
           cmdline = [ 
-            "GRUB_CMDLINE_LINUX=\"resume=UUID=#{@uuid_swap} root=ZFS=#{@pool_name}/ROOT/gentoo init=#{@init} dozfs\""
+            "GRUB_CMDLINE_LINUX=\"resume=UUID=#{@uuid_swap} root=ZFS=#{@pool_name}/ROOT/#{@n} init=#{@init} dozfs\""
           ]
           File.write("#{file}", cmdline.join("\n"), mode: 'a')
         end
@@ -42,15 +42,14 @@ module Getch
 
         def gen_uuid
           @uuid_swap = `lsblk -o "UUID" #{@dev_swap} | tail -1`.chomp()
-          @uuid_boot = `lsblk -o "UUID" #{@dev_boot} | tail -1`.chomp() if @dev_boot
-          @uuid_boot_efi = `lsblk -o "UUID" #{@dev_boot_efi} | tail -1`.chomp() if @dev_boot_efi
+          @uuid_esp = `lsblk -o "UUID" #{@dev_esp} | tail -1`.chomp() if @dev_esp
         end
 
         def data_fstab
-          boot_efi = @dev_boot_efi ? "UUID=#{@uuid_boot_efi} /boot/efi vfat noauto,noatime 1 2" : ''
+          efi = @dev_esp ? "UUID=#{@uuid_esp} /efi vfat noauto,noatime 1 2" : ''
           swap = @dev_swap ? "UUID=#{@uuid_swap} none swap discard 0 0" : ''
 
-          [ boot_efi, swap ]
+          [ efi, swap ]
         end
       end
     end

data/lib/getch/filesystem/zfs/deps.rb

@@ -1,25 +1,31 @@
 module Getch
   module FileSystem
     module Zfs
-      class Deps < Getch::FileSystem::Zfs::Device
-        def initialize
-          super
+      class Deps < Device
+        def make
+          unstable_zfs
           install_deps
           zfs_mountpoint
           auto_module_rebuild
-        end
-
-        def make
           hostid
           options_make
           Getch::Make.new("genkernel --kernel-config=/usr/src/linux/.config all").run!
+          zed_update_path
         end
 
         private
+
+        def unstable_zfs
+          conf = "#{MOUNTPOINT}/etc/portage/package.accept_keywords/zfs"
+          data = [
+            "sys-fs/zfs-kmod",
+            "sys-fs/zfs"
+          ]
+          File.write(conf, data.join("\n"), mode: "w")
+        end
+
         def install_deps
-          exec("euse -E libzfs") if ! Helpers::grep?("#{MOUNTPOINT}/etc/portage/make.conf", /libzfs/)
-          exec("euse -E rootfs") if ! Helpers::grep?("#{MOUNTPOINT}/etc/portage/make.conf", /rootfs/)
-          Getch::Garden.new('-a zfs').run!
+          Getch::Bask.new('-a zfs').run!
           Getch::Make.new("make modules_prepare").run!
           Getch::Make.new("make -j$(nproc)").run!
           Getch::Emerge.new('genkernel sys-fs/zfs').pkg!
@@ -32,11 +38,18 @@ module Getch
           Helpers::touch("#{MOUNTPOINT}/etc/zfs/zfs-list.cache/#{@pool_name}")
           exec("ln -fs /usr/libexec/zfs/zed.d/history_event-zfs-list-cacher.sh /etc/zfs/zed.d/")
           exec("systemctl start zfs-zed.service")
-          Helpers::sys("sed -Ei \"s|/mnt/?|/|\" #{MOUNTPOINT}/etc/zfs/zfs-list.cache/*")
           exec("systemctl enable zfs-zed.service")
           exec("systemctl enable zfs.target")
         end
 
+        def zed_update_path
+          Dir.glob("#{MOUNTPOINT}/etc/zfs/zfs-list.cache/*").each { |f|
+            if !system("sed", "-Ei", "s|#{MOUNTPOINT}/?|/|", f)
+              raise "system exec sed"
+            end
+          }
+        end
+
         def auto_module_rebuild
           g_dir="#{MOUNTPOINT}/etc/portage/env/sys-kernel"
           Helpers::mkdir(g_dir)
@@ -59,8 +72,7 @@ EOF
         end
 
         def hostid
-          hostid_value=`hostid`.chomp
-          File.write("#{MOUNTPOINT}/etc/hostid", hostid_value, mode: 'w')
+          exec "zgenhostid $(hostid)"
         end
 
         def options_make

data/lib/getch/filesystem/zfs/device.rb

@@ -1,17 +1,62 @@
 module Getch
   module FileSystem
     module Zfs
-      class Device
+      class Device < Getch::FileSystem::Device
         def initialize
-          @disk = DEFAULT_OPTIONS[:disk]
-          @user = DEFAULT_OPTIONS[:username]
-          @dev_boot_efi = Helpers::efi? ? "/dev/#{@disk}1" : nil
-          @dev_boot = Helpers::efi? ? nil : "/dev/#{@disk}2"
-          @dev_swap = Helpers::efi? ? "/dev/#{@disk}2" : "/dev/#{@disk}3"
-          @dev_root = Helpers::efi? ? "/dev/#{@disk}3" : "/dev/#{@disk}4"
-          @boot_pool_name = 'bpool'
-          @pool_name = 'zpool'
+          super
+          @id = Helpers::pool_id(@dev_root)
+          @boot_pool_name = "bpool-#{@id}"
+          @pool_name = "rpool-#{@id}"
           @zfs_home = @user ? true : false
+          @n = Getch::OPTIONS[:os]
+        end
+        
+        private
+
+        def search_boot
+          if @efi
+            if @boot_disk
+              @dev_esp = "/dev/#{@boot_disk}1"
+              @dev_boot = "/dev/#{@boot_disk}2" if Getch::OPTIONS[:os] == 'void'
+            else
+              @dev_esp = "/dev/#{@disk}1"
+              @root_part += 1
+              if Getch::OPTIONS[:os] == 'void'
+                @dev_boot = "/dev/#{@disk}#{@root_part}"
+                @root_part += 1
+              end
+            end
+          else
+            if @boot_disk
+              @dev_gpt = "/dev/#{@boot_disk}1"
+              @dev_boot = "/dev/#{@boot_disk}2"
+              @dev_grub = "/dev/#{@boot_disk}"
+            else
+              @dev_gpt = "/dev/#{@disk}1"
+              @dev_boot = "/dev/#{@disk}2"
+              @dev_grub = "/dev/#{@disk}"
+              @root_part += 2
+            end
+          end
+        end
+
+        def search_swap
+          if @cache_disk
+            @dev_swap = "/dev/#{@cache_disk}1"
+            @dev_log = "/dev/#{@cache_disk}2"
+            @dev_cache = "/dev/#{@cache_disk}3"
+          else
+            @dev_swap = "/dev/#{@disk}#{@root_part}"
+            @root_part += 1
+          end
+        end
+
+        def search_root
+          if @root_part == 1
+            @dev_root = "/dev/#{@disk}"
+          else
+            @dev_root = "/dev/#{@disk}#{@root_part}"
+          end
         end
       end
     end

data/lib/getch/filesystem/zfs/encrypt.rb

@@ -13,3 +13,4 @@ require_relative 'encrypt/format'
 require_relative 'encrypt/mount'
 require_relative 'encrypt/config'
 require_relative 'encrypt/deps'
+require_relative 'encrypt/void'

data/lib/getch/filesystem/zfs/encrypt/config.rb

@@ -2,7 +2,7 @@ module Getch
   module FileSystem
     module Zfs
       module Encrypt
-        class Config < Getch::FileSystem::Zfs::Encrypt::Device
+        class Config < Device
           def initialize
             super
             gen_uuid
@@ -18,31 +18,31 @@ module Getch
           end
 
           def systemd_boot
-            return if ! Helpers::efi? 
-            esp = '/boot/efi'
+            return if ! @efi
+            esp = '/efi'
             dir = "#{@root_dir}/#{esp}/loader/entries/"
             datas_gentoo = [
               'title Gentoo Linux',
               'linux /vmlinuz',
               'initrd /initramfs',
-              "options root=ZFS=#{@pool_name}/ROOT/gentoo init=#{@init} dozfs keymap=#{DEFAULT_OPTIONS[:keymap]}"
+              "options root=ZFS=#{@pool_name}/ROOT/#{@n} init=#{@init} dozfs keymap=#{Getch::OPTIONS[:keymap]}"
             ]
             File.write("#{dir}/gentoo.conf", datas_gentoo.join("\n"))
           end
 
           def crypttab
             datas = [
-              "cryptswap UUID=#{@uuid_swap} /dev/urandom swap,cipher=aes-xts-plain64:sha256,size=256"
+              "cryptswap PARTUUID=#{@partuuid_swap} /dev/urandom swap,discard,cipher=aes-xts-plain64:sha256,size=512"
             ]
             File.write("#{@root_dir}/etc/crypttab", datas.join("\n"))
           end
 
           # See https://wiki.gentoo.org/wiki/ZFS#ZFS_root
           def grub
-            return if Helpers::efi?
+            return if @efi
             file = "#{@root_dir}/etc/default/grub"
             cmdline = [ 
-              "GRUB_CMDLINE_LINUX=\"root=ZFS=#{@pool_name}/ROOT/gentoo init=#{@init} dozfs keymap=#{DEFAULT_OPTIONS[:keymap]}\""
+              "GRUB_CMDLINE_LINUX=\"root=ZFS=#{@pool_name}/ROOT/#{@n} init=#{@init} dozfs keymap=#{Getch::OPTIONS[:keymap]}\""
             ]
             File.write("#{file}", cmdline.join("\n"), mode: 'a')
           end
@@ -50,13 +50,13 @@ module Getch
           private
 
           def gen_uuid
-            @uuid_swap = `lsblk -o "UUID" #{@dev_swap} | tail -1`.chomp()
-            @uuid_boot_efi = `lsblk -o "UUID" #{@dev_boot_efi} | tail -1`.chomp() if @dev_boot_efi
+            @partuuid_swap = Helpers::partuuid(@dev_swap)
+            @uuid_esp = `lsblk -o "UUID" #{@dev_esp} | tail -1`.chomp() if @dev_esp
           end
 
           def data_fstab
-            boot_efi = @dev_boot_efi ? "UUID=#{@uuid_boot_efi} /boot/efi vfat noauto,noatime 1 2" : ''
-            swap = @dev_swap ? "/dev/mapper/cryptswap none swap discard 0 0" : ''
+            boot_efi = @dev_esp ? "UUID=#{@uuid_esp} /efi vfat noauto,noatime 1 2" : ''
+            swap = @dev_swap ? "/dev/mapper/cryptswap none swap sw 0 0" : ''
 
             [ boot_efi, swap ]
           end