nutella_framework 0.1.0

data/lib/core/commands/stop.rb

@@ -0,0 +1,62 @@
+require 'core/command'
+require 'core/tmux'
+
+module Nutella
+  class Stop < Command
+    @description = "Stops all or some of the bots in the current project"
+  
+    def run(args=nil)
+      # Is current directory a nutella prj?
+      if !Nutella.currentProject.exist?
+        return
+      end
+    
+      # Extract runid
+      runid = args[0].to_s.empty? ? Nutella.currentProject.config["name"] : Nutella.currentProject.config["name"] + "_" + args[0]
+    
+      # Remove from the list of runs
+      if Nutella.runlist.delete?(runid).nil?
+        console.warn "Run #{runid} doesn't exist. Impossible to stop it."
+        return
+      end
+      # Are we using the internal broker? If yes, stop it
+      if Nutella.runlist.empty? and Nutella.config['broker'] == "localhost" 
+        stopBroker
+      end
+      
+      # Extract project directory
+      @prj_dir = Nutella.currentProject.dir
+    
+      # Stops all the bots
+      Tmux.killSession(runid)
+    
+      # Deletes bots config file if it exists
+      File.delete("#{@prj_dir}/.botsconfig.json") if File.exist?("#{@prj_dir}/.botsconfig.json")
+    
+      # Output success message
+      if runid == Nutella.currentProject.config["name"]
+        console.success "Project #{Nutella.currentProject.config["name"]} stopped"
+      else
+        console.success "Project #{Nutella.currentProject.config["name"]}, run #{args[0]} stopped"
+      end
+    end
+  
+    
+    private
+  
+  
+    def stopBroker
+      pidFile = "#{Nutella.config["broker_dir"]}/bin/.pid"
+      if File.exist?(pidFile) # Does the broker pid file exist?
+        pidF = File.open(pidFile, "rb")
+        pid = pidF.read.to_i
+        pidF.close()
+        Process.kill("SIGKILL", pid)
+        File.delete(pidFile)
+      end
+    end
+  
+  end
+
+  
+end

data/lib/core/nutella_core.rb

@@ -0,0 +1,32 @@
+# Include all commands
+Dir[File.dirname(__FILE__)+"/commands/*.rb"].each do |file|
+  require "core/commands/#{File.basename(file, File.extname(file))}"
+end
+
+module Nutella
+      
+    # Execute command. Returns nil if all is good, otherwise it exceptions out
+    def Nutella.executeCommand (command, args=nil) 
+      # Check that the command exists
+      if commandExists?(command)
+        Object::const_get("Nutella::#{command.capitalize}").new.run(args)
+      else
+        console.error "Unknown command #{command}"
+      end
+    end
+    
+    # Check that a command exists
+    def Nutella.commandExists?(command)
+      return Nutella.const_get("Nutella::#{command.capitalize}").is_a?(Class)
+    rescue NameError
+      return false
+    end
+    
+    # Initialize
+    def Nutella.init
+      Nutella.config["nutella_home"] = NUTELLA_HOME
+      Nutella.config["tmp_dir"] = NUTELLA_HOME+".tmp"
+      Nutella.config["broker_dir"] = NUTELLA_HOME+"broker"
+    end
+
+end

data/lib/core/tmux.rb

@@ -0,0 +1,33 @@
+module Nutella
+  
+  class Tmux
+  
+    def initialize(runId)
+      @runId = runId
+    end
+  
+    def newWindow(bot)
+      if !defined?(@sessions)
+        # If I have no sessions I'm gonna create one and, at the same time, create a new window for the bot
+        `tmux new-session -d -s #{@runId} -n #{bot} &> /dev/null`
+        @sessions = [bot]
+      else
+        # Create new window `bot`
+        # -k destroys it if it can't be created
+        # Pring info about creation of window
+      	`tmux new-window -kP -n #{bot} &> /dev/null` 
+        @sessions.push(bot)
+      end
+      # Select window
+    	`tmux select-window -t #{@runId}:#{@sessions.length-1} &> /dev/null`
+      # Start bot
+      `tmux send-keys "cd bots/#{bot};./startup #{@runId}" C-m`
+    end
+  
+    def self.killSession(runId)
+      `tmux kill-session -t #{runId} &> /dev/null`
+    end
+      
+  end
+  
+end

