gritano 0.4.1 → 0.5.0

data/lib/gritano/console/executor.rb

@@ -0,0 +1,246 @@
+require "terminal-table"
+
+module Gritano
+  module Console
+    class Executor < Gritano::Console::Base
+
+      def initialize(stdin = STDIN, home_dir = Etc.getpwuid.dir, repo_path = Etc.getpwuid.dir)
+        @stdin = stdin
+        @home_dir = home_dir
+        @repo_path = repo_path
+        @ssh_path = File.join(@home_dir, '.ssh')
+        super(@stdin, @home_dir)
+      end
+
+      before_each_command do
+        check_git
+        check_gritano
+        ActiveRecord::Base.establish_connection(
+          YAML::load(File.open(File.join(@home_dir, '.gritano', 'database.yml'))))
+      end
+    
+      add_command "user:list" do |argv|
+        users = User.all
+        msg = Terminal::Table.new do |t|
+          t << ['user']
+          t << :separator
+          users.each do |user|
+            t.add_row [user.login]
+          end
+        end
+        msg = "there is no user registered" if users.count == 0
+        return [true, msg]
+      end
+
+      add_command "user:key:list", "username" do |argv|
+        login, = argv
+        user = User.find_by_login(login)
+        if user
+          keys = user.keys
+          msg = Terminal::Table.new do |t|
+            t << ['keys']
+            t << :separator
+            keys.each do |key|
+              t.add_row [key.name]
+            end
+          end
+          msg = "there is no key registered" if keys.count == 0
+          return [true, msg]
+        else
+          return [false, "User #{login} is not registered"]
+        end
+      end
+
+      add_command "user:repo:list", "username" do |argv|
+        login, = argv
+        user = User.find_by_login(login)
+        if user
+          repos = user.repositories
+          msg = Terminal::Table.new do |t|
+            t << ['repositories']
+            t << :separator
+            repos.each do |repo|
+              t.add_row [repo.name]
+            end
+          end
+          msg = "there is no repository registered" if repos.count == 0
+          return [true, msg]
+        else
+          return [false, "User #{login} is not registered"]
+        end
+      end
+
+      add_command "user:add", "username" do |argv|
+        login, = argv
+        user = User.new(login: login)
+        return [true, "User #{login} added."] if user.save
+        return [false, "#{user.errors.full_messages.join(", ")}."]
+      end
+
+      add_command "user:rm", "username" do |argv|
+        login, = argv
+        user = User.find_by_login(login)
+        if user
+          if user.destroy
+            return [true, "User #{login} removed."] 
+          end
+        end
+        return [false, "User #{login} could not be removed."]
+      end
+
+      add_command "user:key:add", "username keyname < key.pub" do |argv|
+        login, key_name, key_file = argv
+        user = User.find_by_login(login)
+        if user
+          key = user.keys.create(name: key_name, key: @stdin.read)
+          if key.valid?
+            File.open(File.join(@ssh_path, 'authorized_keys'), 'w').write(Key.authorized_keys)
+              return [true, "Key added successfully."]
+          end
+        end
+        return [false, "Key could not be added."]
+      end
+
+      add_command "user:key:rm", "username keyname" do |argv|
+        login, key_name = argv
+        key = Key.where(name: key_name).includes(:user).where("users.login" => login).limit(1)[0]
+        if key
+          if key.destroy
+            File.open(File.join(@ssh_path, 'authorized_keys'), 'w').write(Key.authorized_keys)
+            return [true, "Key removed successfully."]
+          end
+        end
+        return [false, "Key could not be removed."]
+      end
+
+      add_command "user:admin:add", "username" do |argv|
+        login, = argv
+        user = User.find_by_login(login)
+        if user
+          user.admin = true
+          if user.save
+            return [true, "Now, user #{login} is an administrator"]
+          end
+        end
+        return [false, "User #{login} could not be modified"]
+      end
+
+      add_command "user:admin:rm", "username" do |argv|
+        login, = argv
+        user = User.find_by_login(login)
+        if user
+          user.admin = false
+          if user.save
+            return [true, "Now, user #{login} is not an administrator"]
+          end
+        end
+        return [false, "User #{login} could not be modified"]
+      end
+
+      add_command "repo:list" do |argv|
+        repos = Repository.all
+        msg = Terminal::Table.new do |t|
+          t << ['repositories']
+          t << :separator
+          repos.each do |repo|
+            t.add_row [repo.name]
+          end
+        end
+        msg = "there is no repository registered" if repos.count == 0
+        return [true, msg]
+      end
+
+      add_command "repo:add", "reponame.git [username1 username2 ...]*" do |argv|
+        name, user_login = argv
+        repo = Repository.new(name: name, path: @repo_path)
+        if repo.save
+          if user_login
+            argv[1..-1].each do |login|
+              user = User.find_by_login(login)
+              if user
+                user.add_access(repo, :read)
+                user.add_access(repo, :write)
+              end
+            end
+          end
+          return [true, "Repository #{name} created successfully."]
+        end
+        return [false, "Repository #{name} could not be created."]
+      end
+
+      add_command "repo:user:list", "reponame.git" do |argv|
+        name, = argv
+        repo = Repository.find_by_name(name)
+        if repo
+          users = repo.users
+          msg = Terminal::Table.new do |t|
+            t << ['user', 'permission']
+            t << :separator
+            users.each do |user|
+              permissions = ""
+              user.permissions.find_by_repository_id(repo.id) do |p|
+                permissions += "r" if p.is(:read)
+                permissions += "w" if p.is(:write)
+              end
+              t.add_row [user.login, permissions]
+            end
+          end
+          msg = "No user have access to this repository" if users.count == 0
+          return [true, msg]
+        end
+        return [false, "Repository #{name} doesn't exist."]
+      end
+
+      add_command "repo:rm", "reponame.git" do |argv|
+        name, = argv
+        repo = Repository.find_by_name(name)
+        if repo
+          if repo.destroy
+            return [true, "Repository #{name} removed successfully."]
+          end
+        end
+        return [false, "Repository #{name} could not be removed."]
+      end
+
+      add_command "repo:read:add", "reponame.git username" do |argv|
+        repo_name, login = argv
+        user = User.find_by_login(login)
+        repo = Repository.find_by_name(repo_name)
+        if repo and user
+          return [true, "User #{login} has read access to #{repo_name}."] if user.add_access(repo, :read)
+        end
+        return [false, "An error occurred. Permissions was not modified."]
+      end
+
+      add_command "repo:write:add", "reponame.git username" do |argv|
+        repo_name, login = argv
+        user = User.find_by_login(login)
+        repo = Repository.find_by_name(repo_name)
+        if repo and user
+          return [true, "User #{login} has write access to #{repo_name}."] if user.add_access(repo, :write)
+        end
+        return [false, "An error occurred. Permissions was not modified."]
+      end
+
+      add_command "repo:read:rm", "reponame.git username" do |argv|
+        repo_name, login = argv
+        user = User.find_by_login(login)
+        repo = Repository.find_by_name(repo_name)
+        if repo and user
+          return [true, "User #{login} has not read access to #{repo_name}."] if user.remove_access(repo, :read)
+        end
+        return [false, "An error occurred. Permissions was not modified."]
+      end
+
+      add_command "repo:write:rm", "reponame.git username" do |argv|
+        repo_name, login = argv
+        user = User.find_by_login(login)
+        repo = Repository.find_by_name(repo_name)
+        if repo and user
+          return [true, "User #{login} has not write access to #{repo_name}."] if user.remove_access(repo, :write)
+        end
+        return [false, "An error occurred. Permissions was not modified."]
+      end
+    end
+  end
+end

