getch 0.0.8 → 0.1.3

data/lib/getch/filesystem/zfs.rb

@@ -0,0 +1,14 @@
+module Getch
+  module FileSystem
+    module Zfs
+    end
+  end
+end
+
+require_relative 'zfs/device'
+require_relative 'zfs/partition'
+require_relative 'zfs/format'
+require_relative 'zfs/mount'
+require_relative 'zfs/config'
+require_relative 'zfs/deps'
+require_relative 'zfs/encrypt'

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

@@ -0,0 +1,57 @@
+module Getch
+  module FileSystem
+    module Zfs
+      class Config < Getch::FileSystem::Zfs::Device
+        def initialize
+          super
+          gen_uuid
+          @root_dir = MOUNTPOINT
+          @init = '/usr/lib/systemd/systemd'
+        end
+
+        def fstab
+          file = "#{@root_dir}/etc/fstab"
+          datas = data_fstab
+          File.write(file, datas.join("\n"))
+        end
+
+        def systemd_boot
+          return if ! Helpers::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"
+          ]
+          File.write("#{dir}/gentoo.conf", datas_gentoo.join("\n"))
+        end
+
+        # See https://wiki.gentoo.org/wiki/ZFS#ZFS_root
+        def grub
+          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\""
+          ]
+          File.write("#{file}", cmdline.join("\n"), mode: 'a')
+        end
+
+        private
+
+        def gen_uuid
+          @uuid_swap = `lsblk -o "UUID" #{@dev_swap} | tail -1`.chomp()
+          @uuid_esp = `lsblk -o "UUID" #{@dev_esp} | tail -1`.chomp() if @dev_esp
+        end
+
+        def data_fstab
+          efi = @dev_esp ? "UUID=#{@uuid_esp} /efi vfat noauto,noatime 1 2" : ''
+          swap = @dev_swap ? "UUID=#{@uuid_swap} none swap discard 0 0" : ''
+
+          [ efi, swap ]
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,95 @@
+module Getch
+  module FileSystem
+    module Zfs
+      class Deps < Getch::FileSystem::Zfs::Device
+        def make
+          unstable_zfs
+          install_deps
+          zfs_mountpoint
+          auto_module_rebuild
+          hostid
+          options_make
+          Getch::Make.new("genkernel --kernel-config=/usr/src/linux/.config all").run!
+        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
+          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!
+        end
+
+        # See: https://wiki.archlinux.org/index.php/ZFS#Using_zfs-mount-generator
+        def zfs_mountpoint
+          Helpers::mkdir("#{MOUNTPOINT}/etc/zfs/zfs-list.cache")
+          Helpers::touch("#{MOUNTPOINT}/etc/zfs/zfs-list.cache/#{@boot_pool_name}") if @dev_boot
+          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 auto_module_rebuild
+          g_dir="#{MOUNTPOINT}/etc/portage/env/sys-kernel"
+          Helpers::mkdir(g_dir)
+          # See https://wiki.gentoo.org/wiki/Kernel/Upgrade#Automated_build_and_installation
+          content=<<EOF
+post_pkg_postinst() {
+  # BUG: reinstalls of a source will cause errors
+  CURRENT_KV=$(uname -r)
+  # Check to see if genkernel has been run previously for the running kernel and use that config
+  if [[ -f "${EROOT}/etc/kernels/kernel-config-${CURRENT_KV}" ]] ; then
+    genkernel --kernel-config="${EROOT}/etc/kernels/kernel-config-${CURRENT_KV}" all
+  elif [[ -f "${EROOT}/usr/src/linux-${CURRENT_KV}/.config" ]] ; then # Use latest kernel config from current kernel
+    genkernel --kernel-config="${EROOT}/usr/src/linux-${CURRENT_KV}/.config" all
+  else # No valid configs known
+    genkernel all
+  fi
+}
+EOF
+          File.write("#{g_dir}/gentoo-sources", content)
+        end
+
+        def hostid
+          hostid_value=`hostid`.chomp
+          File.write("#{MOUNTPOINT}/etc/hostid", hostid_value, mode: 'w')
+        end
+
+        def options_make
+          bootloader = Helpers::efi? ? 'BOOTLOADER="no"' : 'BOOTLOADER="grub2"'
+          datas = [
+            '',
+            bootloader,
+            'INSTALL="yes"',
+            'MENUCONFIG="no"',
+            'CLEAN="yes"',
+            'SAVE_CONFIG="yes"',
+            'MOUNTBOOT="no"',
+            'MRPROPER="no"',
+            'ZFS="yes"',
+            'MODULEREBUILD="yes"'
+          ]
+          file = "#{MOUNTPOINT}/etc/genkernel.conf"
+          File.write(file, datas.join("\n"), mode: 'a')
+        end
+
+        def exec(cmd)
+          Getch::Chroot.new(cmd).run!
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,58 @@
+module Getch
+  module FileSystem
+    module Zfs
+      class Device < Getch::FileSystem::Device
+        def initialize
+          super
+          @id = Helpers::pool_id(@dev_root)
+          @boot_pool_name = "bpool-#{@id}"
+          @pool_name = "rpool-#{@id}"
+          @zfs_home = @user ? true : false
+        end
+        
+        private
+
+        def search_boot
+          if @efi
+            if @boot_disk
+              @dev_esp = "/dev/#{@boot_disk}1"
+            else
+              @dev_esp = "/dev/#{@disk}1"
+              @root_part += 1
+            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
+  end
+end

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

