morale 0.1.0

data/lib/morale/account.rb

@@ -0,0 +1,119 @@
+require 'morale/flow'
+require 'morale/credentials_store'
+
+module Morale
+  class Account
+    include Morale::Platform
+    
+    class << self
+      include Morale::CredentialsStore
+      include Morale::Flow
+      
+      def subdomain
+        if @subdomain.nil?
+          get_credentials
+          @subdomain = @credentials[0]
+        end
+        @subdomain
+      end
+      
+      def subdomain=(value)
+        @subdomain = value
+        @credentials ||= []
+        @credentials[0] = value
+        write_credentials
+      end
+      
+      def api_key
+        get_credentials
+        @credentials[1]
+      end
+      
+      def project
+        @credentials[2] if !@credentials.nil? && @credentials.length > 2
+      end
+      
+      def project=(value)
+        @credentials ||= Array.new(3)
+        @credentials[2] = value
+        write_credentials
+      end
+      
+      def retry_login?
+        @login_attempts ||= 0
+        @login_attempts += 1
+        @login_attempts < 3
+      end
+      
+      def get_credentials
+        return if @credentials
+        unless @credentials = read_credentials
+          ask_for_and_save_credentials
+        end
+        @credentials
+      end
+      
+      private
+
+      def ask_for_and_save_credentials
+        @credentials = ask_for_credentials
+        write_credentials
+      end
+
+      def ask_for_credentials
+        user ||= nil
+        
+        begin
+          user = ask_for_subdomain if @subdomain.nil?
+        rescue Morale::Client::NotFound
+          say "Email is not registered."
+          return
+        end
+        
+        say "Sign in to Morale."
+
+        if user.nil?
+          say "Email: "
+          user = ask
+        end
+
+        say "Password: "
+        password = running_on_windows? ? ask_for_secret_on_windows : ask_for_secret
+        api_key = Morale::Client.authorize(user, password, @subdomain).api_key
+
+        say "Invalid email/password combination or API key was not generated." if api_key.nil?
+
+        [@subdomain, api_key]
+      end
+      
+      def ask_for_subdomain
+        say "No account specified for Morale."
+        
+        say "Email: "
+        user = ask
+
+        accounts = Morale::Client.accounts user
+        account = nil
+        
+        #TODO: This is the same as the account.list --change
+        retryable(:indefinate => true) do
+          accounts.sort{|a,b| a['account']['group_name'] <=> b['account']['group_name']}.each_with_index do |record, i|
+            say "#{i += 1}. #{record['account']['group_name']}"
+          end
+
+          say "Choose an account: "
+          index = ask
+          account = accounts[index.to_i - 1]
+          
+          if account.nil?
+            say "Invalid account."
+            raise Exception
+          end
+        end
+
+        @subdomain = account['account']['site_address'] unless account.nil?
+        user
+      end
+    end
+  end
+end

data/lib/morale/authorization.rb

@@ -0,0 +1,20 @@
+require 'morale/account'
+
+module Morale
+  class Authorization
+    class << self
+      def client
+        Morale::Client.new(Morale::Account.subdomain, Morale::Account.api_key)
+      end
+      
+      def login
+        Morale::Account.delete_credentials
+        Morale::Account.get_credentials
+      end
+      
+      def retry_login?
+        Morale::Account.retry_login?
+      end
+    end
+  end
+end

data/lib/morale/client.rb

