vmc 0.3.23 → 0.4.0.beta.1

data/vmc-ng/lib/vmc/cli/dots.rb

@@ -0,0 +1,133 @@
+module VMC
+  module Dots
+    DOT_COUNT = 3
+    DOT_TICK = 0.15
+
+    class Skipper
+      def initialize(&ret)
+        @return = ret
+      end
+
+      def skip(&callback)
+        @return.call("SKIPPED", :yellow, callback)
+      end
+
+      def give_up(&callback)
+        @return.call("GAVE UP", :red, callback)
+      end
+
+      def fail(&callback)
+        @return.call("FAILED", :red, callback)
+      end
+    end
+
+    def with_progress(message)
+      unless simple_output?
+        print message
+        dots!
+      end
+
+      skipper = Skipper.new do |status, color, callback|
+        unless simple_output?
+          stop_dots!
+          puts "... #{c(status, color)}"
+        end
+
+        return callback && callback.call
+      end
+
+      begin
+        res = yield skipper
+        unless simple_output?
+          stop_dots!
+          puts "... #{c("OK", :green)}"
+        end
+        res
+      rescue
+        unless simple_output?
+          stop_dots!
+          puts "... #{c("FAILED", :red)}"
+        end
+
+        raise
+      end
+    end
+
+    def color?
+      $stdout.tty?
+    end
+
+    COLOR_CODES = {
+      :black => 0,
+      :red => 1,
+      :green => 2,
+      :yellow => 3,
+      :blue => 4,
+      :magenta => 5,
+      :cyan => 6,
+      :white => 7
+    }
+
+    # colored text
+    #
+    # shouldn't use bright colors, as some color themes abuse
+    # the bright palette (I'm looking at you, Solarized)
+    def c(str, color)
+      return str unless color?
+      "\e[3#{COLOR_CODES[color]}m#{str}\e[0m"
+    end
+    module_function :c
+
+    # bold text
+    def b(str)
+      return str unless color?
+      "\e[1m#{str}\e[0m"
+    end
+    module_function :b
+
+    def dots!
+      @dots ||=
+        Thread.new do
+          before_sync = $stdout.sync
+
+          $stdout.sync = true
+
+          printed = false
+          i = 1
+          until @stop_dots
+            if printed
+              print "\b" * DOT_COUNT
+            end
+
+            print ("." * i).ljust(DOT_COUNT)
+            printed = true
+
+            if i == DOT_COUNT
+              i = 0
+            else
+              i += 1
+            end
+
+            sleep DOT_TICK
+          end
+
+          if printed
+            print "\b" * DOT_COUNT
+            print " " * DOT_COUNT
+            print "\b" * DOT_COUNT
+          end
+
+          $stdout.sync = before_sync
+          @stop_dots = nil
+        end
+    end
+
+    def stop_dots!
+      return unless @dots
+      return if @stop_dots
+      @stop_dots = true
+      @dots.join
+      @dots = nil
+    end
+  end
+end

data/vmc-ng/lib/vmc/cli/service.rb

