vzlimit 0.0.1

data/History.txt

@@ -0,0 +1,7 @@
+=== 0.0.1 2010-01-30
+
+* Initial release
+* Listing/configuriong resources
+  * Memory Limits (privvmpages)
+  * Disk limits
+* ATTENTION: May only work on x86 and x86_64 hardware yet!

data/Manifest.txt

@@ -0,0 +1,17 @@
+History.txt
+Manifest.txt
+README.rdoc
+Rakefile
+bin/vzlimit
+config/website.yml
+lib/vzlimit.rb
+lib/vzlimit/cli.rb
+script/console
+script/destroy
+script/generate
+script/txt2html
+test/test_helper.rb
+test/test_vzlimit.rb
+test/test_vzlimit_cli.rb
+vzlimit.gemspec
+website/index.html

data/README.rdoc

@@ -0,0 +1,25 @@
+= vzlimit
+
+* http://vzlimit.rubyforge.org/
+
+== DESCRIPTION:
+
+vzlimit simplifies the resource management of the virtualization system 'OpenVZ'. It allows to list all the configured limits as well as changing them in human readable/writable units.
+
+== FEATURES/PROBLEMS:
+
+* FIX (list of features or problems)
+
+== SYNOPSIS:
+
+  vzlimit list
+
+== REQUIREMENTS:
+
+* highline
+* rubyforge
+* hirb
+
+== INSTALL:
+
+* sudo gem install vzlimit

data/Rakefile

@@ -0,0 +1,25 @@
+require 'rubygems'
+gem 'hoe', '>= 2.1.0'
+require 'hoe'
+require 'fileutils'
+require './lib/vzlimit'
+
+Hoe.plugin :newgem
+Hoe.plugin :website
+# Hoe.plugin :cucumberfeatures
+
+# Generate all the Rake tasks
+# Run 'rake -T' to see list of generated tasks (from gem root directory)
+$hoe = Hoe.spec 'vzlimit' do
+  self.developer 'Remo Fritzsche', 'dev@remofritzsche.com'
+  #self.post_install_message = 'PostInstall.txt' # TODO remove if post-install message not required
+  self.rubyforge_name       = 'vzlimit' # TODO this is default value
+  self.extra_deps         = [['highline','>= 1.5.1'], ['rubyforge','>= 2.0.3'], ['hirb', '>= 0.2.9']]
+end
+
+require 'newgem/tasks'
+Dir['tasks/**/*.rake'].each { |t| load t }
+
+# TODO - want other tests/tasks run by default? Add them to the list
+# remove_task :default
+# task :default => [:spec, :features]

data/bin/vzlimit

@@ -0,0 +1,10 @@
+#!/usr/bin/env ruby
+#
+#  Created by Remo Fritzsche on 2010-1-30.
+#  Copyright (c) 2010. All rights reserved.
+
+require 'rubygems'
+require File.expand_path(File.dirname(__FILE__) + "/../lib/vzlimit")
+require "vzlimit/cli"
+
+Vzlimit::CLI.new(STDOUT, ARGV)

data/config/website.yml

@@ -0,0 +1,2 @@
+host: remofritzsche@vzlimit.rubyforge.org
+remote_dir: /var/www/gforge-projects/vzlimit

data/lib/vzlimit/cli.rb