@@ -0,0 +1,76 @@
+require 'httparty'
+require "json"
+
+module Morale
+  class Client
+    class Unauthorized  < RuntimeError; end
+    class NotFound < RuntimeError; end
+    
+    include HTTParty
+    format :json
+    
+    API_VERSION = 'v1'
+    
+    attr_accessor :api_key
+    attr_reader   :subdomain
+    
+    def self.authorize(user, password, subdomain)
+      client = new(subdomain)
+      client.unauthorize
+      client.api_key = client.class.post('/in', :body => { :email => user, :password => password })["api_key"]
+      return client
+    end
+    
+    def self.accounts(user)
+      client = new
+      client.unauthorize
+      response = client.class.get("/accounts", :query => { :email => user })
+      raise Unauthorized if response.code == 401
+      raise NotFound if response.code == 404
+      response
+    end
+    
+    def accounts
+      authorize
+      response = self.class.get("/accounts", :query => { :api_key => @api_key })
+      raise Unauthorized if response.code == 401
+      raise NotFound if response.code == 404
+      response
+    end
+    
+    def initialize(subdomain="", api_key="")
+      @api_key = api_key
+      @subdomain = subdomain
+      #TODO: Save the domain in a config file
+      self.class.default_options[:base_uri] = HTTParty.normalize_base_uri("#{subdomain}#{"." unless subdomain.nil? || subdomain.empty?}lvh.me:3000/api/#{API_VERSION}")
+    end
+    
+    def projects
+      authorize
+      response = self.class.get('/projects')
+      raise Unauthorized if response.code == 401
+      raise NotFound if response.code == 404
+      response
+    end
+    
+    def tickets(options={})
+    end
+    
+    def ticket(project_id, command)
+      authorize
+      response = self.class.post("/projects/#{project_id}/tickets", :body => { :command => command })
+      raise Unauthorized if response.code == 401
+      raise NotFound if response.code == 404
+      response
+    end
+    
+    def authorize
+      self.class.basic_auth @subdomain, @api_key
+    end
+    
+    def unauthorize
+      self.class.basic_auth nil, nil
+    end
+    
+  end
+end

data/lib/morale/command.rb

@@ -0,0 +1,70 @@
+require 'rubygems'
+require 'thor'
+require 'morale/commands/account'
+require 'morale/commands/authorization'
+require 'morale/commands/project'
+require 'morale/commands/ticket'
+
+module Morale
+  class Command < Thor
+    include Morale::Platform
+    
+    desc "login", "Signs a user in using their email address and password. Stores the users API key locally to use for access later."
+    def login
+      Morale::Commands::Authorization.login
+    end
+    
+    desc "accounts [EMAIL]", "Gets all the subdomains available for a given email address if provided, else it uses the current api key."    
+    method_options :change => false
+    def accounts(email="")
+      Morale::Commands::Account.list email, options.change
+    end
+    
+    desc "accounts ID", "Changes the current account to the numeric identifier of the account specified."    
+    def account(id)
+      Morale::Commands::Account.select id
+    end
+    
+    desc "projects", "Lists the projects available to the user and the current account."
+    method_options :change => false
+    def projects
+      Morale::Commands::Project.list options.change
+    end
+    
+    desc "projects ID", "Changes the current project to the numeric identifier of the project specified."    
+    def project(id)
+      Morale::Commands::Project.select id
+    end
+    
+    desc "ticket COMMAND", "Creates, updates, or deletes a ticket based on the command specified."
+    method_option :command, :aliases => "-c", :type => :array, :desc => "Specify -c without putting the parameter in a string"
+    def ticket(command="")
+      if command.empty? && !options[:command].nil?
+        command = options[:command].compact.join(" ")
+      end
+      Morale::Commands::Ticket.command command
+    end
+    
+    desc "test", "Testing the ticket output."
+    def test
+      say "+-----+------+-----------------------------------------------------------------------------------------+----------------+----------+----------------+----------+"
+      say "| id  | type | title                                                                                   | created by     | due date | assigned to    | priority |"
+      say "+-----+------+-----------------------------------------------------------------------------------------+----------------+----------+----------------+----------+"
+      say "| 123 | task | hbsdbjhsdjhfdsjfhsdjfhsdjfhsdjfhsdf                                                     | Jimmy P.       | 08/08/11 | Robert P.      | 1        |"
+      say "+-----+------+-----------------------------------------------------------------------------------------+----------------+----------+----------------+----------+"
+    end
+    
+    class << self
+      def client
+        Morale::Authorization.client
+      end
+    end
+    
+    no_tasks do
+      def self.handle_no_task_error(task, has_namespace = $thor_runner) #:nodoc:
+        self.new.ticket ARGV.compact.join(" ")
+      end
+    end
+    
+  end
+end

