getch 0.0.4 → 0.0.9

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

@@ -0,0 +1,22 @@
+module Getch
+  module FileSystem
+    module Lvm
+      module Encrypt
+        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
+            @luks_root = "/dev/mapper/cryptroot"
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,32 @@
+module Getch
+  module FileSystem
+    module Lvm
+      module Encrypt
+        class Format < Getch::FileSystem::Lvm::Encrypt::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 @dev_boot_efi
+            exec("mkfs.#{@fs} -F #{@dev_boot}") if @dev_boot
+            #exec("mkswap -f #{@lv_swap}")
+            exec("mkfs.#{@fs} -F #{@lv_root}")
+            exec("mkfs.#{@fs} -F #{@lv_home}") if @lv_home
+            @state.format
+          end
+
+          private
+          def exec(cmd)
+            Getch::Command.new(cmd).run!
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,64 @@
+require 'fileutils'
+
+module Getch
+  module FileSystem
+    module Lvm
+      module Encrypt
+        class Mount < Getch::FileSystem::Lvm::Encrypt::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 ! @lv_swap
+            system("swapon #{@lv_swap}")
+          end
+
+          def mount_root
+            return if ! @lv_root
+            Dir.mkdir(@root_dir, 0700) if ! Dir.exist?(@root_dir)
+            system("mount #{@lv_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 ! @lv_home
+            if @user != nil then
+              FileUtils.mkdir_p @home_dir, mode: 0700 if ! Dir.exist?(@home_dir)
+              system("mount #{@lv_home} #{@home_dir}")
+            end
+            @state.mount
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,92 @@
+module Getch
+  module FileSystem
+    module Lvm
+      module Encrypt
+        class Partition < Getch::FileSystem::Lvm::Encrypt::Device
+          def initialize
+            super
+            @state = Getch::States.new()
+            @log = Log.new
+            run_partition
+          end
+
+          def run_partition
+            return if STATES[:partition ]
+            clear_struct
+            cleaning
+            partition
+            encrypt
+            lvm
+            @state.partition
+          end
+
+          private
+
+          def clear_struct
+            oldvg = `vgdisplay | grep #{@vg}`.chomp
+            exec("vgremove -f #{@vg}") if oldvg != '' # remove older volume group
+            exec("pvremove -f #{@dev_root}") if oldvg != '' and File.exist? @dev_root # remove older volume group
+
+            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
+
+          def partition
+            if Helpers::efi?
+              exec("sgdisk -n1:1M:+260M -t1:EF00 /dev/#{@disk}")
+              exec("sgdisk -n2:0:+0 -t2:8e00 /dev/#{@disk}")
+            else
+              exec("sgdisk -n1:1MiB:+1MiB -t1:EF02 /dev/#{@disk}")
+              exec("sgdisk -n2:0:+128MiB -t2:8300 /dev/#{@disk}")
+              exec("sgdisk -n3:0:+0 -t3:8e00 /dev/#{@disk}")
+            end
+          end
+
+          def encrypt
+            @log.info("Format root")
+            Helpers::sys("cryptsetup luksFormat #{@dev_root}")
+            @log.debug("Opening root")
+            Helpers::sys("cryptsetup open --type luks #{@dev_root} cryptroot")
+          end
+
+          def lvm
+            mem=`awk '/MemTotal/ {print $2}' /proc/meminfo`.chomp + 'K'
+            exec("pvcreate -f #{@luks_root}")
+            exec("vgcreate -f #{@vg} #{@luks_root}")
+            # Wipe old signature: https://github.com/chef-cookbooks/lvm/issues/45
+            exec("lvcreate -y -Wy -Zy -L 15G -n root #{@vg}")
+            exec("lvcreate -y -Wy -Zy -L #{mem} -n swap #{@vg}")
+            exec("lvcreate -y -Wy -Zy -l 100%FREE -n home #{@vg}") if @user
+            exec("vgchange --available y")
+          end
+
+          # Follow https://wiki.archlinux.org/index.php/Partitioning
+          # Partition_efi
+          # /boot/efi - EFI system partition - 260MB
+          # /         - Root
+
+          # Partition_bios
+          # None      - Bios Boot Partition - 1MiB
+          # /boot     - Boot - 8300
+          # /         - Root
+
+          def exec(cmd)
+            Getch::Command.new(cmd).run!
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,25 @@
+module Getch
+  module FileSystem
+    module Lvm
+      class Format < Getch::FileSystem::Lvm::Device
+        def initialize
+          super
+          @fs = 'ext4'
+          @state = Getch::States.new()
+          format
+        end
+
+        def format
+          return if STATES[:format]
+          puts "Format #{@disk} with #{@fs}"
+          system("mkfs.fat -F32 #{@dev_boot_efi}") if @dev_boot_efi
+          system("mkfs.#{@fs} -F #{@dev_boot}") if @dev_boot
+          system("mkswap -f #{@lv_swap}")
+          system("mkfs.#{@fs} -F #{@lv_root}")
+          system("mkfs.#{@fs} -F #{@lv_home}") if @lv_home
+          @state.format
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,62 @@
+require 'fileutils'
+
+module Getch
+  module FileSystem
+    module Lvm
+      class Mount < Getch::FileSystem::Lvm::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 ! @lv_swap
+          system("swapon #{@lv_swap}")
+        end
+
+        def mount_root
+          return if ! @lv_root
+          Dir.mkdir(@root_dir, 0700) if ! Dir.exist?(@root_dir)
+          system("mount #{@lv_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 ! @lv_home
+          if @user != nil then
+            FileUtils.mkdir_p @home_dir, mode: 0700 if ! Dir.exist?(@home_dir)
+            system("mount #{@lv_home} #{@home_dir}")
+          end
+          @state.mount
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,81 @@
+module Getch
+  module FileSystem
+    module Lvm
+      class Partition < Getch::FileSystem::Lvm::Device
+        def initialize
+          super
+          @state = Getch::States.new()
+          run_partition
+        end
+
+        def run_partition
+          return if STATES[:partition ]
+          clear_struct
+          cleaning
+          partition
+          lvm
+          @state.partition
+        end
+
+        private
+
+        def clear_struct
+          oldvg = `vgdisplay | grep #{@vg}`.chomp
+          exec("vgremove -f #{@vg}") if oldvg != '' # remove older volume group
+          exec("pvremove -f #{@dev_root}") if oldvg != '' and File.exist? @dev_root # remove older volume group
+
+          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
+
+        def partition
+          if Helpers::efi?
+            exec("sgdisk -n1:1M:+260M -t1:EF00 /dev/#{@disk}")
+            exec("sgdisk -n2:0:+0 -t2:8e00 /dev/#{@disk}")
+          else
+            exec("sgdisk -n1:1MiB:+1MiB -t1:EF02 /dev/#{@disk}")
+            exec("sgdisk -n2:0:+128MiB -t2:8300 /dev/#{@disk}")
+            exec("sgdisk -n3:0:+0 -t3:8e00 /dev/#{@disk}")
+          end
+        end
+
+        def lvm
+          mem=`awk '/MemTotal/ {print $2}' /proc/meminfo`.chomp + 'K'
+          exec("pvcreate -f #{@dev_root}")
+          exec("vgcreate -f #{@vg} #{@dev_root}")
+          # Wipe old signature: https://github.com/chef-cookbooks/lvm/issues/45
+          exec("lvcreate -y -Wy -Zy -L 15G -n root #{@vg}")
+          exec("lvcreate -y -Wy -Zy -L #{mem} -n swap #{@vg}")
+          exec("lvcreate -y -Wy -Zy -l 100%FREE -n home #{@vg}") if @user
+          exec("vgchange --available y")
+        end
+
+        # Follow https://wiki.archlinux.org/index.php/Partitioning
+        # Partition_efi
+        # /boot/efi - EFI system partition - 260MB
+        # /         - Root
+
+        # Partition_bios
+        # None      - Bios Boot Partition - 1MiB
+        # /boot     - Boot - 8300
+        # /         - Root
+
+        def exec(cmd)
+          Getch::Command.new(cmd).run!
+        end
+      end
+    end
+  end
+end

data/lib/getch/gentoo.rb

@@ -32,18 +32,19 @@ module Getch
         config.repo
         config.network
         config.systemd(options)
+        config.hostname
         @state.config
       end
 
       def chroot
         chroot = Getch::Gentoo::Chroot.new()
         chroot.update
-        chroot.world
         chroot.systemd
+        chroot.world
         return if STATES[:gentoo_kernel]
         chroot.kernel
         chroot.kernel_deps
-        chroot.install_tools
+        chroot.install_pkgs
       end
 
       def kernel
@@ -51,6 +52,7 @@ module Getch
         source = Getch::Gentoo::Sources.new()
         new
         source.build_kspp
+        source.init_config
         source.build_others
         source.make
         @state.kernel

data/lib/getch/gentoo/boot.rb

@@ -1,22 +1,62 @@
+require 'fileutils'
+
 module Getch
   module Gentoo
     class Boot
       def initialize(opts)
         @disk = opts.disk
         @user = opts.username
+        @config = Getch.class_fs::Config.new()
       end
 
       def start
-        gen_fstab
-        grub
+        @config.fstab
+        bootloader
         password
         umount
       end
 
+      def bootloader
+        if Helpers::efi?
+          bootctl
+        else
+          grub
+        end
+      end
+
+      # bootctl is alrealy installed with the stage3-amd64-systemd
+      def bootctl
+        bootctl_dep
+        puts "Configuring systemd-boot."
+        # ref: https://forums.gentoo.org/viewtopic-p-8118822.html
+        esp = '/boot/efi'
+        exec_chroot("bootctl --path #{esp} install")
+        datas_loader = [
+          'default gentoo',
+          'timeout 3',
+          'editor 0'
+        ]
+        @config.systemd_boot
+        File.write("#{MOUNTPOINT}/#{esp}/loader/loader.conf", datas_loader.join("\n"))
+
+        FileUtils.cp("#{MOUNTPOINT}/usr/src/linux/arch/x86/boot/bzImage", "#{MOUNTPOINT}/#{esp}/vmlinuz", preserve: true)
+
+        initramfs = Dir.glob("#{MOUNTPOINT}/boot/initramfs-*.img")
+        FileUtils.cp("#{initramfs[0]}", "#{MOUNTPOINT}/#{esp}/initramfs", preserve: true) if initramfs != []
+
+        exec_chroot("bootctl --path #{esp} update")
+      end
+
+      def bootctl_dep
+        puts 'Installing systemd-boot...'
+        exec_chroot("euse -p sys-apps/systemd -E gnuefi")
+        Getch::Emerge.new("sys-apps/systemd efivar").pkg!
+      end
+
       def grub
-        return if Helpers::efi?
         puts 'Installing GRUB...'
-        Helpers::emerge("sys-boot/grub:2", MOUNTPOINT)
+        Getch::Emerge.new("sys-boot/grub:2").pkg!
+        @config.grub
         exec_chroot("grub-install /dev/#{@disk}")
         exec_chroot('grub-mkconfig -o /boot/grub/grub.cfg')
       end
@@ -36,23 +76,18 @@ module Getch
 
       def umount
         Helpers::exec_or_die("umount -l /mnt/gentoo/dev{/shm,/pts,}")
-        Helpers::exec_or_die("umount -R #{MOUNTPONT}")
+        Helpers::exec_or_die("umount -R #{MOUNTPOINT}")
         puts "Reboot when you have done"
       end
 
       private
 
-      def gen_fstab
-        mount = Getch::Mount.new(@disk, @user)
-        mount.gen_fstab
-      end
-
       def exec_chroot(cmd)
         script = "chroot #{MOUNTPOINT} /bin/bash -c \"
           source /etc/profile
           #{cmd}
         \""
-        Helpers::exec_or_die(script)
+        Getch::Command.new(script).run!
       end
     end
   end