data/lib/gritano/console/gritano.rb

@@ -0,0 +1,35 @@
+module Gritano
+  module Console
+    class Gritano < Gritano::Console::Base
+
+      def initialize(stdin = STDIN, home_dir = Etc.getpwuid.dir, repo_path = Etc.getpwuid.dir)
+        @stdin = stdin
+        @home_dir = home_dir
+        @repo_path = repo_path
+        @ssh_path = File.join(@home_dir, '.ssh')
+        super(@stdin, @home_dir)
+      end
+
+      add_command "help" do |argv|
+        Gritano.commands = Gritano.commands.merge(Installer.commands).merge(Executor.commands)
+        [true, Gritano.help]
+      end
+
+      add_command "version" do |argv|
+        version = "v#{File.open(File.join(File.dirname(__FILE__), '..', '..', '..', 'VERSION')).readlines.join}"
+        [true, version]
+      end
+
+      def method_missing(meth, *args, &block)
+        params = [meth.to_s.gsub("_", ":")] + args[0]
+        begin
+          installer = Installer.new
+          installer.execute(params)
+        rescue
+          executor = Executor.new(@stdin, @home_dir, @repo_path)
+          executor.execute(params)
+        end
+      end
+    end
+  end
+end

data/lib/gritano/console/installer.rb

