kensa 0.4.2 → 1.0.0.beta1

data/lib/heroku/kensa/client.rb

@@ -0,0 +1,136 @@
+require 'restclient'
+require 'term/ansicolor'
+require 'launchy'
+
+module Heroku
+  module Kensa
+    class Client
+
+      def initialize(args, options)
+        @args    = args
+        @options = options
+      end
+
+      def filename
+        @options[:filename]
+      end
+
+      class CommandInvalid < Exception; end
+
+      def run!
+        command = @args.shift
+        raise CommandInvalid unless command && respond_to?(command)
+        send(command)
+      end
+
+      def init
+        Manifest.new(filename, @options).write
+        Screen.new.message "Initialized new addon manifest in #{filename}"
+      end
+
+      def test
+        case check = @args.shift
+          when "manifest"
+            run_check ManifestCheck
+          when "provision"
+            run_check ManifestCheck
+            run_check ProvisionCheck
+          when "deprovision"
+            id = ARGV.shift || abort("! no id specified; see usage")
+            run_check ManifestCheck
+            run_check DeprovisionCheck, :id => id
+          when "sso"
+            id = ARGV.shift || abort("! no id specified; see usage")
+            run_check ManifestCheck
+            run_check SsoCheck, :id => id
+          else
+            abort "! Unknown test '#{check}'; see usage"
+        end
+      end
+
+      def run
+        abort "! missing command to run; see usage" if ARGV.empty?
+        run_check ManifestCheck
+        run_check AllCheck, :args => ARGV
+      end
+
+      def sso
+        id = ARGV.shift || abort("! no id specified; see usage")
+        data = Yajl::Parser.parse(resolve_manifest).merge(:id => id)
+        sso = Sso.new(data.merge(@options))
+        puts "Opening #{sso.full_url}"
+        Launchy.open sso.full_url
+      end
+
+      def push
+        require_heroku
+        client   = Heroku::Command.run "auth:client", ['--ignore-keys']
+        host     = ENV['ADDONS_HOST'] || 'https://addons.heroku.com'
+        resource = RestClient::Resource.new(host, client.user, client.password)
+        resource['provider/addons'].post(resolve_manifest)
+        puts "Manifest pushed succesfully"
+      end
+
+      private
+
+        def resolve_manifest
+          if File.exists?(filename)
+            File.read(filename)
+          else
+            abort("fatal: #{filename} not found")
+          end
+        end
+
+        def run_check(klass, args={})
+          screen = Screen.new
+          data = Yajl::Parser.parse(resolve_manifest)
+          check = klass.new(data.merge(@options.merge(args)), screen)
+          check.call
+          screen.finish
+        end
+
+        def require_heroku
+          require 'heroku'
+          require 'heroku/command'
+          require 'heroku/commands/auth'
+        rescue LoadError
+          abort("fatal: Could not load the Heroku gem. Plase make sure the Heroku gem is available and up to date")
+        end
+
+
+      class Screen
+        include Term::ANSIColor
+
+        def test(msg)
+          $stdout.puts
+          $stdout.puts
+          $stdout.print "Testing #{msg}"
+        end
+
+        def check(msg)
+          $stdout.puts
+          $stdout.print "  Check #{msg}"
+        end
+
+        def error(msg)
+          $stdout.print "\n", magenta("    ! #{msg}")
+        end
+
+        def result(status)
+          msg = status ? bold("[PASS]") : red(bold("[FAIL]"))
+          $stdout.print " #{msg}"
+        end
+
+        def message(msg)
+          $stdout.puts msg
+        end
+
+        def finish
+          $stdout.puts
+          $stdout.puts
+          $stdout.puts "done."
+        end
+      end
+    end
+  end
+end

data/lib/heroku/kensa/http.rb

