mybot 0.0.1

data/.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/Botfile

@@ -0,0 +1,59 @@
+require "mybot"
+require "fileutils"
+
+include Messages
+
+namespace :mybot do
+  task :install do
+    if File.exists? Mybot::MYBOT_PATH
+      fatal "mybot is already installed"
+    end
+
+    info "installing mybot to #{Mybot::MYBOT_PATH}"
+    FileUtils.mkdir Mybot::MYBOT_PATH
+    FileUtils.cp_r Mybot::VENDOR_PATH, Mybot::MYBOT_PATH
+  end
+
+  task :uninstall do
+    unless File.exists? Mybot::MYBOT_PATH
+      fatal "mybot is not installed"
+    end
+
+    info "uninstalling mybot"
+    if yes?("Do you really want to delete ~/.mybot")
+      FileUtils.rm_rf Mybot::MYBOT_PATH
+    else
+      info "uninstallation aborted"
+    end
+  end
+
+  task :update do
+    unless File.exists? Mybot::MYBOT_PATH
+      fatal "mybot is not installed"
+    end
+
+    info "updating mybot"
+    FileUtils.rm_rf File.join(LIB_PATH, "mybot")
+    FileUtils.mkdir File.join(LIB_PATH, "mybot")
+    FileUtils.cp_r File.join(VENDOR_PATH, "lib", "mybot"),
+      File.join(MYBOT_PATH, "lib")
+
+    FileUtils.rm_rf File.join(TPL_PATH, "mybot")
+    FileUtils.mkdir File.join(TPL_PATH, "mybot")
+    FileUtils.cp_r File.join(VENDOR_PATH, "tpl", "mybot"),
+      File.join(MYBOT_PATH, "tpl")
+  end
+
+  namespace :tasks do
+    task :list do
+      puts "available tasks:"
+      tasks.each do |t|
+        if t[:namespace]
+          puts "  #{t[:namespace]}:#{t[:name]}"
+        else
+          puts "  #{t[:name]}"
+        end
+      end
+    end
+  end
+end

data/Gemfile

@@ -0,0 +1,3 @@
+source :rubygems
+
+gemspec

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2013 Sebastian Sito
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,31 @@
+# Mybot
+
+## Installation
+
+	gem install mybot
+	bot install
+
+## Usage
+**Mybot** searches for tasks in following directories:
+
+  * `~/.mybot/tasks/`
+  * `.mybot/`
+  * and in `./Botfile`
+
+## Tasks
+This is example task
+
+	require "mybot"
+	
+	namespace :foo do
+  	  task :bar do
+	    puts "foo:bar"
+	  end
+	end
+
+and you can run it by typing `bot foo:bar`
+
+## Uninstallation
+
+	bot uninstall
+	gem uninstall -Iax mybot

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/bin/bot

@@ -0,0 +1,6 @@
+#!/usr/bin/env ruby
+
+require "rubygems" unless defined?(Gem)
+require "mybot"
+
+Mybot::Cli.start ARGV

data/lib/mybot.rb

@@ -0,0 +1,20 @@
+module Mybot
+  autoload :Base,     "mybot/base"
+  autoload :Cli,      "mybot/cli"
+  autoload :Command,  "mybot/command"
+  autoload :Commands, "mybot/commands"
+  autoload :Fmt,      "mybot/fmt"
+  autoload :Messages, "mybot/messages"
+  autoload :Node,     "mybot/node"
+  autoload :Recipes,  "mybot/recipes"
+  autoload :Tasks,    "mybot/tasks"
+  autoload :Template, "mybot/template"
+  autoload :VERSION,  "mybot/version"
+
+  GEM_PATH = File.dirname(File.dirname(__FILE__))
+  VENDOR_PATH = File.join(GEM_PATH, "vendor", ".")
+  HOME_PATH = File.expand_path "~"
+  MYBOT_PATH = File.join HOME_PATH, ".mybot"
+  LIB_PATH = File.join MYBOT_PATH, "lib"
+  TPL_PATH = File.join MYBOT_PATH, "tpl"
+end

data/lib/mybot/base.rb

@@ -0,0 +1,8 @@
+module Mybot
+  module Base
+    include Commands
+    include Recipes
+    include Tasks
+    extend self
+  end
+end

data/lib/mybot/cli.rb

@@ -0,0 +1,14 @@
+module Mybot
+  module Cli
+    extend self
+
+    def start(args)
+      unless args[0]
+        abort "usage: bot <task>"
+      end
+
+      Base.load_recipes
+      Base.run_task args[0]
+    end
+  end
+end