data/lib/morale/commands/account.rb

@@ -0,0 +1,61 @@
+require 'morale/account'
+require 'morale/client'
+require 'morale/command'
+require 'morale/authorization'
+
+module Morale::Commands
+  class Account
+    class << self
+      include Morale::Platform
+      
+      def list(email="", change=false)
+        accounts = Morale::Client.accounts(email) unless email.nil? || email.empty?
+
+        begin
+          accounts = Morale::Command.client.accounts if email.nil? || email.empty?
+
+          if !accounts.nil?
+            accounts.sort{|a,b| a['account']['group_name'] <=> b['account']['group_name']}.each_with_index do |record, i|
+              say "#{i += 1}. #{record['account']['group_name']}"
+            end
+            
+            if change
+              say "Choose an account: "
+              index = ask
+              account = accounts[index.to_i - 1]
+              
+              if account.nil?
+                say "Invalid account."
+              end
+              Morale::Account.subdomain = account['account']['site_address'] unless account.nil?
+            end
+          else
+            say "There were no accounts found."
+          end
+        rescue Morale::Client::Unauthorized, Morale::Client::NotFound
+          say "Authentication failure"
+          Morale::Commands::Authorization.login
+          retry if Morale::Authorization.retry_login? && !change
+        end
+      end
+      
+      def select(id)
+        begin
+          accounts = Morale::Command.client.accounts
+          if !accounts.nil?
+            account = accounts[id.to_i - 1]
+            if account.nil?
+              say "Invalid account."
+            end
+            Morale::Account.subdomain = account['account']['site_address'] unless account.nil?
+          else
+            say "There were no accounts found."
+          end
+        rescue Morale::Client::Unauthorized, Morale::Client::NotFound
+          say "Authentication failure"
+          Morale::Commands::Authorization.login
+        end
+      end
+    end
+  end
+end

data/lib/morale/commands/authorization.rb

@@ -0,0 +1,13 @@
+require 'morale/authorization'
+
+module Morale::Commands
+  class Authorization
+    class << self
+      
+      def login
+        Morale::Authorization.login
+      end
+      
+    end
+  end
+end

data/lib/morale/commands/project.rb

@@ -0,0 +1,63 @@
+require 'morale/client'
+require 'morale/command'
+require 'morale/authorization'
+
+module Morale::Commands
+  class Project
+    class << self
+      include Morale::Platform
+      
+      def list(change=false)
+        begin
+          projects = Morale::Command.client.projects
+          if !projects.nil?
+            projects.sort{|a,b| a['project']['name'] <=> b['project']['name']}.each_with_index do |record, i|
+              puts "#{i += 1}. #{record['project']['name']}"
+            end
+            
+            if change
+              say "Choose a project: "
+              index = ask
+              project = projects[index.to_i - 1]
+              
+              if project.nil?
+                say "Invalid project."
+              end
+              Morale::Account.project = project['project']['id'] unless project.nil?
+            end
+          else
+            say "There were no projects found."
+          end
+        rescue Morale::Client::Unauthorized
+          say "Authentication failure"
+          Morale::Commands::Authorization.login
+          retry if Morale::Authorization.retry_login?
+        rescue Morale::Client::NotFound
+          say "Communication failure"
+        end
+      end
+      
+      def select(id)
+        begin
+          projects = Morale::Command.client.projects
+          if !projects.nil?
+            project = projects[id.to_i - 1]
+            if project.nil?
+              say "Invalid project."
+            end
+            Morale::Account.project = project['project']['id'] unless project.nil?
+          else
+            say "There were no projects found."
+          end
+        rescue Morale::Client::Unauthorized
+          say "Authentication failure"
+          Morale::Commands::Authorization.login
+          retry if Morale::Authorization.retry_login?
+        rescue Morale::Client::NotFound
+          say "Communication failure"
+        end
+      end
+      
+    end
+  end
+end