@@ -0,0 +1,53 @@
+require 'restclient'
+
+module Heroku
+  module Kensa
+    module HTTP
+
+      def get(path, params={})
+        path = "#{path}?" + params.map { |k, v| "#{k}=#{v}" }.join("&") unless params.empty?
+        request(:get, [], path)
+      end
+
+      def post(credentials, path, payload=nil)
+        request(:post, credentials, path, payload)
+      end
+
+      def delete(credentials, path, payload=nil)
+        request(:delete, credentials, path, payload)
+      end
+
+      def request(meth, credentials, path, payload=nil)
+        code = nil
+        body = nil
+
+        begin
+          args = [
+            (Yajl::Encoder.encode(payload) if payload),
+            {
+              :accept => "application/json",
+              :content_type => "application/json"
+            }
+          ].compact
+
+          user, pass = credentials
+          body = RestClient::Resource.new(url, user, pass)[path].send(
+            meth,
+            *args
+          ).to_s
+
+          code = 200
+        rescue RestClient::ExceptionWithResponse => boom
+          code = boom.http_code
+          body = boom.http_body
+        rescue Errno::ECONNREFUSED
+          code = -1
+          body = nil
+        end
+
+        [code, body]
+      end
+
+    end
+  end
+end

data/lib/heroku/kensa/manifest.rb

