slicehost 0.1.0

data/History.txt

@@ -0,0 +1,4 @@
+== 0.1.0 2007-12-23
+
+* 1 major enhancement:
+  * Initial release

data/License.txt

@@ -0,0 +1,20 @@
+Copyright (c) 2007 Dr Nic Williams
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/Manifest.txt

@@ -0,0 +1,28 @@
+History.txt
+License.txt
+Manifest.txt
+README.txt
+Rakefile
+config/hoe.rb
+config/requirements.rb
+lib/slicehost.rb
+lib/slicehost/version.rb
+local/slicehost.rb
+local/slicehost.rb.sample
+log/debug.log
+script/destroy
+script/generate
+script/txt2html
+setup.rb
+tasks/deployment.rake
+tasks/environment.rake
+tasks/website.rake
+test/test_admin.rb
+test/test_helper.rb
+test/test_slices.rb
+test/test_zones.rb
+website/index.html
+website/index.txt
+website/javascripts/rounded_corners_lite.inc.js
+website/stylesheets/screen.css
+website/template.rhtml

data/README.txt

@@ -0,0 +1,12 @@
+= Slicehost Helper API
+
+This library provides a simple set of helper methods
+to manage slices and DNS zones/records on your Slicehost
+account (http://slicehost.com).
+
+The current API is very alpha. It was just the simplest thing that worked.
+There are unit tests demonstrating it working and everything.
+
+Future releases will have a nicer, class-based API. 
+
+Contact: Dr Nic Williams, drnicwilliams@gmail.com

data/Rakefile

@@ -0,0 +1,4 @@
+require 'config/requirements'
+require 'config/hoe' # setup Hoe + all gem configuration
+
+Dir['tasks/**/*.rake'].each { |rake| load rake }

data/config/hoe.rb

@@ -0,0 +1,70 @@
+require 'slicehost/version'
+
+AUTHOR = 'Dr Nic Williams'  # can also be an array of Authors
+EMAIL = "drnicwilliams@gmail.com"
+GEM_NAME = 'slicehost' # what ppl will type to install your gem
+RUBYFORGE_PROJECT = 'slicehost' # The unix name for your project
+HOMEPATH = "http://#{RUBYFORGE_PROJECT}.rubyforge.org"
+DOWNLOAD_PATH = "http://rubyforge.org/projects/#{RUBYFORGE_PROJECT}"
+
+@config_file = "~/.rubyforge/user-config.yml"
+@config = nil
+RUBYFORGE_USERNAME = "unknown"
+def rubyforge_username
+  unless @config
+    begin
+      @config = YAML.load(File.read(File.expand_path(@config_file)))
+    rescue
+      puts <<-EOS
+ERROR: No rubyforge config file found: #{@config_file}
+Run 'rubyforge setup' to prepare your env for access to Rubyforge
+ - See http://newgem.rubyforge.org/rubyforge.html for more details
+      EOS
+      exit
+    end
+  end
+  RUBYFORGE_USERNAME.replace @config["username"]
+end
+
+
+REV = nil 
+# UNCOMMENT IF REQUIRED: 
+# REV = `svn info`.each {|line| if line =~ /^Revision:/ then k,v = line.split(': '); break v.chomp; else next; end} rescue nil
+VERS = Slicehost::VERSION::STRING + (REV ? ".#{REV}" : "")
+RDOC_OPTS = ['--quiet', '--title', 'slicehost documentation',
+    "--opname", "index.html",
+    "--line-numbers", 
+    "--main", "README",
+    "--inline-source"]
+
+class Hoe
+  def extra_deps 
+    @extra_deps.reject! { |x| Array(x).first == 'hoe' } 
+    @extra_deps
+  end 
+end
+
+# Generate all the Rake tasks
+# Run 'rake -T' to see list of generated tasks (from gem root directory)
+hoe = Hoe.new(GEM_NAME, VERS) do |p|
+  p.author = AUTHOR 
+  p.description = p.paragraphs_of("README.txt", 1..3).join("\n\n")
+  p.email = EMAIL
+  p.summary = p.description
+  p.url = HOMEPATH
+  p.rubyforge_name = RUBYFORGE_PROJECT if RUBYFORGE_PROJECT
+  p.test_globs = ["test/**/test_*.rb"]
+  p.clean_globs |= ['**/.*.sw?', '*.gem', '.config', '**/.DS_Store']  #An array of file patterns to delete on clean.
+  
+  # == Optional
+  p.changes = p.paragraphs_of("History.txt", 0..1).join("\n\n")
+  #p.extra_deps = []     # An array of rubygem dependencies [name, version], e.g. [ ['active_support', '>= 1.3.1'] ]
+  
+  #p.spec_extras = {}    # A hash of extra values to set in the gemspec.
+  
+end
+
+CHANGES = hoe.paragraphs_of('History.txt', 0..1).join("\\n\\n")
+PATH    = (RUBYFORGE_PROJECT == GEM_NAME) ? RUBYFORGE_PROJECT : "#{RUBYFORGE_PROJECT}/#{GEM_NAME}"
+hoe.remote_rdoc_dir = File.join(PATH.gsub(/^#{RUBYFORGE_PROJECT}\/?/,''), 'rdoc')
+hoe.rsync_args = '-av --delete --ignore-errors'

data/config/requirements.rb

@@ -0,0 +1,17 @@
+require 'fileutils'
+include FileUtils
+
+require 'rubygems'
+%w[rake hoe newgem rubigen].each do |req_gem|
+  begin
+    require req_gem
+  rescue LoadError
+    puts "This Rakefile requires the '#{req_gem}' RubyGem."
+    puts "Installation: gem install #{req_gem} -y"
+    exit
+  end
+end
+
+$:.unshift(File.join(File.dirname(__FILE__), %w[.. lib]))
+
+require 'slicehost'

data/lib/slicehost.rb

@@ -0,0 +1,172 @@
+$:.unshift File.dirname(__FILE__)
+
+require 'rubygems'
+require 'mechanize'
+
+module Slicehost
+end
+
+class Slicehost::Helper
+  Base = "https://manage.slicehost.com"
+
+  # Slice images
+  Arch200708   = 9
+  CentOS5      = 2
+  DebianEtch   = 4
+  Fedora8      = 5
+  UbuntuDapper = 6
+  UbuntuGutsy  = 8
+
+  # Slice sizes
+  Slice256    = 1
+  Slice512    = 2
+  Slice1024   = 3
+  Slice2048   = 4
+
+  attr_reader :agent, :page
+  
+  def initialize
+    @agent =  WWW::Mechanize.new
+  end
+   # !> `&' interpreted as argument prefix
+  def login(email, password)
+    @page = agent.get("#{Base}/login")
+    form = page.form(nil)
+    form.email = email 
+    form.password = password
+    @page = agent.submit(form)
+  end
+
+  def zone?(url)
+    zone_info(url)
+  end
+  
+  def zone_info(url)
+    dotted_url = "#{url}."
+    @page = agent.get("#{Base}/zones")
+    if row = (page/"#dns tr").select { |tr| td = (tr/"td").first; td && td.inner_html == dotted_url }.first
+      info = {}
+      details_td = (row/"td").last
+      info["records"]    = (details_td/"a").first.get_attribute("href")
+      info["zone_id"]    = info["records"].match(/\d+/)[0].to_i
+      info["delete_url"] = (details_td/"a").last.get_attribute("href")
+      delete_click = (details_td/"a").last.get_attribute("onclick")
+      match = delete_click.match(/s\.setAttribute\('name', 'authenticity_token'\); s\.setAttribute\('value', '(\w+)'\);/)
+      info["authenticity_token"] = match[1]
+      info
+    end
+  end
+
+  def create_zone(url)
+    page = agent.get("#{Base}/zones/new")
+    form = page.form(nil)
+    form.origin = url
+    form.ttl = 3600
+    agent.submit(form)
+  end
+  
+  def delete_zone(url)
+    return unless info = zone?(url)
+    agent.post(info["delete_url"], {
+      "_method" => "delete",
+      "authenticity_token" => info["authenticity_token"]
+    })
+  end
+  
+  def record_info(url)
+    return [] unless info = zone_info(url)
+    @page = agent.get(info["records"])
+    (page/"#dns tr")[1..-1].inject([]) do |mem, tr|
+      name, data, record_type = (tr/"td")[0..2].map { |td| td.inner_html }
+      mem << { "name" => name, "data" => data, "record_type" => record_type }
+      mem
+    end.sort { |a, b| a["record_type"] <=> b["record_type"] }
+  end
+  
+  def create_record(url, options)
+    return unless info = zone_info(url)
+    dotted_url = "#{url}."
+    options["record_type"] ||= "A"
+    options["name"] ||= dotted_url
+    options.each_pair { |name, val| options[name] = [val] unless val.is_a?(Array) }
+    @page = agent.get("#{info['records']}/new")
+    form = page.form(nil)
+    form.set_fields(options)
+    @page = agent.submit(form)
+  end
+  
+  def create_records(url, option_list)
+    option_list.each do |options|
+      create_record(url, options)
+    end
+  end
+  
+  def create_simple_zone(url, ipaddr)
+    create_record(url, "data" => ipaddr)
+    create_record(url, "name" => "www", "data" => ipaddr)
+    (1..3).to_a.each do |ns|
+      create_record(url, "data" => "ns#{ns}.slicehost.net.", "record_type" => "NS")
+    end
+  end
+  
+  def create_googleapps_mx(url)
+    dotted_url = "#{url}."
+    mx_records = <<-RECORD
+    ASPMX.L.GOOGLE.COM.
+    ALT1.ASPMX.L.GOOGLE.COM.
+    ALT2.ASPMX.L.GOOGLE.COM.
+    ASPMX2.GOOGLEMAIL.COM.
+    ASPMX3.GOOGLEMAIL.COM.
+    RECORD
+    mx_records = mx_records.split("\n").map { |mx| mx.strip }
+    mx_aux =  %w[1 3 3 5 5 5 ]
+    aux_count = 0
+    create_records(url, mx_records.inject([]) do |mem, mx|
+      mem << { "name" => dotted_url, "data" => mx, "record_type" => "MX", "aux" => mx_aux[aux_count] }
+      aux_count =+ 1
+      mem
+    end)
+  end
+  
+  # WARNING: Calling this method will add a new slice to your account! 
+  # YOU WILL BE BILLED BY SLICEHOST.
+  # +name+ can be an empty string, Slicehost will allocated a random name to your new slice
+  def create_slice(name, options = {})
+    options["name"] = name
+    options["image_id"] ||= UbuntuDapper
+    options["image_id"] = options["image_id"].to_i.to_s # raise error if not Fixnum
+    options["flavor"] ||= Slice256
+    options["flavor"] = options["flavor"].to_i.to_s # raise error if not Fixnum
+    @page = agent.get("#{Base}/slices/new")
+    form = page.form(nil)
+    form.set_fields(options)
+    @page = agent.submit(form)
+  end
+  
+  # Output is an Array of:
+  # {"status"=>"active", "name"=>"slicehost01", "url"=>"/slices/1234", 
+  #  "bandwidth"=>{"total"=>"2.15GB", "inout"=>"0.17 in / 1.99 out"}, 
+  #  "backups"=>"On", "ip"=>"123.456.789.012", "ram"=>"512MB", "age"=>"6 months"}
+  def slices
+    @page = agent.get("#{Base}/slices")
+    (page/"#slice_list tr")[1..-1].map do |tr|
+      td_list = tr/"td"
+      slice_name = (td_list[0]/"a").first
+      slice = {
+        "name"      => slice_name.inner_html,
+        "url"       => slice_name.get_attribute("href"),
+        "status"    => (td_list[1]/"span").first.inner_html.strip,
+        "ram"       => td_list[2].inner_html,
+        "ip"        => (td_list[3]/"span").first.children.first.to_s.strip,
+        "backups"   => td_list[5].inner_html,
+        "age"       => td_list[6].inner_html
+      }
+      bandwidth = td_list[4]
+      slice["bandwidth"] = {
+        "total" => bandwidth.children.first,
+        "inout" => bandwidth.children.last.inner_html
+      }
+      slice
+    end
+  end
+end

data/lib/slicehost/version.rb

@@ -0,0 +1,9 @@
+module Slicehost #:nodoc:
+  module VERSION #:nodoc:
+    MAJOR = 0
+    MINOR = 1
+    TINY  = 0
+
+    STRING = [MAJOR, MINOR, TINY].join('.')
+  end
+end

data/local/slicehost.rb

@@ -0,0 +1,2 @@
+ENV['SLICEHOST_EMAIL'] = "drnicwilliams@gmail.com"
+ENV['SLICEHOST_PASSWORD'] = "UHUst0ck"

data/local/slicehost.rb.sample

@@ -0,0 +1,2 @@
+ENV['SLICEHOST_EMAIL'] = "your@email.com"
+ENV['SLICEHOST_PASSWORD'] = "yourpassword"

data/script/destroy

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+APP_ROOT = 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.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,74 @@
+#!/usr/bin/env ruby
+
+require 'rubygems'
+begin
+  require 'newgem'
+rescue LoadError
+  puts "\n\nGenerating the website requires the newgem RubyGem"
+  puts "Install: gem install newgem\n\n"
+  exit(1)
+end
+require 'redcloth'
+require 'syntax/convertors/html'
+require 'erb'
+require File.dirname(__FILE__) + '/../lib/slicehost/version.rb'
+
+version  = Slicehost::VERSION::STRING
+download = 'http://rubyforge.org/projects/slicehost'
+
+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.rhtml')
+  
+else
+  puts("Usage: #{File.split($0).last} source.txt [template.rhtml] > 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 = fsrc.read
+  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)