getch 0.1.9 → 0.3.3

data/lib/devs.rb

@@ -0,0 +1,199 @@
+# frozen_string_literal: true
+
+require 'getch/device'
+
+module Devs
+  class Settings
+    def initialize(args, options)
+      @args = args
+      @start = args[:start] ||= nil
+      @boot = args[:boot] ||= nil
+      @swap = args[:swap] ||= nil
+      @root = args[:root] ||= nil
+      @options = options
+      @encrypt = options[:encrypt] ||= false
+      @lvm = options[:lvm] ||= false
+      @whole = nil
+      @fs = options[:fs]
+      @tree = TREE[@fs.to_sym]
+      x
+    end
+
+    def x
+      add_boot
+      add_swap
+      add_root
+      add_home
+    end
+
+    protected
+
+    def add_boot
+      @options[:boot_disk] || return
+
+      @fs == 'zfs' ?
+        @tree.new(@options[:boot_disk], start: @start, boot: @boot, systemd_boot: systemd_boot?) :
+        @tree.new(@options[:boot_disk], start: @start, boot: @boot)
+
+      @args.delete :start
+      @args.delete :boot if @boot
+    end
+
+    def add_swap
+      @options[:cache_disk] || return
+
+      @fs == 'zfs' ?
+        @tree.new(@options[:cache_disk], swap: @swap, zfs_cache: true) :
+        @tree.new(@options[:cache_disk], swap: @swap)
+      @args.delete :swap if @swap
+    end
+
+    def add_root
+      if @encrypt or @lvm or @fs == 'zfs' then @whole = true end
+      @args[:whole] = @whole
+
+      if systemd_boot? and @fs == 'zfs' then @args.delete :boot end
+      @tree.new(@options[:disk], @args)
+    end
+
+    def add_home
+      @options[:home_disk] || return
+
+      if @encrypt or @lvm or @fs == 'zfs' then @whole = true end
+      @tree.new(@options[:home_disk], home: true, whole: @whole)
+    end
+
+    private
+
+    def efi?
+      Dir.exist? '/sys/firmware/efi/efivars'
+    end
+
+    def systemd_boot?
+      @options[:os] == 'gentoo' and not @options[:musl] and efi?
+    end
+  end
+
+  class Matrice
+    def initialize(disk, args)
+      @disk = disk
+      @start = args[:start] ||= nil
+      @boot = args[:boot] ||= nil
+      @swap = args[:swap] ||= nil
+      @root = args[:root] ||= nil
+      @home = args[:home] ||= nil
+      @whole = args[:whole] ||= false
+      @device = Getch::Device.new
+      @@part = 1
+      x
+    end
+
+    def x
+      start
+      boot
+      swap
+      root
+      home
+    end
+
+    protected
+
+    def start
+      @start || return
+
+      efi? ? add_efi : add_gpt
+    end
+
+    def add_efi
+      @device.efi "#{@disk}#{@@part}"
+      @@part += 1
+    end
+
+    def add_gpt
+      @device.gpt "#{@disk}#{@@part}"
+      @@part += 1
+    end
+
+    def boot
+      @boot || return
+
+      @device.boot "#{@disk}#{@@part}"
+      @@part += 1
+    end
+
+    def swap
+      @swap || return
+
+      @device.swap "#{@disk}#{@@part}"
+      @@part += 1
+    end
+
+    def root
+      @root || return
+
+      if @whole && @@part == 1
+        @device.root @disk
+      else
+        @device.root "#{@disk}#{@@part}"
+        @@part += 1
+      end
+    end
+
+    def home
+      @home || return
+
+      if @whole && @@part == 1
+        @device.home @disk
+      else
+        @device.home "#{@disk}#{@@part}"
+        @@part += 1
+      end
+    end
+
+    private
+
+    def efi?
+      Dir.exist? '/sys/firmware/efi/efivars'
+    end
+  end
+
+  class MatExt4 < Matrice
+  end
+
+  class MatZfs < Matrice
+    def initialize(disk, args)
+      @zfs_cache = args[:zfs_cache] ||= nil
+      @systemd_boot = args[:systemd_boot] ||= false
+      super
+    end
+
+    def boot
+      @boot || return
+
+      @systemd_boot && return
+
+      @device.boot "#{@disk}#{@@part}"
+      @@part += 1
+    end
+
+    def swap
+      @swap || return
+
+      if @zfs_cache
+        @device.swap "#{@disk}#{@@part}"
+        @@part += 1
+        @device.zlog "#{@disk}#{@@part}"
+        @@part += 1
+        @device.zcache "#{@disk}#{@@part}"
+      else
+        @device.swap "#{@disk}#{@@part}"
+        @@part += 1
+      end
+    end
+  end
+
+  TREE = {
+    ext4: MatExt4,
+    zfs: MatZfs,
+  }
+end