@@ -0,0 +1,52 @@
+module Heroku
+  module Kensa
+    class Manifest
+
+      def initialize(filename = 'addon-manifest.json', options = {})
+        @filename, @options = filename, options
+      end
+
+      def skeleton_json
+        <<-JSON
+{
+  "id": "myaddon",
+  "plans": [
+    { "id": "basic" },
+    { "id": "premium" }
+  ],
+  "api": {
+    "config_vars": [ "MYADDON_URL" ],
+    "username": "heroku",
+    "password": "b1EWrHYXE1R5J71D",#{ sso_key }
+    "production": "https://yourapp.com/",
+    "test": "http://localhost:4567/"
+  }
+}
+JSON
+      end
+
+      def skeleton
+        Yajl::Parser.parse skeleton_json
+      end
+
+      def write
+        open(@filename, 'w') { |f| f << skeleton_json }
+      end
+
+      private
+
+        def sso_key
+          unless @options[:sso] === false
+            %{\n    "sso_salt": #{ generate_password(16).inspect },}
+          end
+        end
+
+        PasswordChars = chars = ['a'..'z', 'A'..'Z', '0'..'9'].map { |r| r.to_a }.flatten
+        def generate_password(size=16)
+          Array.new(size) { PasswordChars[rand(PasswordChars.size)] }.join
+        end
+
+    end
+  end
+end
+

data/lib/heroku/kensa/sso.rb

@@ -0,0 +1,55 @@
+require 'restclient'
+
+module Heroku
+  module Kensa
+    class Sso
+      attr_accessor :id, :url
+
+      def initialize(data)
+        @id   = data[:id]
+        @salt = data['api']['sso_salt']
+
+        env   = data.fetch :env, 'test'
+        @url  = data["api"][env].chomp('/')
+      end
+
+      def path
+        "/heroku/resources/#{id}"
+      end
+
+      def full_url
+        [ url, path, querystring ].join
+      end
+
+      def make_token(t)
+        Digest::SHA1.hexdigest([@id, @salt, t].join(':'))
+      end
+
+      def querystring
+        return '' unless @salt
+
+        t = Time.now.to_i
+        "?token=#{make_token(t)}&timestamp=#{t}&nav-data=#{sample_nav_data}"
+      end
+
+      def sample_nav_data
+        json = Yajl::Encoder.encode({
+          :addon => 'Your Addon',
+          :appname => 'myapp',
+          :addons => [
+            { :slug => 'cron', :name => 'Cron'},
+            { :slug => 'custom_domains+wildcard', :name => 'Custom Domains + Wildcard'},
+            { :slug => 'youraddon', :name => 'Your Addon'},
+          ]
+        })
+        base64_url_variant(json)
+      end
+
+      def base64_url_variant(text)
+        base64 = [text].pack('m').gsub('=', '').gsub("\n", '')
+        base64.tr('+/','-_')
+      end
+
+    end
+  end
+end

data/test/all_check_test.rb

@@ -0,0 +1,26 @@
+require 'test/helper'
+
+class AllCheckTest < Test::Unit::TestCase
+  include Heroku::Kensa
+
+  setup do
+    @data = Manifest.new.skeleton
+    @data['api']['username'] = 'test'
+    @data['api']['password'] = 'secret'
+    @data['api']['test'] += "working"
+    @file = File.dirname(__FILE__) + "/resources/runner.rb"
+  end
+
+  def check; AllCheck; end
+
+  test "valid on script exit 0" do
+    @data[:args] = "ruby #{@file}"
+    assert_valid
+  end
+
+  test "invalid on script exit non 0" do
+    @data[:args] = "ruby #{@file} fail"
+    assert_invalid
+  end
+
+end

data/test/deprovision_check.rb

@@ -1,10 +1,10 @@
-require File.dirname(__FILE__) + "/helper"
+require 'test/helper'
 
 class DeprovisionCheckTest < Test::Unit::TestCase
-  include Heroku::Sensei
+  include Heroku::Kensa
 
   setup do
-    @data = Manifest.skeleton.merge :id => 123
+    @data = Manifest.new.skeleton.merge :id => 123
     @responses = [
       [200, ""],
       [401, ""],

data/test/helper.rb

@@ -1,5 +1,6 @@
 require 'heroku/kensa'
 require 'contest'
+require 'timecop'
 
 class Test::Unit::TestCase
 

data/test/manifest_check_test.rb

@@ -1,12 +1,12 @@
-require File.dirname(__FILE__) + "/helper"
+require 'test/helper'
 
 class ManifestCheckTest < Test::Unit::TestCase
-  include Heroku::Sensei
+  include Heroku::Kensa
 
   def check ; ManifestCheck ; end
 
   setup do
-    @data = Manifest.skeleton
+    @data = Manifest.new.skeleton
     @data["plans"] << {
       "id" => "advanced",
       "name" => "Advanced",
@@ -24,11 +24,6 @@ class ManifestCheckTest < Test::Unit::TestCase
     assert_invalid
   end
 
-  test "has a name" do
-    @data.delete("name")
-    assert_invalid
-  end
-
   test "api key exists" do
     @data.delete("api")
     assert_invalid
@@ -114,29 +109,4 @@ class ManifestCheckTest < Test::Unit::TestCase
     assert_invalid
   end
 
-  test "all plans have a name" do
-    @data["plans"].first.delete("name")
-    assert_invalid
-  end
-
-  test "all plans have a unique name" do
-    @data["plans"].first["name"] = @data["plans"].last["name"]
-    assert_invalid
-  end
-
-  test "plans have a price" do
-    @data["plans"].first.delete("price")
-    assert_invalid
-  end
-
-  test "plans have an integer price" do
-    @data["plans"].first["price"] = "fiddy cent"
-    assert_invalid
-  end
-
-  test "plans have a valid price_unit" do
-    @data["plans"].first["price_unit"] = "first ov da munth"
-    assert_invalid
-  end
-
 end

data/test/manifest_test.rb

@@ -0,0 +1,24 @@
+require 'test/helper'
+
+class ManifestTest < Test::Unit::TestCase
+  include Heroku::Kensa
+
+  context 'manifest' do
+    setup { @manifest = Manifest.new }
+
+    test 'have sso salt' do
+      assert_not_nil @manifest.skeleton['api']['sso_salt']
+    end
+  end
+
+  context 'manifest without sso' do
+    setup do
+      options = { :sso => false }
+      @manifest = Manifest.new 'test.txt', options
+    end
+
+    test 'exclude sso salt' do
+      assert_nil @manifest.skeleton['api']['sso_salt']
+    end
+  end
+end

data/test/provision_check_test.rb

@@ -1,10 +1,10 @@
-require File.dirname(__FILE__) + "/helper"
+require 'test/helper'
 
 class ProvisionCheckTest < Test::Unit::TestCase
-  include Heroku::Sensei
+  include Heroku::Kensa
 
   setup do
-    @data = Manifest.skeleton
+    @data = Manifest.new.skeleton
     @data['api']['username'] = 'test'
     @data['api']['password'] = 'secret'
   end