@@ -0,0 +1,201 @@
+require 'optparse'
+require 'highline'
+require 'hirb'
+
+module Vzlimit
+  class CLI
+    ACTIONS = [:limit, :list]
+    VZ_CONF_DIR = '/etc/vz/conf'
+    
+    def initialize(stdout, arguments)
+      @stdout = stdout
+      @arguments = arguments
+      @h = HighLine.new
+      
+      begin
+        # Check action argument
+        if !@arguments[0] ||! ACTIONS.include?(@arguments[0].to_sym)
+          usage
+        end
+        
+        # list
+        if @arguments[0].to_sym == :list
+          list
+        end
+        
+        # limit
+        if @arguments[0].to_sym == :limit
+          limit
+        end
+      rescue
+        @h.say $!
+      end
+    end
+    
+    def list
+      # Scan directory /etc/vz/conf/
+      conf_files = Dir.entries(VZ_CONF_DIR)
+      
+      vps_list = []
+      
+      conf_files.each do |file|
+        if !(file =~ /^\d\d+\.conf$/)
+          next;
+        end
+        
+        vpsid = /^(\d\d+)/.match(file)[0]
+        conf = File.open(VZ_CONF_DIR + '/' + file, "rb").read
+        
+        # Get memory limits
+        null, memory_soft, memory_hard = /\nPRIVVMPAGES="(\d+):(\d+)"\n/.match(conf).to_a
+        memory_soft = (memory_soft.to_f / 1024 * 4).to_i.to_s
+        memory_hard = (memory_hard.to_f / 1024 * 4).to_i.to_s
+        
+        # Get diskspace limits
+        null, diskspace_soft, diskspace_hard = /\nDISKSPACE="(\d+):(\d+)"\n/.match(conf).to_a
+        diskspace_soft = (diskspace_soft.to_f / 1024).to_i.to_s
+        diskspace_hard = (diskspace_hard.to_f / 1024).to_i.to_s
+        
+        # Get ip
+        ip = /\nIP_ADDRESS="(\d+\.\d+\.\d+\.\d+)"\n/.match(conf)[1]
+        
+        vps_list << {
+          :vpsid            => vpsid,
+          :ip               => ip,
+          :memory_soft      => @h.color(memory_soft + ' MB', :yellow),
+          :memory_hard      => @h.color(memory_hard + ' MB', :red),
+          :diskspace_soft   => @h.color(diskspace_soft + ' MB', :yellow),
+          :diskspace_hard   => @h.color(diskspace_hard + ' MB', :red) 
+        }
+        
+        #vps_list << [
+        #  vpsid + "e[32mDONEe[0m",
+        #  ip,
+        #  memory_soft,
+        #  memory_hard,
+        #  diskspace_soft,
+        #  diskspace_hard
+        #]
+      end
+      
+      #@h.say Hirb::Helpers::Table.render(vps_list, 
+      #  :fields => [:vpsid, :ip, :memory_soft, :memory_hard, :diskspace_soft, :diskspace_hard],
+      #  :headers => {:vpsid => 'CTID', :ip => 'IP', :memory_soft => 'MEMORY SOFT LIMIT', :memory_hard => 'MEMORY HARD LIMIT', :diskspace_soft => 'DISKSPACE SOFT LIMIT', :diskspace_hard => 'DISKSPACE HARD LIMIT'}
+      #)
+      HighLine.use_color=false
+    
+      @h.say sprintf("%-8s\t%-15s\t%-20s\t%-20s\t%-20s\t%-20s", 'CTID', 'IP', 'MEMORY SOFT LIMIT', 'MEMORY HARD LIMIT', 'DISKSPACE SOFT LIMIT', 'DISKSPACE HARD LIMIT') + "\n"
+      @h.say sprintf("%-8s\t%-15s\t%-17s\t%-17s\t%-20s\t%-20s", '='*8, '='*15, '='*17, '='*17, '='*20, '='*20) + "\n"
+    
+      vps_list.each do |vps|
+        @h.say sprintf("%-8s\t%-15s\t%-17s\t\t%-17s\t\t%-20s\t\t%-20s", vps[:vpsid], vps[:ip], vps[:memory_soft], vps[:memory_hard], vps[:diskspace_soft], vps[:diskspace_hard]) + "\n"
+      end
+      
+      #@h.say sprintf("%10sworld", "hello");
+      #@h.say @h.list vps_list, :columns_down
+      
+    end
+
+    def limit
+      @arguments.shift
+      usage = 'Usage: vzlimit limit vpsid [options]'
+      raise usage unless @arguments[0]
+      
+      vpsid = @arguments.shift
+      memory = nil
+      disk = nil
+      
+      while (arg = @arguments.shift) do
+        if ['-m', '--memory'].include? arg
+          memory = @arguments.shift
+          
+          if !(memory =~ /^[-+]?[0-9]*\.?[0-9]+:[-+]?[0-9]*\.?[0-9]+$/)
+            raise "Usage: vzlimit limit vpsid #{arg} soft:hard"
+          end
+        elsif ['-d', '--diskspace'].include? arg
+          disk = @arguments.shift
+          
+          if !(disk =~ /^\d+:\d+$/)
+            raise "Usage: vzlimit limit vpsid #{arg} soft:hard"
+          end
+        else
+          raise "Argument #{arg} not valid for action limit."
+        end
+      end
+      
+      if !memory &&! disk
+        raise "Nothing to set."
+      end
+      
+      # Limit memory
+      limit_memory(vpsid, memory) if memory
+      
+      # Limit disk space
+      limit_disk_space(vpsid, disk) if disk
+    end
+    
+    def limit_memory(vpsid, memory)
+      # Get separate soft and hard limit
+      soft_mb, hard_mb = memory.split(':')
+      
+      # Desaster validation
+      if soft_mb.to_f > hard_mb.to_f || soft_mb.to_f < 0 || hard_mb.to_f < 0
+        raise "Memory limits not valid."
+      end
+      
+      # Convert MB limits to mempages
+      soft = (soft_mb.to_f * 1024 / 4).to_i
+      hard = (soft_mb.to_f * 1024 / 4).to_i
+      
+      # Execute command 
+      execute "vzctl set #{vpsid} --privvmpages #{soft}:#{hard} --save"
+      
+      # Success message
+      @h.say @h.color "Memory limits have been set:", :green
+      @h.say "  VPS-ID:             #{vpsid}"
+      @h.say "  Soft memory limit:  #{soft_mb} MB"
+      @h.say "  Hard memory limit:  #{hard_mb} MB"
+    end
+    
+    def limit_disk_space(vpsid, disk)
+      # Get separate soft and hard limit
+      soft_mb, hard_mb = disk.split(':')
+      
+      # Desaster validation
+      if soft_mb.to_f > hard_mb.to_f || soft_mb.to_f < 0 || hard_mb.to_f < 0
+        raise "Disk limits not valid."
+      end
+      
+      # Convert MB limits to 1K-block
+      soft = (soft_mb.to_f * 1024).to_i
+      hard = (soft_mb.to_f * 1024).to_i
+      
+      # Execute command 
+      execute "vzctl set #{vpsid} --diskspace #{soft}:#{hard} --save"
+      
+      # Success message
+      @h.say @h.color "Diskspace limits have been set:", :green
+      @h.say "  VPS-ID:             #{vpsid}"
+      @h.say "  Soft memory limit:  #{soft_mb} MB"
+      @h.say "  Hard memory limit:  #{hard_mb} MB"
+    end
+    
+    def execute(command)
+      @h.say "Will execute the following command:"
+      @h.say "=================================================================================="
+      @h.say command
+      @h.say "=================================================================================="
+      
+      unless @h.agree @h.color("Execute this command?", :red) + " [yes/no]  "
+        raise 'User aborted.'
+      end
+      
+      retval = system command
+      raise @h.color "Error: Command execution failed, exit status #{$?}.", :red unless retval
+    end
+    
+    def usage
+      @h.say "Usage: vzlimit action [params...]"
+    end
+  end
+end