data/lib/dracut/encrypt.rb

@@ -0,0 +1,36 @@
+# frozen_string_litteral: true
+
+module Dracut
+  class Encrypt < Root
+    def initialize(devs, options)
+      @luks = options[:luks_name]
+      super
+    end
+
+    def generate
+      host_only
+      cmdline
+      luks_key
+    end
+
+    protected
+
+    def get_line
+      root = Getch::Helpers.uuid @root
+      dm_root = get_dm_uuid "root-#{@luks}"
+      "rd.luks.uuid=#{root} root=UUID=#{dm_root} rootfstype=#{@fs}"
+    end
+
+    def luks_key
+      file = "#{@mountpoint}/etc/dracut.conf.d/luks_key.conf"
+      echo file, 'install_items+=" /boot/boot.key /boot/root.key /etc/crypttab "'
+    end
+
+    private
+
+    def get_dm_uuid(name)
+      dm = Getch::Helpers.get_dm name
+      Getch::Helpers.uuid dm
+    end
+  end
+end

data/lib/dracut/hybrid.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module Dracut
+  class Hybrid < Encrypt
+    def initialize(devs, options)
+      super
+      @vg = options[:vg_name] ||= 'vg0'
+    end
+
+    def get_line
+      root = Getch::Helpers.uuid @root
+      "rd.luks.uuid=#{root} rd.lvm.vg=#{@vg} root=/dev/#{@vg}/root"
+    end
+  end
+end

data/lib/dracut/lvm.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+module Dracut
+  class Lvm < Root
+    def initialize(devs, options)
+      super
+      @vg = options[:vg_name] ||= 'vg0'
+    end
+
+    def get_line
+      "rd.lvm.vg=#{@vg} root=/dev/#{@vg}/root resume=/dev/#{@vg}/swap rootfstype=#{@fs}"
+    end
+  end
+end

data/lib/dracut/minimal.rb

@@ -0,0 +1,11 @@
+# frozen_string_litteral: true
+
+module Dracut
+  class Minimal < Root
+    def get_line
+      swap = Getch::Helpers.uuid @swap
+      root = Getch::Helpers.uuid @root
+      "root=UUID=#{root} rootfstype=#{@fs} resume=UUID=#{swap} rootflags=rw,relatime"
+    end
+  end
+end

data/lib/dracut/root.rb

@@ -0,0 +1,45 @@
+# frozen_string_litteral: true
+
+require 'nito'
+require_relative '../getch/log'
+
+module Dracut
+  class Root
+    include NiTo
+
+    def initialize(devs, options)
+      @log = Getch::Log.new
+      @root = devs[:root] ||= nil
+      @swap = devs[:swap] ||= nil
+      @fs = options[:fs] ||= 'ext4'
+      @mountpoint = options[:mountpoint] ||= '/mnt/getch'
+    end
+
+    def generate
+      host_only
+      cmdline
+      others
+    end
+
+    protected
+
+    def host_only
+      file = "#{@mountpoint}/etc/dracut.conf.d/host.conf"
+      echo file, 'hostonly="yes"'
+      echo_a file, 'use_fstab="yes"'
+    end
+
+    # man dracut.cmdline(7)
+    def cmdline
+      file = "#{@mountpoint}/etc/dracut.conf.d/cmdline.conf"
+      line = get_line
+      echo file, "kernel_cmdline=\"#{line}\""
+    end
+
+    def get_line
+    end
+
+    def others
+    end
+  end
+end

data/lib/dracut/zfs.rb