data/lib/mybot/command.rb

@@ -0,0 +1,43 @@
+module Mybot
+  class Command
+    attr_accessor :out, :stdout, :stderr, :channel, :exit_status
+
+    def initialize(node)
+      @node = node
+      @out = @stdout = @stderr = ""
+      @handlers = {}
+      @exit_status = 0
+    end
+
+    def on(s, &block)
+      @handlers[s] = block
+    end
+
+    def handle_stdout(data)
+      @out = data
+      @stdout += data.strip
+      @node.fmt.stdout data
+      @handlers.each do |s, proc|
+        proc.call if @out.include? s
+      end
+    end
+
+    def handle_stderr(data)
+      @out = data
+      @stderr += data.strip
+      @node.fmt.stderr data
+      @handlers.each do |s, proc|
+        proc.call if @out.include? s
+      end
+    end
+
+    def handle_exit
+      @stdout
+    end
+
+    def write(data)
+      @node.fmt.write data
+      @channel.send_data "#{data}\n"
+    end
+  end
+end

data/lib/mybot/commands.rb

@@ -0,0 +1,84 @@
+module Mybot
+  module Commands
+    include Messages
+
+    def node(host, user, options = {})
+      Node.new(host, user, options)
+    end
+
+    def namespace(name)
+      outer_ns, @namespace = @namespace, name.to_s
+      if outer_ns && outer_ns != ""
+        @namespace = "#{outer_ns}:#{@namespace}"
+      end
+
+      yield
+    ensure
+      @namespace = outer_ns
+    end
+
+    def task(name, &block)
+      name, deps = *case name
+      when Hash
+        name.shift
+      else
+        [name, []]
+      end
+
+      namespaced_deps = Array(deps).map do |d|
+        case d
+        when Symbol
+          "#{@namespace}:#{d}"
+        when String
+          if d.include? ":"
+            d
+          else
+            "#{@namespace}:#{d}"
+          end
+        end
+      end.uniq
+
+      task = tasks.find do |t|
+        t[:name] == name.to_s && t[:namespace] == @namespace
+      end
+
+      tasks.push({
+        :name => name.to_s,
+        :namespace => @namespace,
+        :deps => namespaced_deps,
+        :block => block
+      }) unless task
+    end
+
+    def tpl(content)
+      Template.new(content)
+    end
+
+    def tpl_file(file)
+      tpl = File.join(TPL_PATH, file)
+      fatal "template file '#{tpl}' does not exists" unless File.exists?(tpl)
+      Template.new(File.read(tpl))
+    end
+
+    def wait
+      $stdout.print "Press any key to continue..."
+      $stdin.gets
+    end
+
+    def ask(q)
+      $stdout.print "#{q}?: "
+      $stdin.gets.chomp
+    end
+
+    def yes?(q)
+      result = ""
+      loop do
+        $stdout.print "#{q} (y/n)?: "
+        result = $stdin.gets.chomp
+        break if result =~ /y|yes|Y|YES|Yes|n|no|N|NO|No/
+      end
+
+      result =~ /y|yes|Y|YES|Yes/
+    end
+  end
+end

data/lib/mybot/fmt.rb

@@ -0,0 +1,87 @@
+require "colored"
+
+module Mybot
+  class Fmt
+    attr_accessor :use_color, :show_cmd,
+      :show_stdout, :show_stderr, :show_stdin
+
+    def initialize(host, user)
+      @host, @user = host, user
+      @use_color = true
+      @show_cmd = true
+      @show_stdout = true
+      @show_stderr = true
+      @show_stdin = true
+    end
+
+    def run(cmd)
+      run_format = "RUN"
+      run_format = run_format.blue.bold if @use_color
+      $stdout.print "#{prompt} #{run_format} #{cmd}\n" if @show_cmd
+    end
+
+    def stdout(data)
+      data = data.green if @use_color
+      $stdout.print(data) if @show_stdout
+    end
+
+    def stderr(data)
+      data = data.red if @use_color
+      $stdout.print(data) if @show_stderr
+    end
+
+    def write(data)
+      write_format = "WRITE"
+      write_format = write_format.blue.bold if @use_color
+      enter_format = "<ENTER>"
+      enter_format = enter_format.red.bold if @use_color
+      if @show_stdin
+        $stdout.print "#{prompt} #{write_format} #{data}#{enter_format}\n"
+      end
+    end
+
+    def upload(from, to)
+      upload_format = "UPLOAD"
+      upload_format = upload_format.blue.bold if @use_color
+
+      if @show_cmd
+        $stdout.print "#{prompt} #{upload_format} #{from} -> #{to}\n"
+      end
+    end
+
+    def download(from, to)
+      download_format = "DOWNLOAD"
+      download_format = download_format.blue.bold if @use_color
+      
+      
+      if @show_cmd
+        $stdout.print "#{prompt} #{download_format} #{from} -> #{to}\n"
+      end
+    end
+
+    def progress(n)
+      if @show_cmd
+        $stdout.print "\r#{n}%"
+        puts if n == 100
+      end
+    end
+
+    def error(msg)
+      error_format = "ERROR"
+      error_format = error_format.bold.red if @use_color
+      $stdout.print "#{prompt} #{error_format} #{msg}\n"
+    end
+
+    def fatal(msg)
+      error(msg)
+      abort
+    end
+
+  private
+
+    def prompt
+      p ="[#{@user}@#{@host}]"
+      p = p.bold if @use_color
+    end
+  end
+end