@@ -0,0 +1,122 @@
+require "vmc/cli/command"
+
+module VMC
+  class Service < Command
+    desc "create", "Create a service"
+    group :services, :manage
+    flag(:type) { |choices|
+      ask "What kind?", :choices => choices
+    }
+    flag(:name) { |vendor|
+      random = sprintf("%x", rand(1000000))
+      ask "Name?", :default => "#{vendor}-#{random}"
+    }
+    def create
+      choices = []
+      manifests = {}
+      client.system_services.each do |type, vendors|
+        vendors.each do |vendor, versions|
+          versions.each do |version, _|
+            choice = "#{vendor} #{version}"
+            manifests[choice] = {
+              :type => type,
+              :vendor => vendor,
+              :version => version
+            }
+
+            choices << choice
+          end
+        end
+      end
+
+      type = input(:type, choices)
+      meta = manifests[type]
+
+      # --type redis should work, and just ignore version
+      unless meta
+        _, meta = manifests.find { |k, _| k.split.first == type }
+        fail "Unknown service type." unless meta
+      end
+
+      service = client.service(input(:name, meta[:vendor]))
+      service.type = meta[:type]
+      service.vendor = meta[:vendor]
+      service.version = meta[:version]
+      service.tier = "free"
+
+      with_progress("Creating service") do
+        service.create!
+      end
+    end
+
+    desc "bind", "Bind a service to an application"
+    group :services, :manage
+    flag(:name) { |choices|
+      ask "Which service?", :choices => choices
+    }
+    flag(:app) { |choices|
+      ask "Which application?", :choices => choices
+    }
+    def bind(name = nil, appname = nil)
+      name ||= input(:name, client.services.collect(&:name))
+      appname ||= input(:app, client.apps.collect(&:name))
+
+      with_progress("Binding #{c(name, :blue)} to #{c(appname, :blue)}") do
+        client.app(appname).bind(name)
+      end
+    end
+
+    desc "unbind", "Unbind a service from an application"
+    group :services, :manage
+    flag(:name) { |choices|
+      ask "Which service?", :choices => choices
+    }
+    flag(:app) { |choices|
+      ask "Which application?", :choices => choices
+    }
+    def unbind(name = nil, appname = nil)
+      appname ||= input(:app, client.apps.collect(&:name))
+
+      app = client.app(appname)
+      name ||= input(:name, app.services)
+
+      with_progress("Unbinding #{c(name, :blue)} from #{c(appname, :blue)}") do
+        app.unbind(name)
+      end
+    end
+
+    desc "delete", "Delete a service"
+    group :services, :manage
+    flag(:really) { |name, color|
+      force? || ask("Really delete #{c(name, color)}?", :default => false)
+    }
+    flag(:name) { |choices|
+      ask "Delete which service?", :choices => choices
+    }
+    flag(:all, :default => false)
+    def delete(name = nil)
+      if input(:all)
+        return unless input(:really, "ALL SERVICES", :red)
+
+        with_progress("Deleting all services") do
+          client.services.collect(&:delete!)
+        end
+
+        return
+      end
+
+      unless name
+        services = client.services
+        fail "No services." if services.empty?
+
+        name = input(:name, services.collect(&:name))
+      end
+
+      return unless input(:really, name, :blue)
+
+      with_progress("Deleting #{c(name, :blue)}") do
+        client.service(name).delete!
+      end
+    end
+  end
+end

data/vmc-ng/lib/vmc/cli/user.rb

@@ -0,0 +1,72 @@
+require "vmc/cli/command"
+
+module VMC
+  class User < Command
+    desc "create [EMAIL]", "Create a user"
+    group :admin, :user, :hidden => true
+    flag(:email) {
+      ask("Email")
+    }
+    flag(:password) {
+      ask("Password", :echo => "*", :forget => true)
+    }
+    flag(:verify) {
+      ask("Verify Password", :echo => "*", :forget => true)
+    }
+    def create(email = nil)
+      email ||= input(:email)
+      password = input(:password)
+      verify = input(:verify)
+
+      if password != verify
+        fail "Passwords don't match."
+      end
+
+      with_progress("Creating user") do
+        client.register(email, password)
+      end
+    end
+
+    desc "delete [EMAIL]", "Delete a user"
+    group :admin, :user, :hidden => true
+    flag(:really) { |email|
+      force? || ask("Really delete user #{c(email, :blue)}?", :default => false)
+    }
+    def delete(email)
+      return unless input(:really, email)
+
+      with_progress("Deleting #{c(email, :blue)}") do
+        client.user(email).delete!
+      end
+    ensure
+      forget(:really)
+    end
+
+    desc "passwd [EMAIL]", "Update a user's password"
+    group :admin, :user, :hidden => true
+    flag(:email) {
+      ask("Email")
+    }
+    flag(:password) {
+      ask("Password", :echo => "*", :forget => true)
+    }
+    flag(:verify) {
+      ask("Verify Password", :echo => "*", :forget => true)
+    }
+    def passwd(email = nil)
+      email ||= input(:email)
+      password = input(:password)
+      verify = input(:verify)
+
+      if password != verify
+        fail "Passwords don't match."
+      end
+
+      with_progress("Changing password") do
+        user = client.user(email)
+        user.password = password
+        user.update!
+      end
+    end
+  end
+end