@@ -0,0 +1,37 @@
+module Gritano
+  module Console
+    class Installer < Gritano::Console::Base
+
+      def initialize(stdin = STDIN, home_dir = Etc.getpwuid.dir)
+        @home_dir = home_dir
+        @stdin = stdin
+        super(@stdin, @home_dir)
+      end
+
+      before_each_command do
+        check_git
+      end
+
+      add_command "setup:prepare" do |argv|
+        Dir.mkdir(File.join(@home_dir, '.gritano')) unless File.exist?(File.join(@home_dir, '.gritano'))
+        Dir.mkdir(File.join(@home_dir, '.ssh')) unless File.exist?(File.join(@home_dir, '.ssh'))
+        File.open(File.join(@home_dir, '.gritano', 'database.yml'), "w") do |f|
+          f.write("adapter: sqlite3\ndatabase: #{File.join(Etc.getpwuid.dir, '.gritano', 'database.db')}\n")
+        end
+        if File.exist?(File.join(@home_dir, '.gritano', 'database.db'))
+          FileUtils.rm(File.join(@home_dir, '.gritano', 'database.db'))
+        end
+        return [true, "configuration has been generated"]
+      end
+
+      add_command "setup:install" do |argv|
+        ActiveRecord::Base.establish_connection(
+          YAML::load(File.open(File.join(@home_dir, '.gritano', 'database.yml'))))
+        ActiveRecord::Migrator.migrate(
+          File.join(File.dirname(__FILE__),'..', '..', '..', 'db/migrate'), 
+          ENV["VERSION"] ? ENV["VERSION"].to_i : nil )
+         [true, "gritano has been installed"]
+      end
+    end
+  end
+end

data/spec/{console_spec.rb → executor_spec.rb}

@@ -1,95 +1,100 @@
 require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
 
-describe Gritano::Console do
+describe Gritano::Console::Executor do
   
   before(:each) do
     stdin = double()
     stdin.stub(:read).and_return("Your SSHKEY here...")
-    @console = Gritano::Console.new(stdin)
+    @console = Gritano::Console::Executor.new(stdin)
   end
   
   it "should respond to gritano user:add igorbonadio" do
     @console.should_receive(:user_add)
-    @console.execute("user:add igorbonadio".split(' '))
+    @console.execute_without_filters("user:add igorbonadio".split(' '))
   end
   
   it "should respond to gritano user:rm igorbonadio" do
     @console.should_receive(:user_rm)
-    @console.execute("user:rm igorbonadio".split(' '))
+    @console.execute_without_filters("user:rm igorbonadio".split(' '))
   end
   
   it "should respond to gritano user:key:add username keyname < key.pub" do
     @console.should_receive(:user_key_add)
-    @console.execute("user:key:add igorbonadio keyname".split(' '))
+    @console.execute_without_filters("user:key:add igorbonadio keyname".split(' '))
   end
   
   it "should respond to gritano user:key:rm username keyname" do
     @console.should_receive(:user_key_rm)
-    @console.execute("user:key:rm username keyname".split(' '))
+    @console.execute_without_filters("user:key:rm username keyname".split(' '))
   end
   
   it "should respond to gritano user:list" do
     @console.should_receive(:user_list)
-    @console.execute("user:list".split(' '))
+    @console.execute_without_filters("user:list".split(' '))
   end
   
   it "should respond to gritano user:key:list username" do
     @console.should_receive(:user_key_list)
