lxc-config/hooks/lib/lxc/container.rb

require "erb"
require "netaddr"
require "pathname"
require "fileutils"

module Lxc
  class Container
    def initialize(data, name:, ipv4: nil, ipv6: nil, ula: nil, **options)
      @data = data
      @data["network"] ||= {}
      @data["network"][name] = {}

      zone = @data["zone"] || {}
      @ipv4_subnet = NetAddr::CIDR.create(zone["v4_subnet"] || "192.168.10.0/24")
      @ipv6_subnet = NetAddr::CIDR.create(zone["v6_subnet"] || "fd7d:aed0:18aa::/48")
      @ula_subnet  = NetAddr::CIDR.create(zone["ula_subnet"] || "fdc5:bdb8:b81::/48")

      @container_root = Pathname.new(zone["lxc_root"]).join(name)

      network = data["network"]
      @name = name
      @ipv4 = ipv4 || find_address(@ipv4_subnet, collect_subnets(network, "ipv4"))
      @ipv6 = ipv6 || find_address(@ipv6_subnet, collect_subnets(network, "ipv6"))
      @ula  = ula  || find_address(@ula_subnet,  collect_subnets(network, "ula"))
      @options = options
    end

    def write_config(config_path)
      c = @data["network"][@name] || {}
      c["ipv4"] = NetAddr::CIDR.create(@ipv4).to_s(Short: true)
      c["ipv6"] = NetAddr::CIDR.create(@ipv6).to_s(Short: true)
      c["ula"] = NetAddr::CIDR.create(@ula).to_s(Short: true)
      c["group"] = @options[:group] if @options[:group]
      c["vars"] = @options[:vars] if @options[:vars]

      opts = @options.merge(name: @name,
                            ipv4: format_address(@ipv4, @ipv4_subnet.to_i(:netmask)),
                            ipv6: format_address(@ipv6, @ipv6_subnet.to_i(:netmask)),
                            ula: format_address(@ula, @ula_subnet.to_i(:netmask)))

      config_dir = File.dirname(config_path)
      local_conf = File.join(config_dir, "local.conf")
      unless File.exists?(local_conf)
        FileUtils.touch(local_conf)
      end
      opts[:local_conf] = local_conf

      fstab = @container_root.join("fstab")
      opts[:fstab]  = fstab if File.exists?(fstab)
      opts[:rootfs] = @data["zone"]["shared_rootfs"] || opts[:rootfs] || @container_root.join("rootfs")

      templ = Template.new(CONFIG_ROOT.join("hooks", "templates", "config.erb"))
      templ.write(config_path, opts)
    end

    private
    def format_address(address, netmask)
      NetAddr::CIDR.create(address, Mask: netmask).desc(IP: true, Short: true)
    end

    def collect_subnets(network, type)
      addrs = []
      network.each do |k,v|
        if v[type]
          addrs << NetAddr::CIDR.create(v[type])
        end
      end
      addrs
    end

    def find_address(subnet, assigned_subnets)
      subnet.enumerate(Limit: 1E4, Short: true)[1..1E4].each do |cidr|
        assigned = assigned_subnets.find { |s| s.contains?(cidr) || s == cidr }
        return cidr unless assigned
      end
    end
  end
end