data/lib/morale/commands/ticket.rb

@@ -0,0 +1,36 @@
+require 'morale/client'
+require 'morale/command'
+require 'morale/authorization'
+require 'hirb'
+
+module Morale::Commands
+  class Ticket
+    class << self
+      include Morale::Platform
+      
+      def command(command)
+        ticket = Morale::Command.client.ticket Morale::Account.project, command
+        print ticket
+      end
+      
+      private
+      
+      def print(ticket)
+        due_date = Date.parse(ticket['due_date']).strftime("%b. %d") unless ticket['due_date'].nil?
+        assigned_to = "#{ticket['assigned_to']['user']['first_name']} #{(ticket['assigned_to']['user']['last_name']).to_s[0,1]}." unless ticket['assigned_to'].nil?
+        
+        say Hirb::Helpers::Table.render [{
+          :id => ticket['identifier'], 
+          :type => ticket['type'],
+          :title => ticket['title'],
+          :created_by => "#{ticket['created_by']['user']['first_name']} #{(ticket['created_by']['user']['last_name']).to_s[0,1]}.",
+          :due_date => due_date,
+          :assigned_to => assigned_to,
+          :priority => ticket['priority']
+          }], 
+          :fields => [:id, :type, :title, :created_by, :due_date, :assigned_to, :priority],
+          :headers => { :created_by => "created by", :due_date => "due date", :assigned_to => "assigned to" }
+      end
+    end
+  end
+end

data/lib/morale/credentials_store.rb

@@ -0,0 +1,41 @@
+require 'morale/platform'
+
+module Morale
+  module CredentialsStore
+    include Morale::Platform
+    
+    attr_accessor :credentials
+    
+    def location
+      ENV['CREDENTIALS_LOCATION'] || default_location
+    end
+    
+    def default_location
+      "#{home_directory}/.morale/credentials"
+    end
+    
+    def read_credentials
+      File.exists?(location) and File.read(location).split("\n")
+    end
+    
+    def write_credentials
+      FileUtils.mkdir_p(File.dirname(location))
+      f = File.open(location, 'w')
+      f.puts self.credentials
+      f.close
+      set_credentials_permissions
+    end
+    
+    def delete_credentials
+      FileUtils.rm_f(location)
+      @credentials = nil
+    end
+    
+    private
+    
+    def set_credentials_permissions
+      FileUtils.chmod 0700, File.dirname(location)
+      FileUtils.chmod 0600, location
+    end
+  end
+end

data/lib/morale/flow.rb

@@ -0,0 +1,17 @@
+module Morale
+  module Flow
+    def retryable(options = {}, &block)
+      opts = { :tries => 1, :indefinate => false, :on => Exception }.merge(options)
+
+      retry_exception, retries, indefinately = opts[:on], opts[:tries], opts[:indefinate]
+
+      begin
+        return yield
+      rescue retry_exception
+        retry if indefinately || (retries -= 1) > 0
+      end
+
+      yield
+    end
+  end
+end

data/lib/morale/platform.rb

