wf_node_api 0.6.0

data/lib/wf_node_api/config_check.rb

@@ -0,0 +1,50 @@
+=begin
+__        ___     _ _       _____                ____ __  __
+\ \      / / |__ (_) |_ ___|  ___|   _ ___  ___ / ___|  \/  |
+ \ \ /\ / /| '_ \| | __/ _ \ |_ | | | / __|/ _ \ |   | |\/| |
+  \ V  V / | | | | | ||  __/  _|| |_| \__ \  __/ |___| |  | |
+   \_/\_/  |_| |_|_|\__\___|_|   \__,_|___/\___|\____|_|  |_|
+                                            Container Manager
+
+Copyright (C) 2015 David Prandzioch <kontakt@davidprandzioch.de>
+
+This program is free software; you can redistribute it and/or
+modify it under the terms of the GNU General Public License
+as published by the Free Software Foundation; either version 2
+of the License, or (at your option) any later version.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program; if not, write to the Free Software
+Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
+=end
+
+module ConfigCheck
+
+  # Runs the configuration check
+  #
+  # @raise [RuntimeError]
+  def self.run
+    if self.valid_container_type?($container_type) == false
+      raise RuntimeError, 'container_type is invalid'
+    end
+  end
+
+  # Checks if the submitted container type is registered and valid
+  #
+  # @param container_type [String] The container type to check
+  #
+  # @raise [ArgumentError] If the container type is not set
+  #
+  # @return [Boolean] Result of the validation
+  def self.valid_container_type?(container_type)
+    raise ArgumentError 'container_type not set' if container_type.nil?
+    valid_types = ['lxc', 'vserver']
+
+    valid_types.include?(container_type)
+  end
+end

data/lib/wf_node_api/container_manager.rb