@@ -0,0 +1,35 @@
+# frozen_string_literal: true
+
+module Dracut
+  class Zfs < Root
+    def initialize(devs, options)
+      super
+      @zfs = options[:zfs_name] ||= 'pool'
+      @encrypt = options[:encrypt]
+      @os = options[:os]
+    end
+
+    def others
+      file = "#{@mountpoint}/etc/dracut.conf.d/zfs.conf"
+      echo file, 'nofsck="yes"'
+      echo_a file, 'omit_dracutmodules+=" btrfs "'
+    end
+
+    # See https://wiki.gentoo.org/wiki/ZFS#ZFS_root
+    # https://github.com/openzfs/zfs/blob/master/contrib/dracut/README.dracut.markdown
+    def get_line
+      @encrypt ?
+        without :
+        with_swap
+    end
+
+    def without
+      "root=zfs:r#{@zfs}/ROOT/#{@os} zfs.force=1 zfs.zfs_arc_max=536870912"
+    end
+
+    def with_swap
+      swap = Getch::Helpers.uuid @swap
+      "resume=UUID=#{swap} root=zfs:r#{@zfs}/ROOT/#{@os} zfs.force=1 zfs.zfs_arc_max=536870912"
+    end
+  end
+end

data/lib/dracut.rb

@@ -0,0 +1,11 @@
+# frozen_string_litteral: true
+
+module Dracut
+end
+
+require_relative 'dracut/root'
+require_relative 'dracut/minimal'
+require_relative 'dracut/lvm'
+require_relative 'dracut/encrypt'
+require_relative 'dracut/hybrid'
+require_relative 'dracut/zfs'

data/lib/fstab/encrypt.rb

@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+
+module Fstab
+  class Encrypt < Root
+    def initialize(devs, options)
+      super
+      @luks = options[:luks_name]
+    end
+
+    def write_boot
+      @boot || return
+
+      dm = Getch::Helpers.get_dm "boot-#{@luks}"
+      uuid = gen_uuid dm
+      line = "UUID=#{uuid} /boot #{@fs} noauto,rw,relatime 0 0"
+      echo_a @conf, line
+    end
+
+    def write_swap
+      @swap || return
+
+      line = "/dev/mapper/swap-#{@luks} none swap rw,noatime,discard 0 0"
+      echo_a @conf, line
+    end
+
+    def write_root
+      @root || return
+
+      dm = Getch::Helpers.get_dm "root-#{@luks}"
+      uuid = gen_uuid dm
+      line = "UUID=#{uuid} / #{@fs} rw,relatime 0 1"
+      echo_a @conf, line
+    end
+
+    def write_home
+      @home || return
+
+      dm = Getch::Helpers.get_dm "home-#{@luks}"
+      uuid = gen_uuid dm
+      line = "UUID=#{uuid} /home #{@fs} rw,relatime 0 2"
+      echo_a @conf, line
+    end
+  end
+end

data/lib/fstab/hybrid.rb

@@ -0,0 +1,34 @@
+
+module Fstab
+  # Hybrid for Lvm + Encryption
+  class Hybrid < Encrypt
+    def initialize(devs, options)
+      super
+      @vg = options[:vg_name] ||= 'vg0'
+      @luks = options[:luks_name]
+    end
+
+    # The swap UUID based on the lvm volume /dev/vg/swap
+    def write_swap
+      # The both use /etc/crypttab
+      if Helpers.runit? or Helpers.systemd?
+        echo_a @conf, "/dev/mapper/swap-#{@luks} none swap sw 0 0"
+      else
+        dm = Getch::Helpers.get_dm "#{@vg}-swap"
+        uuid = Getch::Helpers.uuid dm
+        line = "UUID=#{uuid} none swap sw 0 0"
+        echo_a @conf, line
+      end
+    end
+
+    def write_root
+      line = "/dev/#{@vg}/root / #{@fs} rw,relatime 0 1"
+      echo_a @conf, line
+    end
+
+    def write_home
+      line = "/dev/#{@vg}/home /home #{@fs} rw,relatime 0 2"
+      echo_a @conf, line
+    end
+  end
+end

data/lib/fstab/lvm.rb