data/lib/mybot/messages.rb

@@ -0,0 +1,20 @@
+module Mybot
+  module Messages
+    def info(msg)
+      puts "info: #{msg}"
+    end
+
+    def warning(msg)
+      puts "warning: #{msg}"
+    end
+
+    def error(msg)
+      puts "error: #{msg}"
+    end
+
+    def fatal(msg)
+      puts "fatal: #{msg}"
+      abort
+    end
+  end
+end

data/lib/mybot/node.rb

@@ -0,0 +1,124 @@
+require "net/ssh"
+require "net/sftp"
+
+module Mybot
+  class Node
+    attr_accessor :host, :user, :ssh, :sftp, :fmt
+
+    def initialize(host, user, options = {})
+      @host, @user, @options = host, user, options
+      @fmt = Fmt.new(@host, @user)
+    end
+
+    def ssh
+      @ssh ||= Net::SSH.start(@host, @user, @options)
+    end
+
+    def sftp
+      @sftp ||= Net::SFTP.start(@host, @user, @options)
+    end
+
+    def run(options, &block)
+      @fmt.fatal "cmd not specified" unless options[:cmd]
+
+      command = Command.new(self)
+      yield command if block_given?
+
+      ssh.open_channel do |ch|
+        ch.request_pty do |ch, ok|
+          @fmt.fatal "cannot request pty" unless ok
+        end
+
+        command.channel = ch
+        @fmt.run(options[:cmd])
+
+        ch.exec options[:cmd] do |ch, ok|
+          @fmt.fatal "run: cannot exec #{options[:cmd]}" unless ok
+
+          ch.on_data do |ch, data|
+            command.handle_stdout(data)
+          end
+
+          ch.on_extended_data do |ch, data|
+            command.handle_stderr(data)
+          end
+
+          ch.on_request("exit-status") do |ch, data|
+            command.exit_status = data.read_long
+          end
+
+          ch.on_close do |ch|
+            return command.handle_exit
+          end
+        end
+      end
+
+      ssh.loop
+      return command.stdout_all
+    end
+
+    def exists?(options)
+      @fmt.fatal "exists?: file not specified" unless options[:file]
+      
+      options[:cmd] = "test -e #{options[:file]} && echo EXISTS"
+      run(options).include?("EXISTS") ? true : false
+    end
+
+    def create(options)
+      @fmt.fatal "create: file not specified" unless options[:file]
+      
+      options[:cmd] = "mktemp /tmp/mybot.XXXXX"
+      tmp = run(options)
+
+      options[:content] ||= ""
+      sftp.file.open(tmp, "w") { |f| f.write options[:content] }
+      options[:cmd] = "mv #{tmp} #{options[:file]}"
+      run(options)
+    end
+
+    def read(options)
+      @fmt.fatal "read: file not specified" unless options[:file]
+
+      options[:cmd] = "cat #{options[:file]}"
+      run(options) do |cmd|
+        cmd.on "No such file or directory" do
+          @fmt.error "read: file '#{options[:file]}' does not exists"
+          return ""
+        end
+      end
+    end
+
+    def upload(options)
+      @fmt.fatal "upload: from not specified" unless options[:from]
+      @fmt.fatal "upload: to not specified" unless options[:to]
+      @fmt.upload(options[:from], options[:to])
+
+      sftp.upload!(options[:from], options[:to]) do |event, uploader, *args|
+        case event
+        when :put
+          n = (args[1].to_f * 100 / args[0].size.to_f).to_i
+          @fmt.progress(n)
+        when :finish
+          @fmt.progress(100)
+        end
+      end
+    end
+
+    def download(options)
+      @fmt.fatal "download: from not specified" unless options[:from]
+      @fmt.fatal "download: to not specified" unless options[:to]
+      @fmt.download(options[:from], options[:to])
+
+      sftp.download!(options[:from], options[:to]) do |event, uploader, *args|
+        case event
+        when :get
+          size = args[0].size ? args[0].size : sftp.stat!(options[:from]).size
+          n = (args[1].to_f * 100 / size.to_f).to_i
+          @fmt.progress(n)
+        when :finish
+          @fmt.progress(100)
+        end
+      end
+    end
+  end
+end