@@ -0,0 +1,64 @@
+module Morale
+  module Platform
+    def home_directory
+      running_on_windows? ? ENV['USERPROFILE'] : ENV['HOME']
+    end
+    
+    def running_on_windows?
+      RUBY_PLATFORM =~ /mswin32|mingw32/
+    end
+    
+    def say(message="", color=nil, force_new_line=(message.to_s !~ /( |\t)$/))
+      message = message.to_s
+      message = set_color(message, color) if color
+
+      spaces = ""
+
+      if force_new_line
+        $stdout.puts(spaces + message)
+      else
+        $stdout.print(spaces + message)
+      end
+      $stdout.flush
+    end
+    
+    def ask
+      input = $stdin.gets
+      input.strip! unless input.nil?
+    end
+    
+    def ask_for_secret_on_windows
+      require "Win32API"
+      char = nil
+      secret = ''
+
+      while char = Win32API.new("crtdll", "_getch", [ ], "L").Call do
+        break if char == 10 || char == 13 # received carriage return or newline
+        if char == 127 || char == 8 # backspace and delete
+          secret.slice!(-1, 1)
+        else
+          # windows might throw a -1 at us so make sure to handle RangeError
+          (secret << char.chr) rescue RangeError
+        end
+      end
+      puts
+      return secret
+    end
+
+    def ask_for_secret
+      echo_off
+      secret = ask
+      puts
+      echo_on
+      return secret
+    end
+    
+    def echo_off
+      system "stty -echo"
+    end
+
+    def echo_on
+      system "stty echo"
+    end
+  end
+end

data/lib/morale.rb

@@ -0,0 +1,5 @@
+module Morale
+  VERSION = "0.1.0"
+end
+
+require 'morale/command'

data/morale.gemspec

@@ -0,0 +1,70 @@
+
+Gem::Specification.new do |s|
+  s.specification_version = 2 if s.respond_to? :specification_version=
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.rubygems_version = '1.3.5'
+
+  s.name              = 'morale'
+  s.version           = '0.1.0'
+  s.date              = '2011-09-29'
+  s.rubyforge_project = 'morale'
+
+  s.summary     = "Command line interface to create & manage tickets on Morale."
+  s.description = "Client library and command-line tool to manage tickets and control your account on Morale."
+
+  s.authors  = ["Brilliant Fantastic"]
+  s.email    = 'support@teammorale.com'
+  s.homepage = 'http://teammorale.com'
+
+  s.require_paths = %w[lib]
+  s.executables = ["morale"]
+
+  s.rdoc_options = ["--charset=UTF-8"]
+  s.extra_rdoc_files = %w[README.md LICENSE]
+
+  s.add_dependency('hirb', "~> 0.5.0")
+  s.add_dependency('httparty', "~> 0.7.8")
+  s.add_dependency('json', "~> 1.4.6")
+  s.add_dependency('thor', "~> 0.14.6")
+
+  #s.add_development_dependency('DEVDEPNAME', [">= 1.1.0", "< 2.0.0"])
+
+  ## Leave this section as-is. It will be automatically generated from the
+  ## contents of your Git repository via the gemspec task. DO NOT REMOVE
+  ## THE MANIFEST COMMENTS, they are used as delimiters by the task.
+  # = MANIFEST =
+  s.files = %w[
+    Gemfile
+    Gemfile.lock
+    LICENSE
+    README.md
+    Rakefile
+    bin/morale
+    features/accounts.feature
+    features/login.feature
+    features/projects.feature
+    features/step_definitions/models.rb
+    features/support/env.rb
+    features/support/hooks.rb
+    features/tickets.feature
+    lib/morale.rb
+    lib/morale/account.rb
+    lib/morale/authorization.rb
+    lib/morale/client.rb
+    lib/morale/command.rb
+    lib/morale/commands/account.rb
+    lib/morale/commands/authorization.rb
+    lib/morale/commands/project.rb
+    lib/morale/commands/ticket.rb
+    lib/morale/credentials_store.rb
+    lib/morale/flow.rb
+    lib/morale/platform.rb
+    morale.gemspec
+    spec/morale/account_spec.rb
+    spec/morale/client_spec.rb
+    spec/morale/command_spec.rb
+    spec/morale/credentials_store_spec.rb
+    spec/spec_helper.rb
+  ]
+  # = MANIFEST =
+end

data/spec/morale/account_spec.rb

@@ -0,0 +1,11 @@
+require 'spec_helper'
+require 'morale/account'
+
+describe Morale::Account do
+  describe "#subdomain" do
+    it "should store the subdomain in the credentials file" do
+      Morale::Account.subdomain = "blah"
+      File.read(Morale::Account.location).should =~ /blah/
+    end
+  end
+end