data/vmc-ng/lib/vmc/constants.rb

@@ -0,0 +1,10 @@
+module VMC
+  OLD_TARGET_FILE = "~/.vmc_target"
+  OLD_TOKENS_FILE = "~/.vmc_token"
+
+  CONFIG_DIR = "~/.vmc"
+  PLUGINS_FILE = "#{CONFIG_DIR}/plugins.yml"
+  TARGET_FILE = "#{CONFIG_DIR}/target"
+  TOKENS_FILE = "#{CONFIG_DIR}/tokens.yml"
+  CRASH_FILE = "#{CONFIG_DIR}/crash"
+end

data/vmc-ng/lib/vmc/detect.rb

@@ -0,0 +1,64 @@
+module VMC
+  class Detector
+    def initialize(client, path)
+      @client = client
+      @path = path
+    end
+
+    def all_frameworks
+      info = @client.info
+      info["frameworks"] || {}
+    end
+
+    def frameworks
+      info = @client.info
+
+      matches = {}
+      all_frameworks.each do |name, meta|
+        matched = false
+
+        # e.g. standalone has no detection
+        next if meta["detection"].nil?
+
+        meta["detection"].first.each do |file, match|
+          files =
+            if File.file? @path
+              if File.fnmatch(file, @path)
+                [@path]
+              else
+                []
+              end
+            else
+              Dir.glob("#@path/#{file}")
+            end
+
+          unless files.empty?
+            if match == true
+              matched = true
+            elsif match == false
+              matched = false
+              break
+            else
+              files.each do |f|
+                contents = File.open(f, &:read)
+                if contents =~ Regexp.new(match)
+                  matched = true
+                end
+              end
+            end
+          end
+        end
+
+        if matched
+          matches[name] = meta
+        end
+      end
+
+      if matches.size == 1
+        default = matches.keys.first
+      end
+
+      [matches, default]
+    end
+  end
+end

data/vmc-ng/lib/vmc/errors.rb

@@ -0,0 +1,17 @@
+module VMC
+  class UserError < RuntimeError
+    def initialize(msg)
+      @message = msg
+    end
+
+    def to_s
+      @message
+    end
+  end
+
+  class NotAuthorized < UserError
+    def initialize
+      @message = "Not authorized."
+    end
+  end
+end

data/vmc-ng/lib/vmc/plugin.rb

@@ -0,0 +1,40 @@
+require "set"
+require "yaml"
+
+require "vmc/constants"
+require "vmc/cli"
+
+module VMC
+  module Plugin
+    @@plugins = []
+
+    def self.load_all
+      # auto-load gems with 'vmc-plugin' in their name
+      enabled =
+        Set.new(
+          Gem::Specification.find_all { |s|
+            s.name =~ /vmc-plugin/
+          }.collect(&:name))
+
+      # allow explicit enabling/disabling of gems via config
+      plugins = File.expand_path(VMC::PLUGINS_FILE)
+      if File.exists?(plugins) && yaml = YAML.load_file(plugins)
+        enabled += yaml["enabled"] if yaml["enabled"]
+        enabled -= yaml["disabled"] if yaml["disabled"]
+      end
+
+      # load up each gem's 'plugin' file
+      #
+      # we require this file specifically so people can require the gem
+      # without it plugging into VMC
+      enabled.each do |gemname|
+        require "#{gemname}/plugin"
+      end
+    end
+  end
+
+  def self.Plugin(target = CLI, &blk)
+    # SUPER FANCY PLUGIN SYSTEM
+    target.class_eval &blk
+  end
+end