getch 0.0.5 → 0.1.0

data/lib/getch/filesystem/ext4/format.rb

@@ -0,0 +1,30 @@
+module Getch
+  module FileSystem
+    module Ext4
+      class Format < Getch::FileSystem::Ext4::Device
+        def initialize
+          super
+          @fs = 'ext4'
+          @state = Getch::States.new()
+          format
+        end
+
+        def format
+          return if STATES[:format]
+          puts "Format #{@disk} with #{@fs}"
+          exec("mkfs.fat -F32 #{@dev_boot_efi}") if Helpers::efi?
+          exec("mkswap -f #{@dev_swap}")
+          exec("mkfs.#{@fs} -F #{@dev_root}")
+          exec("mkfs.#{@fs} -F #{@dev_home}") if @dev_home
+          @state.format
+        end
+
+        private
+
+        def exec(cmd)
+          Getch::Command.new(cmd).run!
+        end
+      end
+    end
+  end
+end

data/lib/getch/filesystem/ext4/mount.rb

@@ -0,0 +1,62 @@
+require 'fileutils'
+
+module Getch
+  module FileSystem
+    module Ext4
+      class Mount < Getch::FileSystem::Ext4::Device
+        def initialize
+          super
+          @root_dir = MOUNTPOINT
+          @boot_dir = "#{@root_dir}/boot"
+          @boot_efi_dir = "#{@root_dir}/boot/efi"
+          @home_dir = @user ? "#{@root_dir}/home/#{@user}" : nil
+          @state = Getch::States.new()
+        end
+
+        def run
+          return if STATES[:mount]
+          mount_swap
+          mount_root
+          mount_boot
+          mount_home
+          mount_boot_efi
+          @state.mount
+        end
+
+        private
+
+        def mount_swap
+          return if ! @dev_swap
+          system("swapon #{@dev_swap}")
+        end
+
+        def mount_root
+          return if ! @dev_root
+          Dir.mkdir(@root_dir, 0700) if ! Dir.exist?(@root_dir)
+          system("mount #{@dev_root} #{@root_dir}")
+        end
+
+        def mount_boot_efi
+          return if ! @dev_boot_efi
+          FileUtils.mkdir_p @boot_efi_dir, mode: 0700 if ! Dir.exist?(@boot_efi_dir)
+          system("mount #{@dev_boot_efi} #{@boot_efi_dir}")
+        end
+
+        def mount_boot
+          return if ! @dev_boot
+          FileUtils.mkdir_p @boot_dir, mode: 0700 if ! Dir.exist?(@boot_dir)
+          system("mount #{@dev_boot} #{@boot_dir}")
+        end
+
+        def mount_home
+          return if ! @dev_home
+          if @user != nil then
+            FileUtils.mkdir_p @home_dir, mode: 0700 if ! Dir.exist?(@home_dir)
+            system("mount #{@dev_home} #{@home_dir}")
+          end
+          @state.mount
+        end
+      end
+    end
+  end
+end

data/lib/getch/filesystem/ext4/partition.rb

@@ -0,0 +1,75 @@
+module Getch
+  module FileSystem
+    module Ext4
+      class Partition < Getch::FileSystem::Ext4::Device
+        def initialize
+          super
+          @state = Getch::States.new()
+          run_partition
+        end
+
+        def run_partition
+          return if STATES[:partition ]
+          clear_struct
+          cleaning
+          if Helpers::efi?
+            partition_efi
+          else
+            partition_bios
+          end
+          @state.partition
+        end
+
+        private
+
+        def clear_struct
+          exec("sgdisk -Z /dev/#{@disk}")
+          exec("wipefs -a /dev/#{@disk}")
+        end
+
+        def cleaning
+          puts
+          print "Cleaning data on #{@disk}, can be long, avoid this on Flash Memory (SSD,USB,...) ? (n,y) "
+          case gets.chomp
+          when /^y|^Y/
+            bloc=`blockdev --getbsz /dev/#{@disk}`.chomp
+            exec("dd if=/dev/urandom of=/dev/#{@disk} bs=#{bloc} status=progress")
+          else
+            return
+          end
+        end
+
+        # Follow https://wiki.archlinux.org/index.php/Partitioning
+        def partition_efi
+          # /boot/efi - EFI system partition - 260MB
+          # /         - Root
+          # swap      - Linux Swap - size of the ram
+          # /home     - Home
+          mem=`awk '/MemTotal/ {print $2}' /proc/meminfo`.chomp + 'K'
+
+          exec("sgdisk -n1:1M:+260M -t1:EF00 /dev/#{@disk}")
+          exec("sgdisk -n2:0:+15G -t2:8304 /dev/#{@disk}")
+          exec("sgdisk -n3:0:+#{mem} -t3:8200 /dev/#{@disk}")
+          exec("sgdisk -n4:0:0 -t4:8302 /dev/#{@disk}") if @dev_home
+        end
+
+        def partition_bios
+          # None      - Bios Boot Partition - 1MiB
+          # /         - Root
+          # swap      - Linux Swap - size of the ram
+          # /home     - Home
+          mem=`awk '/MemTotal/ {print $2}' /proc/meminfo`.chomp + 'K'
+
+          exec("sgdisk -n1:1MiB:+1MiB -t1:EF02 /dev/#{@disk}")
+          exec("sgdisk -n2:0:+15G -t2:8304 /dev/#{@disk}")
+          exec("sgdisk -n3:0:+#{mem} -t3:8200 /dev/#{@disk}")
+          exec("sgdisk -n4:0:0 -t4:8302 /dev/#{@disk}") if @dev_home
+        end
+
+        def exec(cmd)
+          Getch::Command.new(cmd).run!
+        end
+      end
+    end
+  end
+end