@@ -0,0 +1,286 @@
+=begin
+__        ___     _ _       _____                ____ __  __
+\ \      / / |__ (_) |_ ___|  ___|   _ ___  ___ / ___|  \/  |
+ \ \ /\ / /| '_ \| | __/ _ \ |_ | | | / __|/ _ \ |   | |\/| |
+  \ V  V / | | | | | ||  __/  _|| |_| \__ \  __/ |___| |  | |
+   \_/\_/  |_| |_|_|\__\___|_|   \__,_|___/\___|\____|_|  |_|
+                                            Container Manager
+
+Copyright (C) 2015 David Prandzioch <kontakt@davidprandzioch.de>
+
+This program is free software; you can redistribute it and/or
+modify it under the terms of the GNU General Public License
+as published by the Free Software Foundation; either version 2
+of the License, or (at your option) any later version.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program; if not, write to the Free Software
+Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
+=end
+
+# Manager for containers, meta class
+class ContainerManager
+  # Checks the installation with a simple test procedure
+  #
+  # @raise [StandardError]
+  def check
+    name = (0...8).map { (65 + rand(26)).chr }.join
+    puts "==> Random name for the container: #{name}"
+
+    # containers should return an array
+    puts "==> Listing containers"
+    res = containers()
+    #puts res.inspect
+
+    raise 'containers() failed' unless res.is_a?(Array)
+    puts "==> PASS"
+
+    # container should not work
+    puts "==> Getting container info on non-existant container (should not work)"
+
+    begin
+      res = container(name)
+      raise 'container() failed'
+    rescue => e
+      puts "==> PASS"
+    end
+
+    # delete should not work
+    puts "==> Delete non-existing container (should not work)"
+
+    begin
+      res = delete(name)
+      raise 'delete() failed'
+    rescue => e
+      puts "==> PASS"
+    end
+
+    # start should not work
+    puts "==> Starting non-existing container (should not work)"
+
+    begin
+      res = start(name)
+      raise 'start() failed'
+    rescue => e
+      puts "==> PASS"
+    end
+
+    # stop should not work
+    puts "==> Stopping non-existing container (should not work)"
+
+    begin
+      res = stop(name)
+      raise 'stop() failed'
+    rescue => e
+      puts "==> PASS"
+    end
+
+    # kill should not work
+    puts "==> Killing non-existing container (should not work)"
+
+    begin
+      res = kill(name)
+      raise 'kill() failed'
+    rescue => e
+      puts "==> PASS"
+    end
+
+    # create should work
+    puts "==> Create container"
+    create_container(name, '1.2.3.4', 1, 64, 1)
+    puts "==> PASS"
+
+    # start should work
+    puts "==> Starting container"
+    start(name)
+    puts "==> PASS"
+
+    # stop should work
+    puts "==> Stopping container"
+    stop(name)
+    puts "==> PASS"
+
+    # kill should not work
+    puts "==> Killing stopped container (should not work)"
+
+    begin
+      res = kill(name)
+      raise 'kill() failed'
+    rescue => e
+      puts "==> PASS"
+    end
+
+    # stop should not work
+    puts "==> Stopping stopped container (should not work)"
+
+    begin
+      res = stop(name)
+      raise 'stop() failed'
+    rescue => e
+      puts "==> PASS"
+    end
+
+    # start should work
+    puts "==> Starting container"
+    start(name)
+    puts "==> PASS"
+
+    # kill should work
+    puts "==> Killing container"
+    kill(name)
+    puts "==> PASS"
+
+    # delete should work
+    puts "==> Deleting container"
+    delete(name)
+    puts "==> PASS"
+  end
+
+  # Initializes the manager with the matching adapter
+  #
+  # @param container_type [String] Container adapter
+  #
+  # @raise [ArgumentError] if the container type is invalid
+  def initialize(container_type)
+    if container_type == 'lxc'
+      @adapter = ContainerManagerAdapter::Lxc.new
+    elsif container_type == 'vserver'
+      @adapter = ContainerManagerAdapter::Vserver.new
+    else
+      raise ArgumentError, 'invalid container type'
+    end
+  end
+
+  # Lists all available containers
+  #
+  # @return [Array]
+  def containers
+    @adapter.containers
+  end
+
+  # Returns information for a single container
+  #
+  # @param name [String] The container name
+  #
+  # @raise [NotFoundError]
+  #
+  # @return [Hash] Hash with information about the container
+  def container(name)
+    if !@adapter.exist?(name)
+      raise ::NotFoundError
+    end
+
+    @adapter.container(name)
+  end
+
+  # Returns the amount of free cpu cores
+  #
+  # @param resman [ResourceManager] A ResourceManager instance
+  #
+  # @return [Integer] The amount of free cpu cores
+  def free_cpu_core_count(resman)
+    @adapter.free_cpu_core_count(resman)
+  end
+
+  # Starts a container with the given name
+  #
+  # @param name [String] The container name
+  #
+  # @raise [RuntimeError]
+  # @raise [NotFoundError]
+  #
+  # @return [String] CLI output
+  def start(name)
+    if !@adapter.exist?(name)
+      raise ::NotFoundError
+    end
+
+    @adapter.start(name)
+  end
+
+  # Stops a container with the given name
+  #
+  # @param name [String] The container name
+  #
+  # @raise [RuntimeError]
+  # @raise [NotFoundError]
+  #
+  # @return [String] CLI output
+  def stop(name)
+    if !@adapter.exist?(name)
+      raise ::NotFoundError
+    end
+
+    @adapter.stop(name)
+  end
+
+  # Kills a container with the given name
+  #
+  # @param name [String] The container name
+  #
+  # @raise [RuntimeError]
+  # @raise [NotFoundError]
+  #
+  # @return [String] CLI output
+  def kill(name)
+    if !@adapter.exist?(name)
+      raise ::NotFoundError
+    end
+
+    @adapter.kill(name)
+  end
+
+  # Deletes a container with the given name
+  #
+  # @param name [String] The container name
+  #
+  # @raise [RuntimeError]
+  # @raise [NotFoundError]
+  #
+  # @return [String] CLI output
+  def delete(name)
+    if !@adapter.exist?(name)
+      raise ::NotFoundError
+    end
+
+    @adapter.delete(name)
+  end
+
+  # Creates a container with the given parameters
+  #
+  # @param name             [String]    The container name
+  # @param ip_address       [String]    A valid IPv4 address
+  # @param disk_size_gb     [Integer]   The disk size in GB
+  # @param memory_limit_mb  [Integer]   The memory limit in MB
+  # @param cpu_core_count   [Integer]   Amount of Vcores to assign
+  #
+  # @raise [RuntimeError]
+  # @raise [NotFoundError]
+  # @raise [ArgumentError]
+  #
+  # @return [String] CLI output
+  def create_container(name, ip_address, disk_size_gb, memory_limit_mb, cpu_core_count)
+    disk_size_gb = disk_size_gb.to_i
+    memory_limit_mb = memory_limit_mb.to_i
+    cpu_core_count = cpu_core_count.to_i
+
+    if name.nil? || name.empty? || (/^([A-Za-z0-9_]+)$/ =~ name).nil?
+      raise ArgumentError, 'container name is invalid'
+    end
+
+    if ip_address.nil? || ip_address.empty? || (/^#{Resolv::IPv4::Regex}$/ =~ ip_address).nil?
+      raise ArgumentError, 'the submitted ip address is not a valid ipv4 address'
+    end
+
+    if @adapter.exist?(name)
+      raise ::NotFoundError
+    end
+
+    @adapter.create_container(name, ip_address, disk_size_gb, memory_limit_mb, cpu_core_count)
+  end
+end