data/lib/mybot/recipes.rb

@@ -0,0 +1,23 @@
+module Mybot
+  module Recipes
+    def recipes
+      @recipes ||= (
+        Dir[File.join(GEM_PATH, "Botfile")] +
+        Dir[File.join(MYBOT_PATH, "tasks", "**", "*.rb")].flatten +
+        Dir[File.join(".mybot", "**", "*.rb")].flatten +
+        Dir["Botfile"]
+      ).reject { |f| !File.exists?(f) }
+    end
+
+    def load_recipes
+      recipes.each { |r| self.load_recipe(r) }
+    end
+
+    def load_recipe(r)
+      instance_eval(File.read(r), __FILE__, __LINE__)
+    rescue Exception => e
+      puts "error: cannot load recipe #{r}"
+      raise(e)
+    end
+  end
+end

data/lib/mybot/tasks.rb

@@ -0,0 +1,35 @@
+module Mybot
+  module Tasks
+    include Messages
+
+    def tasks
+      @tasks ||= []
+    end
+
+    def find_task(name)
+      ns = name.split ":"
+      task = ns.pop
+      ns = ns.join ":"
+
+      tasks.find do |t|
+        t[:name] == task && (t[:namespace] == ns || t[:namespace].nil?)
+      end
+    end
+
+    def run_task(name)
+      task = find_task(name)
+
+      unless task
+        error "cannot find task '#{name}'"
+        find_task("mybot:tasks:list")[:block].call
+        abort
+      end
+
+      task[:deps].each do |d|
+        run_task(d)
+      end
+
+      task[:block].call
+    end
+  end
+end

data/lib/mybot/template.rb

@@ -0,0 +1,14 @@
+require "ostruct"
+require "erb"
+
+module Mybot
+	class Template
+		def initialize(tpl)
+			@tpl = tpl
+		end
+
+		def render(params = {})
+			ERB.new(@tpl).result(OpenStruct.new(params).instance_eval { binding })
+		end
+	end
+end

data/lib/mybot/version.rb

@@ -0,0 +1,3 @@
+module Mybot
+  VERSION = "0.0.1"
+end

data/mybot.gemspec

@@ -0,0 +1,26 @@
+# encoding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'mybot/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "mybot"
+  spec.version       = Mybot::VERSION
+  spec.authors       = ["Sebastian Sito"]
+  spec.email         = ["sebastian@hypenode.com"]
+  spec.description   = %q{Provision, manage and monitor stuff}
+  spec.summary       = %q{Mybot, your personal bot}
+  spec.homepage      = "https://github.com/sebastiansito/mybot"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files`.split($/)
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency "thor"
+  spec.add_dependency "fog"
+  spec.add_dependency "colored"
+  spec.add_dependency "net-ssh"
+  spec.add_dependency "net-sftp"
+end

metadata

@@ -0,0 +1,150 @@
+--- !ruby/object:Gem::Specification
+name: mybot
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Sebastian Sito
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-04-12 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: fog
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: colored
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: net-ssh
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: net-sftp
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Provision, manage and monitor stuff
+email:
+- sebastian@hypenode.com
+executables:
+- bot
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Botfile
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/bot
+- lib/mybot.rb
+- lib/mybot/base.rb
+- lib/mybot/cli.rb
+- lib/mybot/command.rb
+- lib/mybot/commands.rb
+- lib/mybot/fmt.rb
+- lib/mybot/messages.rb
+- lib/mybot/node.rb
+- lib/mybot/recipes.rb
+- lib/mybot/tasks.rb
+- lib/mybot/template.rb
+- lib/mybot/version.rb
+- mybot.gemspec
+- vendor/lib/mybot/.gitkeep
+- vendor/tasks/.gitkeep
+- vendor/tpl/mybot/.gitkeep
+homepage: https://github.com/sebastiansito/mybot
+licenses:
+- MIT
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: Mybot, your personal bot
+test_files: []