data/lib/getch/filesystem/lvm.rb

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

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

@@ -0,0 +1,63 @@
+module Getch
+  module FileSystem
+    module Lvm
+      class Config < Getch::FileSystem::Lvm::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 = '/boot/efi'
+          dir = "#{@root_dir}/#{esp}/loader/entries/"
+          datas_gentoo = [
+            'title Gentoo Linux',
+            'linux /vmlinuz',
+            'initrd /initramfs',
+            "options resume=UUID=#{@uuid_swap} root=UUID=#{@uuid_root} init=#{@init} dolvm rw"
+          ]
+          File.write("#{dir}/gentoo.conf", datas_gentoo.join("\n"))
+        end
+
+        def grub
+          return if Helpers::efi?
+          file = "#{@root_dir}/etc/default/grub"
+          cmdline = [ 
+            "GRUB_CMDLINE_LINUX=\"resume=UUID=#{@uuid_swap} root=UUID=#{@uuid_root} init=#{@init} dolvm rw\""
+          ]
+          File.write("#{file}", cmdline.join("\n"), mode: 'a')
+        end
+
+        private
+
+        def gen_uuid
+          @uuid_swap = `lsblk -o "UUID" #{@lv_swap} | tail -1`.chomp() if @lv_swap
+          @uuid_root = `lsblk -o "UUID" #{@lv_root} | tail -1`.chomp() if @lv_root
+          @uuid_dev_root = `lsblk -o "UUID" #{@dev_root} | tail -1`.chomp() if @dev_root
+          @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_home = `lsblk -o "UUID" #{@lv_home} | tail -1`.chomp() if @lv_home
+        end
+
+        def data_fstab
+          boot_efi = @dev_boot_efi ? "UUID=#{@uuid_boot_efi} /boot/efi vfat noauto,noatime 1 2" : ''
+          boot = @dev_boot ? "UUID=#{@uuid_boot} /boot ext4 noauto,noatime 1 2" : ''
+          swap = @lv_swap ? "UUID=#{@uuid_swap} none swap discard 0 0" : ''
+          root = @lv_root ? "UUID=#{@uuid_root} / ext4 defaults 0 1" : ''
+          home = @lv_home ? "UUID=#{@uuid_home} /home/#{@user} ext4 defaults 0 2" : ''
+
+          [ boot_efi, boot, swap, root, home ]
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,57 @@
+module Getch
+  module FileSystem
+    module Lvm
+      class Deps
+        def initialize
+          if Helpers::efi?
+            install_efi
+          else
+            install_bios
+          end
+          install_deps
+        end
+
+        def make
+          options_make
+          Getch::Make.new("genkernel --kernel-config=/usr/src/linux/.config all").run!
+        end
+
+        private
+        def options_make
+          grub = Helpers::efi? ? 'BOOTLOADER="no"' : 'BOOTLOADER="grub2"'
+          datas = [
+            '',
+            grub,
+            'INSTALL="yes"',
+            'MENUCONFIG="no"',
+            'CLEAN="yes"',
+            'SAVE_CONFIG="yes"',
+            'MOUNTBOOT="yes"',
+            'MRPROPER="no"',
+            'LVM="yes"',
+          ]
+          file = "#{MOUNTPOINT}/etc/genkernel.conf"
+          File.write(file, datas.join("\n"), mode: 'a')
+        end
+
+        def install_efi
+        end
+
+        def install_bios
+          exec("euse -p sys-boot/grub -E device-mapper")
+        end
+
+        def install_deps
+          exec("euse -E lvm")
+          Getch::Emerge.new('genkernel lvm2').pkg!
+          Getch::Garden.new('-a lvm').run!
+          exec("systemctl enable lvm2-monitor")
+        end
+
+        def exec(cmd)
+          Getch::Chroot.new(cmd).run!
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,19 @@
+module Getch
+  module FileSystem
+    module Lvm
+      class 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_root = Helpers::efi? ? "/dev/#{@disk}2" : "/dev/#{@disk}3"
+          @vg = 'vg0'
+          @lv_root = "/dev/mapper/#{@vg}-root"
+          @lv_swap = "/dev/mapper/#{@vg}-swap"
+          @lv_home = @user ? "/dev/mapper/#{@vg}-home" : nil
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,15 @@
+module Getch
+  module FileSystem
+    module Lvm
+      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/lvm/encrypt/config.rb