-    @console.execute("user:key:list username".split(' '))
+    @console.execute_without_filters("user:key:list username".split(' '))
   end
   
   it "should respond to gritano user:repo:list username" do
     @console.should_receive(:user_repo_list)
-    @console.execute("user:repo:list username".split(' '))
+    @console.execute_without_filters("user:repo:list username".split(' '))
   end
   
   it "should respond to gritano repo:user:list reponame.git" do
     @console.should_receive(:repo_user_list)
-    @console.execute("repo:user:list reponame.git".split(' '))
+    @console.execute_without_filters("repo:user:list reponame.git".split(' '))
   end
   
   it "should respond to gritano user:admin:add igorbonadio" do
     @console.should_receive(:user_admin_add)
-    @console.execute("user:admin:add igorbonadio".split(' '))
+    @console.execute_without_filters("user:admin:add igorbonadio".split(' '))
   end
   
   it "should respond to gritano user:admin:rm igorbonadio" do
     @console.should_receive(:user_admin_rm)
-    @console.execute("user:admin:rm igorbonadio".split(' '))
+    @console.execute_without_filters("user:admin:rm igorbonadio".split(' '))
   end
   
   it "should respond to gritano repo:add tmp/reponame.git" do
     @console.should_receive(:repo_add)
-    @console.execute("repo:add tmp/reponame.git".split(' '))
+    @console.execute_without_filters("repo:add tmp/reponame.git".split(' '))
   end
   
   it "should respond to gritano repo:rm tmp/reponame.git" do
     @console.should_receive(:repo_rm)
-    @console.execute("repo:rm tmp/reponame.git".split(' '))
+    @console.execute_without_filters("repo:rm tmp/reponame.git".split(' '))
   end
   
   it "should respond to gritano repo:read:add igorbonadio tmp/reponame.git" do
     @console.should_receive(:repo_read_add)
-    @console.execute("repo:read:add igorbonadio tmp/reponame.git".split(' '))
+    @console.execute_without_filters("repo:read:add igorbonadio tmp/reponame.git".split(' '))
   end
   
   it "should respond to gritano repo:write:add igorbonadio tmp/reponame.git" do
     @console.should_receive(:repo_write_add)
-    @console.execute("repo:write:add igorbonadio tmp/reponame.git".split(' '))
+    @console.execute_without_filters("repo:write:add igorbonadio tmp/reponame.git".split(' '))
   end
   
   it "should respond to gritano repo:read:rm igorbonadio tmp/reponame.git" do
     @console.should_receive(:repo_read_rm)
-    @console.execute("repo:read:rm igorbonadio tmp/reponame.git".split(' '))
+    @console.execute_without_filters("repo:read:rm igorbonadio tmp/reponame.git".split(' '))
   end
   
   it "should respond to gritano repo:write:rm igorbonadio tmp/reponame.git" do
     @console.should_receive(:repo_write_rm)
-    @console.execute("repo:write:rm igorbonadio tmp/reponame.git".split(' '))
+    @console.execute_without_filters("repo:write:rm igorbonadio tmp/reponame.git".split(' '))
   end
   
   it "should respond to gritano repo:list" do
     @console.should_receive(:repo_list)
-    @console.execute("repo:list".split(' '))
+    @console.execute_without_filters("repo:list".split(' '))
+  end
+
+  it "should respond to gritano help" do
+    @console.should_receive(:help)
+    @console.execute_without_filters("help".split(' '))
   end
 end

data/spec/gritano_spec.rb

@@ -0,0 +1,20 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+describe Gritano::Console::Gritano do
+  
+  before(:each) do
+    stdin = double()
+    stdin.stub(:read).and_return("Your SSHKEY here...")
+    @console = Gritano::Console::Gritano.new(stdin)
+  end
+  
+  it "should respond to gritano version" do
+    @console.should_receive(:version)
+    @console.execute("version".split(' '))
+  end
+  
+  it "should respond to gritano help" do
+    @console.should_receive(:help)
+    @console.execute("help".split(' '))
+  end
+end