traquitana 0.0.11 → 0.0.12

data/lib/migrator.rb

@@ -0,0 +1,25 @@
+require "yaml"
+
+module Traquitana
+   class Migrator
+      def run
+         old_file = "./traq/config.yml"
+         new_file = "./config/traq.yml"
+
+         return false if !File.exists?(old_file) || File.exists?(new_file)
+
+         STDOUT.puts "Migrating old config file ..."
+         contents = YAML.load(File.read(old_file))
+         contents = contents.inject({}) {|hash,val| hash[val.first.to_s] = val.last; hash}.reject {|k,v| k=="ignore"}.to_yaml
+         File.open(new_file,"w") {|f| f<<contents}
+
+         File.unlink(old_file)
+         first_run = "#{File.dirname(old_file)}/.first_run"
+         File.unlink(first_run) if File.exists?(first_run)
+
+         dir = "#{File.dirname(old_file)}"
+         Dir.unlink(dir) if Dir.exists?(dir)
+         true
+      end
+   end
+end

data/lib/packager.rb

@@ -0,0 +1,44 @@
+require "tmpdir"
+require "zip/zip"
+
+module Traquitana
+   class Packager
+      attr_reader :id
+      attr_accessor :verbose
+
+      def initialize(dir="")
+         @dir     = dir
+         @id      = Time.now.strftime("%Y%m%d%H%M%S%L")
+         @verbose = verbose
+      end
+
+      def list_file
+         "#{@id}.list"
+      end
+
+      def zip_file
+         "#{@id}.zip"
+      end
+
+      def pack
+         list_path   = "#{Dir.tmpdir}/#{self.list_file}"
+         zip_path    = "#{Dir.tmpdir}/#{self.zip_file}"
+         list        = Traquitana::Selector.new(@dir).files
+         regex       = @dir.to_s.size<1 ? "" : Regexp.new("^#{@dir}") 
+
+         # write list file
+         STDOUT.puts "Creating the list file: #{list_path}" if @verbose
+         File.open(list_path,"w") {|file| file << list.map {|f| f.sub(regex,"")}.join("\n") }
+
+         # write zip file
+         STDOUT.puts "Creating the zip file : #{zip_path}" if @verbose
+         Zip::ZipFile.open(zip_path ,"w") do |zip_file|
+            for file in list
+               strip = file.sub(regex,"")
+               zip_file.add(strip,file)
+            end
+         end
+         [list_path,zip_path]
+      end
+   end
+end

data/lib/selector.rb

@@ -0,0 +1,24 @@
+module Traquitana
+   class Selector
+      def initialize(dir="")
+         @dir = dir
+      end
+
+      def files
+         config = Traquitana::Config.instance
+         selected = []
+         for file in config.list
+            send, *ignore = *file
+            mask = "#{@dir}#{send}"
+            send_list = Dir.glob(mask).select { |f| File.file?(f) }
+            for ignore_mask in ignore
+               mask = "#{@dir}#{ignore_mask}"
+               ignore_list = Dir.glob(mask).select { |f| File.file?(f) }
+               send_list = send_list - ignore_list if ignore_list.size>0
+            end
+            selected.push(*send_list)
+         end
+         selected
+      end
+   end
+end

data/lib/ssh.rb

@@ -0,0 +1,43 @@
+require "net/scp"
+require "net/ssh"
+
+module Traquitana
+   class SSH
+      attr_reader :host, :user, :options
+
+      def initialize(host,user,options=nil)
+         @host    = host
+         @user    = user
+         @options = options || {}
+			STDOUT.puts "Connecting to #{@host} using user #{@user}"
+      end
+
+      def execute(cmds,verbose=false)
+         rst = []
+         Net::SSH.start(@host,@user,@options) do |ssh|
+            for cmd in cmds
+               STDOUT.puts "Executing #{cmd} on remote host ..." if verbose
+               rst << ssh.exec!(cmd)
+            end
+         end
+         rst
+      end
+
+      def send_files(col,updater=nil)
+			Net::SCP.start(@host,@user,@options) do |scp|
+            for files in col
+               from, to = *files
+               next if from.nil? || to.nil?
+               scp.upload!(from,to) do |ch,name,sent,total|
+                  if !updater.nil?
+                     updater.name  = to
+                     updater.total = total
+                     updater.update(sent)
+                  end
+               end
+               updater.reset if !updater.nil?
+            end
+         end
+      end
+   end
+end