@@ -0,0 +1,74 @@
+module Getch
+  module FileSystem
+    module Lvm
+      module Encrypt
+        class Config < Getch::FileSystem::Lvm::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 ! Helpers::efi? 
+            esp = '/boot/efi'
+            dir = "#{@root_dir}/#{esp}/loader/entries/"
+            datas_gentoo = [
+              'title Gentoo Linux',
+              'linux /vmlinuz',
+              'initrd /initramfs',
+              "options crypt_root=UUID=#{@uuid_dev_root} root=#{@lv_root} init=#{@init} keymap=#{DEFAULT_OPTIONS[:keymap]} dolvm rw"
+            ]
+            File.write("#{dir}/gentoo.conf", datas_gentoo.join("\n"))
+          end
+
+          def crypttab
+            datas = [
+              "cryptswap #{@lv_swap} /dev/urandom swap,cipher=aes-xts-plain64:sha256,size=256"
+            ]
+            File.write("#{@root_dir}/etc/crypttab", datas.join("\n"))
+          end
+
+          def grub
+            return if Helpers::efi?
+            file = "#{@root_dir}/etc/default/grub"
+            cmdline = [ 
+              "GRUB_CMDLINE_LINUX=\"crypt_root=UUID=#{@uuid_dev_root} root=#{@lv_root} init=#{@init} dolvm rw slub_debug=P page_poison=1 slab_nomerge pti=on vsyscall=none spectre_v2=on spec_store_bypass_disable=seccomp iommu=force keymap=#{DEFAULT_OPTIONS[:keymap]}\"",
+              "GRUB_ENABLE_CRYPTODISK=y"
+            ]
+            File.write("#{file}", cmdline.join("\n"), mode: 'a')
+          end
+
+          private
+
+          def gen_uuid
+            @uuid_swap = `lsblk -o "UUID" #{@lv_swap} | tail -1`.chomp() if @lv_swap
+            @uuid_root = `lsblk -d -o "UUID" #{@lv_root} | tail -1`.chomp() if @lv_root
+            @uuid_dev_root = `lsblk -d -o "UUID" #{@dev_root} | tail -1`.chomp() if @dev_root
+            @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_home = `lsblk -o "UUID" #{@lv_home} | tail -1`.chomp() if @lv_home
+          end
+
+          def data_fstab
+            boot_efi = @dev_boot_efi ? "UUID=#{@uuid_boot_efi} /boot/efi vfat noauto,noatime 1 2" : ''
+            boot = @dev_boot ? "UUID=#{@uuid_boot} /boot ext4 noauto,noatime 1 2" : ''
+            swap = @lv_swap ? "/dev/mapper/cryptswap none swap discard 0 0" : ''
+            root = @lv_root ? "UUID=#{@uuid_root} / ext4 defaults 0 1" : ''
+            home = @lv_home ? "UUID=#{@uuid_home} /home/#{@user} ext4 defaults 0 2" : ''
+
+            [ boot_efi, boot, swap, root, home ]
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,63 @@
+module Getch
+  module FileSystem
+    module Lvm
+      module Encrypt
+        class Deps
+          def initialize
+            if Helpers::efi?
+              install_efi
+            else
+              install_bios
+            end
+            install_deps
+          end
+
+          def make
+            options_make
+            Getch::Make.new("genkernel --kernel-config=/usr/src/linux/.config all").run!
+          end
+
+          private
+          def options_make
+            grub = Helpers::efi? ? 'BOOTLOADER="no"' : 'BOOTLOADER="grub2"'
+            datas = [
+              '',
+              grub,
+              'INSTALL="yes"',
+              'MENUCONFIG="no"',
+              'CLEAN="yes"',
+              'KEYMAP="yes"',
+              'SAVE_CONFIG="yes"',
+              'MOUNTBOOT="yes"',
+              'MRPROPER="no"',
+              'LVM="yes"',
+              'LUKS="yes"',
+            ]
+            file = "#{MOUNTPOINT}/etc/genkernel.conf"
+            File.write(file, datas.join("\n"), mode: 'a')
+          end
+
+          def install_efi
+          end
+
+          def install_bios
+            exec("euse -p sys-boot/grub -E device-mapper")
+          end
+
+          def install_deps
+            make_conf = "#{MOUNTPOINT}/etc/portage/make.conf"
+            exec("euse -E lvm") if ! Helpers::grep?(make_conf, /lvm/)
+            exec("euse -E cryptsetup") if ! Helpers::grep?(make_conf, /cryptsetup/)
+            Getch::Emerge.new('genkernel systemd sys-fs/cryptsetup lvm2').pkg!
+            Getch::Garden.new('-a lvm').run!
+            exec("systemctl enable lvm2-monitor")
+          end
+
+          def exec(cmd)
+            Getch::Chroot.new(cmd).run!
+          end
+        end
+      end
+    end
+  end
+end