data/lib/logging/nutella_logger-remote.rb

@@ -0,0 +1,31 @@
+module Nutella
+  
+  class NutellaLoggerRemote < NutellaLogger
+    
+    def debug(message, code=nil)
+      @log.debug(message)
+      code
+    end
+    
+    def info(message, code=nil)
+      @log.info(message)
+      code
+    end
+    
+    def success(message, code=nil)
+      @log.info(ANSI.green + message + ANSI.reset)
+      code
+    end
+    
+    def warn(message, code=nil)
+      @log.warn(ANSI.yellow + message + ANSI.reset)
+      code
+    end
+    
+    def error(message, code=nil)
+      @log.error(ANSI.red + message + ANSI.reset)
+      code
+    end
+    
+  end
+end

data/lib/logging/nutella_logger.rb

@@ -0,0 +1,42 @@
+require 'ansi/code'
+require 'logging'
+
+module Nutella
+  
+  class NutellaLogger
+    
+    def initialize(name)
+      @log = Logging.logger[name]
+      @log.add_appenders(Logging.appenders.stdout(
+        :layout => Logging.layouts.pattern(:pattern => '%m\n')
+        )
+      )
+    end
+    
+    def debug(message, code=nil)
+      @log.debug(ANSI.cyan + message + ANSI.reset)
+      code
+    end
+    
+    def info(message, code=nil)
+      @log.info(message)
+      code
+    end
+    
+    def success(message, code=nil)
+      @log.info(ANSI.green + message + ANSI.reset)
+      code
+    end
+    
+    def warn(message, code=nil)
+      @log.warn(ANSI.yellow + message + ANSI.reset)
+      code
+    end
+    
+    def error(message, code=nil)
+      @log.error(ANSI.red + message + ANSI.reset)
+      code
+    end
+    
+  end
+end

data/lib/logging/nutella_logging.rb

@@ -0,0 +1,35 @@
+require 'singleton'
+require 'logging/nutella_logger'
+require 'logging/nutella_logger-remote'
+
+module Nutella
+
+  class NutellaLogging
+    include Singleton
+    
+    def initialize
+      console = NutellaLogger.new("console")
+      log = NutellaLoggerRemote.new("log")
+      @loggers = {"log" => log, "console" => console}
+    end
+    
+    def logger(name)
+      @loggers[name]
+    end 
+    
+  end
+  
+end
+
+
+module Kernel
+  
+  def console
+    Nutella::NutellaLogging.instance.logger('console')
+  end
+  
+  def log
+    Nutella::NutellaLogging.instance.logger('log')
+  end
+  
+end

data/lib/nutella_framework.rb

@@ -0,0 +1,17 @@
+# Nutella main script
+require 'logging/nutella_logging'
+require 'core/nutella_core'
+require 'cli/nutella_cli'
+require 'config/config'
+require 'config/runlist'
+require 'config/project'
+
+module Nutella
+  NUTELLA_HOME = File.dirname(__FILE__)+"/../"
+  
+  # Store some constants and defaults in the configuration file
+  if Nutella.config.empty?
+    Nutella.init
+  end
+  
+end

data/test/config/test_config.rb

@@ -0,0 +1,47 @@
+require 'helper'
+
+module Nutella
+  
+  class TestNutellaConfig < MiniTest::Test
+    
+    def setup
+      Nutella.config.send(:removeConfigFile)
+    end
+  
+    should "set a key value" do
+      assert_equal "value1", Nutella.config["key1"]="value1"
+    end
+  
+    should "return 'nil' if a key doesn't exist" do   
+      assert_nil Nutella.config["fakekey"]
+    end
+
+    should "return the value associated with a key whenever that key exists" do
+      Nutella.config["key2"]="value2"
+      assert_equal "value2", Nutella.config["key2"]
+    end
+
+    should "retun true if a key exists" do
+      Nutella.config["key3"]="value3"
+      assert Nutella.config.has_key?("key3")
+    end
+
+    should "retun false if a key doens't exist" do
+      refute Nutella.config.has_key?("key4")
+    end
+  
+    should "access nested hashes" do
+      Nutella.config["key5"]={"k55" => "v55"}
+      assert_equal "v55", Nutella.config["key5"]["k55"]
+    end
+  
+    def teardown
+      Nutella.config.send(:removeConfigFile)
+    end
+  
+  end
+  
+  
+end
+
+