data/lib/vzlimit.rb

@@ -0,0 +1,9 @@
+$:.unshift(File.dirname(__FILE__)) unless
+  $:.include?(File.dirname(__FILE__)) || $:.include?(File.expand_path(File.dirname(__FILE__)))
+
+require 'rubygems'
+
+module Vzlimit
+  VERSION = '0.0.1'
+
+end

data/script/console

@@ -0,0 +1,10 @@
+#!/usr/bin/env ruby
+# File: script/console
+irb = RUBY_PLATFORM =~ /(:?mswin|mingw)/ ? 'irb.bat' : 'irb'
+
+libs =  " -r irb/completion"
+# Perhaps use a console_lib to store any extra methods I may want available in the cosole
+# libs << " -r #{File.dirname(__FILE__) + '/../lib/console_lib/console_logger.rb'}"
+libs <<  " -r #{File.dirname(__FILE__) + '/../lib/vzlimit.rb'}"
+puts "Loading vzlimit gem"
+exec "#{irb} #{libs} --simple-prompt"

data/script/destroy

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+APP_ROOT = File.expand_path(File.join(File.dirname(__FILE__), '..'))
+
+begin
+  require 'rubigen'
+rescue LoadError
+  require 'rubygems'
+  require 'rubigen'
+end
+require 'rubigen/scripts/destroy'
+
+ARGV.shift if ['--help', '-h'].include?(ARGV[0])
+RubiGen::Base.use_component_sources! [:rubygems, :newgem, :newgem_theme, :test_unit]
+RubiGen::Scripts::Destroy.new.run(ARGV)

data/script/generate

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+APP_ROOT = File.expand_path(File.join(File.dirname(__FILE__), '..'))
+
+begin
+  require 'rubigen'
+rescue LoadError
+  require 'rubygems'
+  require 'rubigen'
+end
+require 'rubigen/scripts/generate'
+
+ARGV.shift if ['--help', '-h'].include?(ARGV[0])
+RubiGen::Base.use_component_sources! [:rubygems, :newgem, :newgem_theme, :test_unit]
+RubiGen::Scripts::Generate.new.run(ARGV)