@@ -0,0 +1,25 @@
+# frozen_string_litteral: true
+
+module Fstab
+  class Lvm < Root
+    def initialize(devs, options)
+      super
+      @vg = options[:vg_name]
+    end
+
+    def write_swap
+      line = "/dev/#{@vg}/swap swap swap rw,noatime,discard 0 0"
+      echo_a @conf, line
+    end
+
+    def write_root
+      line = "/dev/#{@vg}/root / #{@fs} rw,relatime 0 1"
+      echo_a @conf, line
+    end
+
+    def write_home
+      line = "/dev/#{@vg}/home /home #{@fs} rw,relatime 0 2"
+      echo_a @conf, line
+    end
+  end
+end

data/lib/fstab/minimal.rb

@@ -0,0 +1,6 @@
+# frozen_string_litteral: true
+
+module Fstab
+  class Minimal < Root
+  end
+end

data/lib/fstab/root.rb

@@ -0,0 +1,93 @@
+# frozen_string_litteral: true
+
+require 'nito'
+require_relative '../getch/log'
+
+module Fstab
+  class Root
+    include NiTo
+
+    def initialize(devs, options)
+      @log = Getch::Log.new
+      @efi = devs[:efi]   ||= nil
+      @boot = devs[:boot] ||= nil
+      @swap = devs[:swap] ||= nil
+      @root = devs[:root] ||= nil
+      @home = devs[:home] ||= nil
+      @fs = options[:fs]  ||= 'ext4'
+      @mountpoint = options[:mountpoint] ||= '/mnt/getch'
+      @conf = "#{@mountpoint}/etc/fstab"
+    end
+
+    def generate
+      @log.info 'Generating fstab...'
+      write_efi
+      write_boot
+      write_swap
+      write_root
+      write_home
+      write_tmp
+      @log.result_ok
+    end
+
+    protected
+
+    def write_efi
+      @efi || return
+
+      uuid = gen_uuid @efi
+      line = "UUID=#{uuid} /efi vfat noauto,rw,relatime 0 0"
+      echo_a @conf, line
+    end
+
+    def write_boot
+      @boot || return
+
+      uuid = gen_uuid @boot
+      line = "UUID=#{uuid} /boot #{@fs} noauto,rw,relatime 0 0"
+      echo_a @conf, line
+    end
+
+    def write_swap
+      @swap || return
+
+      uuid = gen_uuid @swap
+      line = "UUID=#{uuid} swap swap rw,noatime,discard 0 0"
+      echo_a @conf, line
+    end
+
+    def write_root
+      @root || return
+
+      uuid = gen_uuid @root
+      line = "UUID=#{uuid} / #{@fs} rw,relatime 0 1"
+      echo_a @conf, line
+    end
+
+    def write_home
+      @home || return
+
+      uuid = gen_uuid @home
+      line = "UUID=#{uuid} /home #{@fs} rw,relatime 0 2"
+      echo_a @conf, line
+    end
+
+    def write_tmp
+      Getch::Helpers.systemd? && return
+
+      line = 'tmpfs /tmp tmpfs defaults,nosuid,nodev 0 0'
+      echo_a @conf, line
+    end
+
+    private
+
+    def gen_uuid(dev)
+      device = dev.delete_prefix('/dev/')
+      Dir.glob('/dev/disk/by-uuid/*').each do |f|
+        link = File.readlink(f)
+        return f.delete_prefix('/dev/disk/by-uuid/') if link.match(/#{device}$/)
+      end
+      @log.fatal "No uuid found for #{device}"
+    end
+  end
+end

data/lib/fstab/zfs.rb

@@ -0,0 +1,23 @@
+module Fstab
+  class Zfs < Root
+    def initialize(devs, options)
+      super
+      @encrypt = options[:encrypt]
+    end
+
+    def generate
+      @log.info 'Generating fstab...'
+      write_efi
+      write_swap
+      @log.result_ok
+    end
+
+    def write_swap
+      uuid = gen_uuid @swap
+      @encrypt ?
+        line = "/dev/mapper/swap-luks none swap sw 0 0" :
+        line = "UUID=#{uuid} swap swap rw,noatime,discard 0 0"
+      echo_a @conf, line
+    end
+  end
+end

data/lib/fstab.rb

@@ -0,0 +1,11 @@
+# frozen_string_litteral: true
+
+module Fstab
+end
+
+require_relative 'fstab/root'
+require_relative 'fstab/minimal'
+require_relative 'fstab/lvm'
+require_relative 'fstab/encrypt'
+require_relative 'fstab/hybrid'
+require_relative 'fstab/zfs'