data/test/config/test_project.rb

@@ -0,0 +1,32 @@
+require 'helper'
+
+module Nutella
+  
+  class TestProject < MiniTest::Test
+    
+    def setup
+      Nutella.executeCommand "new", ["test_project"]
+      Dir.chdir NUTELLA_HOME + "test_project"
+    end
+    
+    should "return true if the dir is a nutella project" do
+      assert Nutella.currentProject.exist?
+    end
+    
+    should "return false if the dir is not a nutella project" do
+      Dir.chdir NUTELLA_HOME
+      refute Nutella.currentProject.exist?
+    end
+    
+    should "return the correct version of nutella as read from the project configuration file" do
+      assert_equal File.open(NUTELLA_HOME+"VERSION", "rb").read, Nutella.currentProject.config["nutella_version"]
+    end
+    
+    
+    def teardown
+      FileUtils.rm_rf(NUTELLA_HOME + "test_project")
+      Dir.chdir NUTELLA_HOME
+    end
+    
+  end
+end

data/test/config/test_runlist.rb

@@ -0,0 +1,46 @@
+require 'helper'
+
+module Nutella
+  
+  class TestRunList < MiniTest::Test
+    
+    def setup
+      Nutella.runlist.send(:removeRunListFile)
+    end
+    
+    should "return true if the list is empty" do
+      assert Nutella.runlist.empty?
+    end
+
+    should "return false if the list is not empty" do
+      refute_nil Nutella.runlist.add? "run1"
+      refute Nutella.runlist.empty?
+    end
+
+    should "return empty array if the list is empty" do
+      assert_empty Nutella.runlist.to_a
+    end
+
+    should "return an array of runs in the list if not empty" do
+      refute_nil Nutella.runlist.add? "run1"
+      refute_nil Nutella.runlist.add? "run2"
+      assert_equal ["run1", "run2"], Nutella.runlist.to_a
+    end
+    
+    should "return nil if trying to add the same element twice" do
+      refute_nil Nutella.runlist.add? "run1"
+      assert_nil Nutella.runlist.add? "run1"
+    end
+    
+    should "return properly when deleting an item" do
+      refute_nil Nutella.runlist.add? "run1"
+      refute_nil Nutella.runlist.delete? "run1"
+      assert_nil Nutella.runlist.delete? "run1"
+    end
+    
+    def teardown
+      Nutella.runlist.send(:removeRunListFile)
+    end
+    
+  end
+end

data/test/helper.rb

@@ -0,0 +1,36 @@
+require 'simplecov'
+
+module SimpleCov::Configuration
+  def clean_filters
+    @filters = []
+  end
+end
+
+SimpleCov.configure do
+  clean_filters
+  load_profile 'test_frameworks'
+end
+
+ENV["COVERAGE"] && SimpleCov.start do
+  add_filter "/.rvm/"
+end
+require 'rubygems'
+require 'bundler'
+begin
+  Bundler.setup(:default, :development)
+rescue Bundler::BundlerError => e
+  $stderr.puts e.message
+  $stderr.puts "Run `bundle install` to install missing gems"
+  exit e.status_code
+end
+require 'minitest/autorun'
+require 'shoulda'
+
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+require 'nutella_framework'
+
+class MiniTest::Test
+end
+
+MiniTest.autorun

data/test/logging/test_logging.rb

@@ -0,0 +1,16 @@
+require 'helper'
+
+module Nutella
+  class TestNutellaLogger < MiniTest::Test
+    
+    # should "log to console with color and error code" do
+#       assert_equal(453, console.error("This is the message that needs to be logged", 453))
+#     end
+    
+    # shuold "log to appenders and not console" do
+#       log.error("This is an error message that needs to be logged", 354)
+#     end
+    
+  end
+end  
+