@@ -0,0 +1,15 @@
+module Getch
+  module FileSystem
+    module Zfs
+      module Encrypt
+      end
+    end
+  end
+end
+
+require_relative 'encrypt/device'
+require_relative 'encrypt/partition'
+require_relative 'encrypt/format'
+require_relative 'encrypt/mount'
+require_relative 'encrypt/config'
+require_relative 'encrypt/deps'

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

@@ -0,0 +1,67 @@
+module Getch
+  module FileSystem
+    module Zfs
+      module Encrypt
+        class Config < Getch::FileSystem::Zfs::Encrypt::Device
+          def initialize
+            super
+            gen_uuid
+            @root_dir = MOUNTPOINT
+            @init = '/usr/lib/systemd/systemd'
+            crypttab
+          end
+
+          def fstab
+            file = "#{@root_dir}/etc/fstab"
+            datas = data_fstab
+            File.write(file, datas.join("\n"))
+          end
+
+          def systemd_boot
+            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]}"
+            ]
+            File.write("#{dir}/gentoo.conf", datas_gentoo.join("\n"))
+          end
+
+          def crypttab
+            datas = [
+              "cryptswap PARTUUID=#{@partuuid_swap} /dev/urandom swap,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 @efi
+            file = "#{@root_dir}/etc/default/grub"
+            cmdline = [ 
+              "GRUB_CMDLINE_LINUX=\"root=ZFS=#{@pool_name}/ROOT/gentoo init=#{@init} dozfs keymap=#{DEFAULT_OPTIONS[:keymap]}\""
+            ]
+            File.write("#{file}", cmdline.join("\n"), mode: 'a')
+          end
+
+          private
+
+          def gen_uuid
+            @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_esp ? "UUID=#{@uuid_esp} /efi vfat noauto,noatime 1 2" : ''
+            swap = @dev_swap ? "/dev/mapper/cryptswap none swap discard 0 0" : ''
+
+            [ boot_efi, swap ]
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,97 @@
+module Getch
+  module FileSystem
+    module Zfs
+      module Encrypt
+        class Deps < Getch::FileSystem::Zfs::Encrypt::Device
+          def make
+            unstable_zfs
+            install_deps
+            zfs_mountpoint
+            auto_module_rebuild
+            hostid
+            options_make
+            Getch::Make.new("genkernel --kernel-config=/usr/src/linux/.config all").run!
+          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
+            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!
+          end
+
+          # See: https://wiki.archlinux.org/index.php/ZFS#Using_zfs-mount-generator
+          def zfs_mountpoint
+            Helpers::mkdir("#{MOUNTPOINT}/etc/zfs/zfs-list.cache")
+            Helpers::touch("#{MOUNTPOINT}/etc/zfs/zfs-list.cache/#{@boot_pool_name}") if @dev_boot
+            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 auto_module_rebuild
+            g_dir="#{MOUNTPOINT}/etc/portage/env/sys-kernel"
+            Helpers::mkdir(g_dir)
+            # See https://wiki.gentoo.org/wiki/Kernel/Upgrade#Automated_build_and_installation
+            content=<<EOF
+post_pkg_postinst() {
+  # BUG: reinstalls of a source will cause errors
+  CURRENT_KV=$(uname -r)
+  # Check to see if genkernel has been run previously for the running kernel and use that config
+  if [[ -f "${EROOT}/etc/kernels/kernel-config-${CURRENT_KV}" ]] ; then
+    genkernel --kernel-config="${EROOT}/etc/kernels/kernel-config-${CURRENT_KV}" all
+  elif [[ -f "${EROOT}/usr/src/linux-${CURRENT_KV}/.config" ]] ; then # Use latest kernel config from current kernel
+    genkernel --kernel-config="${EROOT}/usr/src/linux-${CURRENT_KV}/.config" all
+  else # No valid configs known
+    genkernel all
+  fi
+}
+EOF
+            File.write("#{g_dir}/gentoo-sources", content)
+          end
+
+          def hostid
+            hostid_value=`hostid`.chomp
+            File.write("#{MOUNTPOINT}/etc/hostid", hostid_value, mode: 'w')
+          end
+
+          def options_make
+            bootloader = Helpers::efi? ? 'BOOTLOADER="no"' : 'BOOTLOADER="grub2"'
+            datas = [
+              '',
+              bootloader,
+              'INSTALL="yes"',
+              'MENUCONFIG="no"',
+              'CLEAN="yes"',
+              'KEYMAP="yes"',
+              'SAVE_CONFIG="yes"',
+              'MOUNTBOOT="no"',
+              'MRPROPER="no"',
+              'ZFS="yes"',
+              'MODULEREBUILD="yes"'
+            ]
+            file = "#{MOUNTPOINT}/etc/genkernel.conf"
+            File.write(file, datas.join("\n"), mode: 'a')
+          end
+
+          def exec(cmd)
+            Getch::Chroot.new(cmd).run!
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,60 @@
+module Getch
+  module FileSystem
+    module Zfs
+      module Encrypt
+        class Device < Getch::FileSystem::Device
+          def initialize
+            super
+            @id = Helpers::pool_id(@dev_root)
+            @boot_pool_name = "bpool-#{@id}"
+            @pool_name = "rpool-#{@id}"
+            @zfs_home = @user ? true : false
+          end
+
+          private
+
+          def search_boot
+            if @efi
+              if @boot_disk
+                @dev_esp  = "/dev/#{@boot_disk}1"
+              else
+                @dev_esp  = "/dev/#{@disk}1"
+                @root_part += 1
+              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
+    end
+  end
+end