data/lib/traquitana.rb

@@ -0,0 +1,3 @@
+%w(traquitana/version config selector deployer packager ssh bar migrator cleaner).each do |file|
+   require File.dirname(__FILE__)+"/"+file
+end

data/lib/traquitana/version.rb

@@ -0,0 +1,3 @@
+module Traquitana
+  VERSION = "0.0.12"
+end

data/spec/bar_spec.rb

@@ -0,0 +1,45 @@
+require "minitest/autorun"
+require "#{File.expand_path(File.dirname(__FILE__))}/../lib/traquitana.rb"
+
+describe Traquitana::Bar do
+   before do
+      @bar = Traquitana::Bar.new
+      @bar.total = 100
+   end
+
+   describe "limits" do
+      it "should have a total" do
+         @bar.must_respond_to :total
+      end
+      it "should have a current value" do
+         @bar.must_respond_to :current
+      end
+   end
+
+   describe "updates" do
+      it "should have a name" do
+         @bar.must_respond_to :name
+      end
+      it "should have a update method" do
+         @bar.must_respond_to :update
+      end
+      it "should have a indicator method" do
+         @bar.must_respond_to :indicator
+      end
+      it "should return the correct string for 0%" do
+         @bar.indicator(0).must_equal "____________________"
+      end
+      it "should return the correct string for 25%" do
+         @bar.indicator(25).must_equal "#####_______________"
+      end
+      it "should return the correct string for 50%" do
+         @bar.indicator(50).must_equal "##########__________"
+      end
+      it "should return the correct string for 75%" do
+         @bar.indicator(75).must_equal "###############_____"
+      end
+      it "should return the correct string for 100%" do
+         @bar.indicator(100).must_equal "####################"
+      end
+   end
+end

data/spec/cleaner_spec.rb

@@ -0,0 +1,22 @@
+require "minitest/autorun"
+require "#{File.expand_path(File.dirname(__FILE__))}/../lib/traquitana.rb"
+
+describe Traquitana::Cleaner do
+   before do
+      @cleaner = Traquitana::Cleaner.new 
+      @config  = Traquitana::Config.instance
+      @config.load
+   end
+
+   it "should have a run method" do
+      @cleaner.must_respond_to :run  
+   end
+
+   it "should run cleaner on remote host" do
+      network = MiniTest::Mock.new
+      network.expect(:execute,nil,[["find #{@config.directory}/traq -type f -iname '*.zip' -o -iname '*.list' | sort | head -n-2 | xargs rm $1"]])
+      @cleaner.network = network
+      @cleaner.run
+      network.verify
+   end
+end

data/spec/config/config/traq.yml

@@ -0,0 +1,25 @@
+# Sample configuration--- 
+directory: /tmp/traq_test
+user: taq
+list: 
+- Rakefile
+- config.ru
+- Gemfile
+- - config/application.rb
+- - config/environment.rb
+- - config/initializers/**/*
+- - config/environments/production.rb
+- - config/locales/**/*
+- - config/routes.rb
+- - app/**/*
+- - db/migrate/**/*
+- - public/javascripts/**/*
+- - public/stylesheets/**/*
+- - lib/**/*
+- - public/images/**/*
+  - public/images/uploads/**/*
+  - public/images/themes/**/*
+password: fill your password here
+host: localhost
+server: passenger
+shell: bash -l -c 

data/spec/config/network_test.txt

@@ -0,0 +1 @@
+network test

data/spec/config_spec.rb