data/script/txt2html

@@ -0,0 +1,71 @@
+#!/usr/bin/env ruby
+
+load File.dirname(__FILE__) + "/../Rakefile"
+require 'rubyforge'
+require 'redcloth'
+require 'syntax/convertors/html'
+require 'erb'
+
+download = "http://rubyforge.org/projects/#{$hoe.rubyforge_name}"
+version  = $hoe.version
+
+def rubyforge_project_id
+  RubyForge.new.configure.autoconfig["group_ids"][$hoe.rubyforge_name]
+end
+
+class Fixnum
+  def ordinal
+    # teens
+    return 'th' if (10..19).include?(self % 100)
+    # others
+    case self % 10
+    when 1: return 'st'
+    when 2: return 'nd'
+    when 3: return 'rd'
+    else    return 'th'
+    end
+  end
+end
+
+class Time
+  def pretty
+    return "#{mday}#{mday.ordinal} #{strftime('%B')} #{year}"
+  end
+end
+
+def convert_syntax(syntax, source)
+  return Syntax::Convertors::HTML.for_syntax(syntax).convert(source).gsub(%r!^<pre>|</pre>$!,'')
+end
+
+if ARGV.length >= 1
+  src, template = ARGV
+  template ||= File.join(File.dirname(__FILE__), '/../website/template.html.erb')
+else
+  puts("Usage: #{File.split($0).last} source.txt [template.html.erb] > output.html")
+  exit!
+end
+
+template = ERB.new(File.open(template).read)
+
+title = nil
+body = nil
+File.open(src) do |fsrc|
+  title_text = fsrc.readline
+  body_text_template = fsrc.read
+  body_text = ERB.new(body_text_template).result(binding)
+  syntax_items = []
+  body_text.gsub!(%r!<(pre|code)[^>]*?syntax=['"]([^'"]+)[^>]*>(.*?)</\1>!m){
+    ident = syntax_items.length
+    element, syntax, source = $1, $2, $3
+    syntax_items << "<#{element} class='syntax'>#{convert_syntax(syntax, source)}</#{element}>"
+    "syntax-temp-#{ident}"
+  }
+  title = RedCloth.new(title_text).to_html.gsub(%r!<.*?>!,'').strip
+  body = RedCloth.new(body_text).to_html
+  body.gsub!(%r!(?:<pre><code>)?syntax-temp-(\d+)(?:</code></pre>)?!){ syntax_items[$1.to_i] }
+end
+stat = File.stat(src)
+created = stat.ctime
+modified = stat.mtime
+
+$stdout << template.result(binding)

data/test/test_helper.rb

@@ -0,0 +1,3 @@
+require 'stringio'
+require 'test/unit'
+require File.dirname(__FILE__) + '/../lib/vzlimit'

data/test/test_vzlimit.rb

@@ -0,0 +1,11 @@
+require File.dirname(__FILE__) + '/test_helper.rb'
+
+class TestVzlimit < Test::Unit::TestCase
+
+  def setup
+  end
+  
+  def test_truth
+    assert true
+  end
+end

data/test/test_vzlimit_cli.rb

@@ -0,0 +1,14 @@
+require File.join(File.dirname(__FILE__), "test_helper.rb")
+require 'vzlimit/cli'
+
+class TestVzlimitCli < Test::Unit::TestCase
+  def setup
+    Vzlimit::CLI.execute(@stdout_io = StringIO.new, [])
+    @stdout_io.rewind
+    @stdout = @stdout_io.read
+  end
+  
+  def test_print_default_output
+    assert_match(/To update this executable/, @stdout)
+  end
+end

data/vzlimit.gemspec

@@ -0,0 +1,49 @@
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{vzlimit}
+  s.version = "0.0.1"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Remo Fritzsche"]
+  s.date = %q{2010-01-30}
+  s.default_executable = %q{vzlimit}
+  s.description = %q{FIX (describe your package)}
+  s.email = ["dev@remofritzsche.com"]
+  s.executables = ["vzlimit"]
+  s.extra_rdoc_files = ["History.txt", "Manifest.txt", "PostInstall.txt", "website/index.txt"]
+  s.files = ["History.txt", "Manifest.txt", "PostInstall.txt", "README.rdoc", "Rakefile", "bin/vzlimit", "config/website.yml.sample", "lib/vzlimit.rb", "lib/vzlimit/cli.rb", "script/console", "script/destroy", "script/generate", "script/txt2html", "test/test_helper.rb", "test/test_vzlimit.rb", "test/test_vzlimit_cli.rb", "website/index.html", "website/index.txt", "website/javascripts/rounded_corners_lite.inc.js", "website/stylesheets/screen.css", "website/template.html.erb"]
+  s.homepage = %q{http://github.com/#{github_username}/#{project_name}}
+  s.post_install_message = %q{PostInstall.txt}
+  s.rdoc_options = ["--main", "README.rdoc"]
+  s.require_paths = ["lib"]
+  s.rubyforge_project = %q{vzlimit}
+  s.rubygems_version = %q{1.3.5}
+  s.summary = %q{FIX (describe your package)}
+  s.test_files = ["test/test_helper.rb", "test/test_vzlimit.rb", "test/test_vzlimit_cli.rb"]
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
+      s.add_runtime_dependency(%q<highline>, [">= 1.5.1"])
+      s.add_runtime_dependency(%q<rubyforge>, [">= 2.0.3"])
+      s.add_development_dependency(%q<rubyforge>, [">= 2.0.3"])
+      s.add_development_dependency(%q<gemcutter>, [">= 0.3.0"])
+      s.add_development_dependency(%q<hoe>, [">= 2.5.0"])
+    else
+      s.add_dependency(%q<highline>, [">= 1.5.1"])
+      s.add_dependency(%q<rubyforge>, [">= 2.0.3"])
+      s.add_dependency(%q<rubyforge>, [">= 2.0.3"])
+      s.add_dependency(%q<gemcutter>, [">= 0.3.0"])
+      s.add_dependency(%q<hoe>, [">= 2.5.0"])
+    end
+  else
+    s.add_dependency(%q<highline>, [">= 1.5.1"])
+    s.add_dependency(%q<rubyforge>, [">= 2.0.3"])
+    s.add_dependency(%q<rubyforge>, [">= 2.0.3"])
+    s.add_dependency(%q<gemcutter>, [">= 0.3.0"])
+    s.add_dependency(%q<hoe>, [">= 2.5.0"])
+  end
+end

data/website/index.html

@@ -0,0 +1,15 @@
+<html>
+	<head>
+		<title>vzlimit - managing OpenVZ limits with comfort</title>
+		<style type="text/css">
+			body
+			{
+				font-family:	Georgia;
+			}
+		</style>
+	</head>
+	<body>
+		<h1>vzlimit</h1>
+		<p>Website comming soon.</p>
+	</body>
+</html>

metadata

@@ -0,0 +1,135 @@
+--- !ruby/object:Gem::Specification 
+name: vzlimit
+version: !ruby/object:Gem::Version 
+  version: 0.0.1
+platform: ruby
+authors: 
+- Remo Fritzsche
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-01-30 00:00:00 +01:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: highline
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 1.5.1
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: rubyforge
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 2.0.3
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: hirb
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 0.2.9
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: rubyforge
+  type: :development
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 2.0.3
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: gemcutter
+  type: :development
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 0.3.0
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: hoe
+  type: :development
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 2.5.0
+    version: 
+description: vzlimit simplifies the resource management of the virtualization system 'OpenVZ'. It allows to list all the configured limits as well as changing them in human readable/writable units.
+email: 
+- dev@remofritzsche.com
+executables: 
+- vzlimit
+extensions: []
+
+extra_rdoc_files: 
+- History.txt
+- Manifest.txt
+files: 
+- History.txt
+- Manifest.txt
+- README.rdoc
+- Rakefile
+- bin/vzlimit
+- config/website.yml
+- lib/vzlimit.rb
+- lib/vzlimit/cli.rb
+- script/console
+- script/destroy
+- script/generate
+- script/txt2html
+- test/test_helper.rb
+- test/test_vzlimit.rb
+- test/test_vzlimit_cli.rb
+- vzlimit.gemspec
+- website/index.html
+has_rdoc: true
+homepage: http://vzlimit.rubyforge.org/
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --main
+- README.rdoc
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: vzlimit
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 3
+summary: vzlimit simplifies the resource management of the virtualization system 'OpenVZ'
+test_files: 
+- test/test_helper.rb
+- test/test_vzlimit.rb
+- test/test_vzlimit_cli.rb