@@ -0,0 +1,64 @@
+require "yaml"
+require "fileutils"
+require "minitest/autorun"
+require "#{File.expand_path(File.dirname(__FILE__))}/../lib/traquitana.rb"
+
+describe Traquitana::Config do
+   before do
+      @config = Traquitana::Config.instance
+   end
+
+   describe "paths" do
+      it "should have a filename method" do
+         @config.must_respond_to(:filename)   
+      end
+
+      it "should have a filename called traq.yml" do
+         File.basename(@config.filename).must_equal("traq.yml")
+      end
+
+      it "should have a default file method" do
+         @config.must_respond_to(:default)
+      end
+   end
+
+   describe "configs" do
+      it "should respond to configuration dynamic methods" do
+         @config.banana.must_equal ""
+      end
+
+      it "should respond with the correct value" do
+         @config.load
+         @config.user.must_equal "taq"
+      end
+
+      it "should have basic information on the default file" do
+         contents = YAML.load(File.read(@config.default))
+         contents["directory"].wont_be_nil
+         contents["server"].wont_be_nil
+         contents["list"].wont_be_nil
+         contents["host"].wont_be_nil
+      end
+   end
+
+   describe "setup" do
+      it "should have a method named setup" do
+         @config.must_respond_to(:setup)
+      end
+
+      it "should do nothing if the configuration file exists" do
+         @config.setup.must_equal false
+      end
+
+      it "should write the configuration file if it doesn't exists" do
+         contents = File.read(@config.filename)
+         File.unlink(@config.filename)
+
+         assert !File.exists?(@config.filename)
+         @config.setup.must_equal true
+         assert File.exists?(@config.filename)
+
+         File.open(@config.filename,"w") {|file| file << contents}
+      end
+   end
+end

data/spec/deploy_spec.rb

@@ -0,0 +1,11 @@
+require "yaml"
+require "fileutils"
+require "minitest/autorun"
+require "#{File.expand_path(File.dirname(__FILE__))}/../lib/traquitana.rb"
+
+describe Traquitana::Deployer do
+   before do
+      @config = Traquitana::Config.instance
+      @deploy = Traquitana::Deployer.new 
+   end
+end

data/spec/migrator_spec.rb

@@ -0,0 +1,75 @@
+require "minitest/autorun"
+require "#{File.expand_path(File.dirname(__FILE__))}/../lib/traquitana.rb"
+
+describe Traquitana::Migrator do
+   before do
+      @migrator = Traquitana::Migrator.new
+
+      @old_file = "#{File.expand_path(File.dirname(__FILE__))}/../traq/config.yml"
+      @old_cont = File.read(@old_file)
+
+      @new_file = "#{File.expand_path(File.dirname(__FILE__))}/../config/traq.yml"
+      @new_cont = File.read(@new_file)
+
+      File.unlink(@new_file)
+   end
+
+   after do
+      FileUtils.mkdir(File.dirname(@old_file)) if !Dir.exists?(File.dirname(@old_file))
+      FileUtils.mkdir(File.dirname(@new_file)) if !Dir.exists?(File.dirname(@new_file))
+
+      File.open(@new_file,"w") {|f| f<<@new_cont}
+      File.open(@old_file,"w") {|f| f<<@old_cont}
+   end
+
+   it "should have a run method" do
+      @migrator.must_respond_to(:run)
+   end
+   
+   it "should return false when there isn't an old config file" do
+      contents = File.read(@old_file)
+      File.unlink(@old_file)
+      assert !@migrator.run
+      File.open(@old_file,"w") {|f| f<<contents}         
+   end
+
+   it "should return false if there is a new config file" do
+      File.open(@new_file,"w") {|f| f<<@new_cont}
+      assert !@migrator.run
+   end
+
+   it "should return true when there is an old config file" do
+      assert @migrator.run
+   end
+
+   it "should have the new file after run" do
+      File.unlink(@new_file) if File.exists?(@new_file)
+      @migrator.run
+      assert File.exists?(@new_file)
+   end
+
+   it "should not have the old file after run" do
+      @migrator.run
+      assert !File.exists?(@old_file)
+   end
+
+   it "should not have the old dir after run" do
+      @migrator.run
+      assert !Dir.exists?(File.dirname(@old_file))
+   end
+
+   it "should not have the ignore clause on the new file" do
+      @migrator.run
+      assert !YAML.load(File.read(@new_file)).include?("ignore")
+   end
+
+   it "should have all the keys of the old file on the new one" do
+      old_content = YAML.load(File.read(@old_file)).reject {|k,v| k==:ignore}
+      @migrator.run
+      new_content = YAML.load(File.read(@new_file))
+      old_content.each do |key,val|
+         assert new_content.include?(key.to_s)
+         assert new_content[key.to_s].to_s==old_content[key].to_s
+      end
+   end
+end