bugzyrb 0.1.0

data/lib/common/cmdapp.rb

@@ -0,0 +1,280 @@
+#!/usr/bin/env ruby -w
+=begin
+  * Name          : cmdapp.rb
+  * Description   : some basic command line things
+  *               : Moving some methods from todorb.rb here
+  * Author        : rkumar
+  * Date          : 2010-06-20 11:18 
+  * License:       Ruby License
+
+=end
+require 'common/sed'
+
+ERRCODE = 1
+
+module Cmdapp
+
+  ## 
+  # external dependencies:
+  #  @app_default_action - action to run if none specified
+  #  @app_file_path - data file we are backing up, or reading into array
+  #  @app_serial_path - serial_number file
+  ##
+  # check whether this action is mapped to some alias and *changes*
+  # variables@action and @argv if true.
+  # @param [String] action asked by user
+  # @param [Array] rest of args on command line
+  # @return [Boolean] whether it is mapped or not.
+  #
+  def check_aliases action, args
+    return false unless @aliases
+    ret = @aliases[action]
+    if ret
+      a = ret.shift
+      b = [*ret, *args]
+      @action = a
+      @argv = b
+      #puts " #{@action} ; argv: #{@argv} "
+      return true
+    end
+    return false
+  end
+  ## 
+  # runs method after checking if valid or alias.
+  # If not found prints help.
+  # @return [0, ERRCODE] success 0.
+  def run
+    @action = @argv[0] || @app_default_action
+    @action = @action.downcase
+
+
+    ret = 0
+    @argv.shift
+    if respond_to? @action
+      ret = send(@action, @argv)
+    else
+      # check aliases
+      if check_aliases @action, @argv
+        ret = send(@action, @argv)
+      else
+        help @argv
+        ret = ERRCODE
+      end
+    end
+    ret ||= 0
+    ret = 0 if ret != ERRCODE
+    return ret
+  end
+  # not required if using Subcommand
+  def help args
+    if @actions.nil? 
+      if defined? @commands
+        unless @commands.empty?
+          @actions = @commands
+        end
+      end
+    end
+    if @actions
+      puts "Actions are "
+      @actions.each_pair { |name, val| puts "#{name}\t#{val}" }
+    end
+    puts " "
+    if @aliases
+      puts "Aliases are "
+      @aliases.each_pair { |name, val| puts "#{name}:\t#{val.join(' ')}" }
+    end
+    0
+  end
+  ## 
+  def alias_command name, *args
+    @aliases ||= {}
+    @aliases[name.to_s] = args
+  end
+  def add_action name, descr
+    @actions ||= {}
+    @actions[name.to_s] = desc
+  end
+
+  ##
+  # reads serial_number file, returns serialno for this app
+  # and increments the serial number and writes back.
+  def _get_serial_number
+    require 'fileutils'
+    appname = @appname
+    filename = @app_serial_path || "serial_numbers"
+    h = {}
+    # check if serial file existing in curr dir. Else create
+    if File.exists?(filename)
+      File.open(filename).each { |line|
+        #sn = $1 if line.match regex
+        x = line.split ":"
+        h[x[0]] = x[1].chomp
+      }
+    end
+    sn = h[appname] || 1
+    # update the sn in file
+    nsn = sn.to_i + 1
+    # this will create if not exists in addition to storing if it does
+    h[appname] = nsn
+    # write back to file
+    File.open(filename, "w") do |f|
+      h.each_pair {|k,v| f.print "#{k}:#{v}\n"}
+    end
+    return sn
+  end
+  ##
+  # After doing a redo of the numbering, we need to reset the numbers for that app
+  def _set_serial_number number
+    appname = @appname
+    pattern = Regexp.new "^#{appname}:.*$"
+    filename = @app_serial_path || "serial_numbers"
+    # during testing redo this file does not exist, so i get errors
+    if !File.exists? filename
+      _get_serial_number
+    end
+    _backup filename
+    change_row filename, pattern, "#{appname}:#{number}"
+  end
+
+  def _backup filename=@app_file_path
+    require 'fileutils'
+    FileUtils.cp filename, "#{filename}.org"
+  end
+  def die text
+    $stderr.puts text
+    exit ERRCODE
+  end
+  # prints messages to stderr
+  # All messages should go to stderr.
+  # Keep stdout only for output which can be used by other programs
+  def message text
+    $stderr.puts text
+  end
+  # print to stderr only if verbose set
+  def verbose text
+    message(text) if @options[:verbose]
+  end
+  # print to stderr only if verbose set
+  def warning text
+    print_red("WARNING: #{text}") 
+  end
+  def print_red text
+    message "#{RED}#{text}#{CLEAR}"
+  end
+  def print_green text
+    message "#{GREEN}#{text}#{CLEAR}"
+  end
+
+  ##
+  # load data into array as item and task
+  # @see save_array to write
+  def load_array
+    #return if $valid_array
+    $valid_array = false
+    @data = []
+    File.open(@app_file_path).each do |line|
+      # FIXME: use @app_delim
+      row = line.chomp.split "\t"
+      @data << row
+    end
+    $valid_array = true
+  end
+  ## 
+  # saves the task array to disk
+  # Please use load_array to load, and not populate
+  def save_array
+    raise "Cannot save array! Please use load_array to load" if $valid_array == false
+
+    File.open(@app_file_path, "w") do |file| 
+      # FIXME: use join with @app_delim
+      @data.each { |row| file.puts "#{row[0]}\t#{row[1]}" }
+    end
+  end
+  ##
+  # retrieve version info updated by jeweler.
+  # Typically used by --version option of any command.
+  # @return [String, nil] version as string, or nil if file not found
+  def version_info
+    # thanks to Roger Pack on ruby-forum for how to get to the version
+    # file
+    filename = File.open(File.dirname(__FILE__) + "/../../VERSION")
+    v = nil
+    if File.exists?(filename)
+      v = File.open(filename).read.chomp if File.exists?(filename)
+    #else
+      #$stderr.puts "could not locate file #{filename}. " 
+      #puts `pwd`
+    end
+    v
+  end
+
+  # reads multiple lines of input until EOF (Ctrl-d)
+  # and returns as a string.
+  # Add newline after each line
+  # @return [String, nil] newline delimited string, or nil
+  def get_lines
+    lines = nil
+    #$stdin.flush
+    $stdin.each_line do |line|
+      line.chomp!
+      if line =~ /^bye$/
+        break
+      end
+      if lines
+        lines << "\n" + line
+      else
+        lines = line
+      end
+    end
+    return lines
+  end
+  
+
+# edits given text using EDITOR
+# @param [String] text to edit
+# @return [String, nil] edited string, or nil if no change
+def edit_text text
+  # 2010-06-29 10:24 
+  require 'fileutils'
+  require 'tempfile'
+  ed = ENV['EDITOR'] || "vim"
+  temp = Tempfile.new "tmp"
+  File.open(temp,"w"){ |f| f.write text }
+  mtime =  File.mtime(temp.path)
+  #system("#{ed} #{temp.path}")
+  system(ed, temp.path)
+
+  newmtime = File.mtime(temp.path)
+  newstr = nil
+  if mtime < newmtime
+    # check timestamp, if updated ..
+    #newstr = ""
+    #File.open(temp,"r"){ |f| f.each {|r| newstr << r } }
+    newstr = File.read(temp)
+    #puts "I got: #{newstr}"
+  else
+    #puts "user quit without saving"
+  end
+  return newstr
+end
+
+##
+# reads up template, and substirutes values from myhash
+# @param [String] template text
+# @param [Hash] values to replace in template
+# @return [String] template output
+# NOTE: probably better to use rdoc/template which can handle arrays as well.
+def template_replace template, myhash
+  #tmpltext=File::read(template);
+
+  t = template.dup
+  t.gsub!( /##(.*?)##/ ) {
+    #raise "Key '#{$1}' found in template but the value has not been set" unless ( myhash.has_key?( $1 ) )
+    myhash[ $1 ].to_s
+  }
+  t
+end
+
+
+
+end

data/lib/common/colorconstants.rb

@@ -0,0 +1,79 @@
+#!/usr/bin/env ruby 
+
+# Color constants so we can print onto console using print or puts
+# @example 
+# string = "hello ruby"
+# puts " #{RED}#{BOLD}#{UNDERLINE}#{string}#{CLEAR}"
+#
+# http://wiki.bash-hackers.org/scripting/terminalcodes
+# ripped off highline gem
+module ColorConstants
+
+  # Embed in a String to clear all previous ANSI sequences.  This *MUST* be 
+  # done before the program exits!
+  # 
+  CLEAR      = "\e[0m"
+  # An alias for CLEAR.
+  RESET      = CLEAR
+  NORMAL      = CLEAR
+  # Erase the current line of terminal output.
+  ERASE_LINE = "\e[K"
+  # Erase the character under the cursor.
+  ERASE_CHAR = "\e[P"
+  # The start of an ANSI bold sequence.
+  BOLD       = "\e[1m"
+  # The start of an ANSI dark sequence.  (Terminal support uncommon.)
+  DARK       = "\e[2m"
+  DIM        = DARK
+  # The start of an ANSI underline sequence.
+  UNDERLINE  = "\e[4m"
+  # An alias for UNDERLINE.
+  UNDERSCORE = UNDERLINE
+  # The start of an ANSI blink sequence.  (Terminal support uncommon.)
+  BLINK      = "\e[5m"
+  # The start of an ANSI reverse sequence.
+  REVERSE    = "\e[7m"
+  # The start of an ANSI concealed sequence.  (Terminal support uncommon.)
+  CONCEALED  = "\e[8m"
+
+  # added from http://understudy.net/custom.html
+  BOLD_OFF       = "\e[22m"
+  UNDERILNE_OFF  = "\e[24m"
+  BLINK_OFF      = "\e[25m"
+  REVERSE_OFF    = "\e[27m"
+
+  # Set the terminal's foreground ANSI color to black.
+  BLACK      = "\e[30m"
+  # Set the terminal's foreground ANSI color to red.
+  RED        = "\e[31m"
+  # Set the terminal's foreground ANSI color to green.
+  GREEN      = "\e[32m"
+  # Set the terminal's foreground ANSI color to yellow.
+  YELLOW     = "\e[33m"
+  # Set the terminal's foreground ANSI color to blue.
+  BLUE       = "\e[34m"
+  # Set the terminal's foreground ANSI color to magenta.
+  MAGENTA    = "\e[35m"
+  # Set the terminal's foreground ANSI color to cyan.
+  CYAN       = "\e[36m"
+  # Set the terminal's foreground ANSI color to white.
+  WHITE      = "\e[37m"
+
+  # Set the terminal's background ANSI color to black.
+  ON_BLACK   = "\e[40m"
+  # Set the terminal's background ANSI color to red.
+  ON_RED     = "\e[41m"
+  # Set the terminal's background ANSI color to green.
+  ON_GREEN   = "\e[42m"
+  # Set the terminal's background ANSI color to yellow.
+  ON_YELLOW  = "\e[43m"
+  # Set the terminal's background ANSI color to blue.
+  ON_BLUE    = "\e[44m"
+  # Set the terminal's background ANSI color to magenta.
+  ON_MAGENTA = "\e[45m"
+  # Set the terminal's background ANSI color to cyan.
+  ON_CYAN    = "\e[46m"
+  # Set the terminal's background ANSI color to white.
+  ON_WHITE   = "\e[47m"
+
+end

data/lib/common/db.rb

@@ -0,0 +1,249 @@
+#!/usr/bin/env ruby -w
+require 'rubygems'
+require 'sqlite3'
+require 'pp'
+require 'arrayfields'
+module Database
+  class DB
+    attr_accessor :db
+
+    def initialize dbname="bugzy.sqlite"
+      raise "#{dbname} does not exist. Try --help" unless File.exists? dbname
+      @db = SQLite3::Database.new(dbname)
+      $now = Time.now
+      $num = rand(100)
+      $default_user = ENV['LOGNAME'] || ENV['USER']
+    end
+
+    # returns many rows 
+    # @param [Fixnum] id for table with 1:n rows
+    # @return [Array, nil] array if rows, else nil
+    def select table
+      #puts " --- #{table} ---  "
+      @db.type_translation = true
+      rows = []
+      @db.execute( "select * from #{table} " ) do |row|
+        if block_given?
+          yield row
+        else
+          rows << row
+        end
+      end
+      return nil if rows.empty?
+      return rows
+    end
+    # returns many rows 
+    # @param [String] sql statement
+    # @return [Array, nil] array if rows, else nil
+    def run text
+      #puts " --- #{text} ---  "
+      @db.type_translation = true
+      rows = []
+      @db.execute( text ) do |row|
+        if block_given?
+          yield row
+        else
+          rows << row
+        end
+      end
+      return nil if rows.empty?
+      return rows
+    end
+    def select_where table, *wherecond
+      #puts " --- #{table} --- #{wherecond} "
+      @db.type_translation = true
+      wherestr = nil
+      rows = []
+      if wherecond and !wherecond.empty?
+        fields, values = separate_field_values wherecond
+        #wherestr = "" unless wherestr
+        wherestr = " where #{fields.join(" and ")} "
+        if wherestr
+          #puts " wherestr #{wherestr}, #{values} "
+          #stmt = @db.prepare("select * from #{table} #{wherestr} ", *values)
+          @db.execute( "select * from #{table} #{wherestr}", *values  ) do |row|
+            if block_given?
+              yield row
+            else
+              rows << row
+            end
+          end
+        end
+      end
+      return nil if rows.empty?
+      return rows
+    end
+  ## 
+  # insert a issue or bug report into the database
+  # @params
+  # @return [Fixnum] last row id
+  def bugs_insert(status, severity, type, assigned_to, start_date, due_date, comment_count, priority, title, description, fix, created_by = $default_user)
+    # id = $num
+    # status = "CODE" 
+    # severity = "CODE" 
+    # type = "CODE" 
+    # assigned_to = "CODE" 
+    # start_date = $now
+    # due_date = $now
+    # comment_count = $num
+    # priority = "CODE" 
+    # title = "CODE" 
+    # description = "Some long text" 
+    # fix = "Some long text" 
+    # date_created = $now
+    # date_modified = $now
+    @db.execute(" insert into bugs (  status, severity, type, assigned_to, start_date, due_date, comment_count, priority, title, description, fix, created_by ) values (  ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)",  
+                status, severity, type, assigned_to, start_date, due_date, comment_count, priority, title, description, fix, created_by )
+    rowid = @db.get_first_value( "select last_insert_rowid();")
+    return rowid
+  end
+  ## takes a hash and creates an insert statement for table and inserts data.
+  # Advantage is that as we add columns, this will add the column to the insert, so we
+  # don't need to keep modifying in many places.
+  # @param [String] name of table to insert data into
+  # @param [Hash] values to insert, keys must be table column names
+  # @return [Fixnum] newly inserted rowid
+  def table_insert_hash table, hash
+    str = "INSERT INTO #{table} ("
+    qstr = [] # question marks
+    fields = [] # field names
+    bind_vars = [] # values to insert
+    hash.each_pair { |name, val| 
+      fields << name
+      bind_vars << val
+      qstr << "?"
+    }
+    fstr = fields.join(",")
+    str << fstr
+    str << ") values ("
+    str << qstr.join(",")
+    str << ")"
+    puts str
+    @db.execute(str, *bind_vars)
+    rowid = @db.get_first_value( "select last_insert_rowid();")
+    return rowid
+  end
+  def max_bug_id
+    id = @db.get_first_value( "select max(id) from bugs;")
+    return id
+  end
+  def sql_comments_insert id, comment, created_by = $default_user
+    #date_created = date_cr | Time.now
+    @db.execute("insert into comments (id, comment, created_by) values (?,?,?)", id, comment, created_by ) 
+    rowid = @db.get_first_value( "select last_insert_rowid();")
+    return rowid
+  end
+  def sql_logs_insert id, field, log, created_by = $default_user
+    #date_created = date_cr | Time.now
+    @db.execute("insert into log (id, field, log, created_by) values (?,?,?,?)", id, field, log, created_by ) 
+  end
+  def sql_delete_bug id
+    message "deleting #{id}"
+    @db.execute( "delete from bugs where id = ?", id )
+    @db.execute( "delete from comments where id = ?", id )
+    @db.execute( "delete from log where id = ?", id )
+  end
+
+
+    ##
+    # update a row from bugs based on id, giving one fieldname and value
+    # @param [Fixnum] id unique key
+    # @param [String] fieldname 
+    # @param [String] value to update
+    # @example sql_bugs_update 9, :name, "Roger"
+    def sql_update table, id, field, value
+      @db.execute( "update #{table} set #{field} = ?, date_modified = ? where id = ?", value,$now, id)
+    end
+
+    # update a row from bugs based on id, can give multiple fieldnames and values
+    # @param [Fixnum] id unique key
+    # @return [Array] alternating fieldname and value
+    # @example sql_bugs_update_mult 9, :name, "Roger", :age, 29, :country, "SWI"
+    def sql_bugs_update_mult id, *fv
+      fields = []
+      values = []
+      fv << "date_modified"
+      fv << $now
+      fv.each_with_index do |f, i| 
+        if i % 2 == 0
+          fields << "#{f} = ?"
+        else
+          values << f
+        end
+      end
+
+      print( "update bugs set #{fields.join(" ,")} where id = ?", *values, id)
+      @db.execute( "update bugs set #{fields.join(" ,")} where id = ?", *values, id)
+    end
+    # 
+    # return a single row from table based on rowid
+    # @param [String] table name
+    # @param [Fixnum] rowid
+    # @return [Array] resultset (based on arrayfield)
+    def sql_select_rowid table, id
+      # @db.results_as_hash = true
+      return nil if id.nil? or table.nil?
+      row = @db.get_first_row( "select * from #{table} where rowid = ?", id )
+      return row
+    end
+
+    def separate_field_values array
+      fields = []
+      values = []
+      array.each_with_index do |f, i| 
+        if i % 2 == 0
+          fields << "#{f} = ?"
+        else
+          values << f
+        end
+      end
+      return fields, values
+    end
+    def dummy
+       id = $num
+       status = "OPEN" 
+       severity = "CRI" 
+       type = "BUG" 
+       assigned_to = "rahul" 
+       start_date = $now
+       due_date = $now
+       comment_count = 0
+       priority = "A" 
+       title = "some title" 
+       description = "Some long text fro this bug too" 
+       fix = nil #"Some long text" 
+       date_created = $now
+       date_modified = $now
+       created_by = $default_user
+       bugs_insert(status, severity, type, assigned_to, start_date, due_date, comment_count, priority, title, description, fix, created_by)
+      #bugs_insert(id, status, severity, type, assigned_to, start_date, due_date, comment_count, priority, title, description, fix, date_created, date_modified)
+    end
+
+  end # class
+end # module
+if __FILE__ == $PROGRAM_NAME
+  include Database
+  # some tests. change bugs with real name
+  db = DB.new
+  db.dummy
+  puts  "\n------------ all row -----\n"
+  #db.select "bugs" do |r|
+    #puts r
+    ##puts r.join(" | ")
+  #end
+  res = db.select "bugs" 
+  if res
+    puts "count: #{res.count}"
+    #puts res.public_methods
+    res.each do |e| 
+      puts e.join(" | ")
+    end
+  end
+  db.sql_update "bugs", 1, "fix", "A fix added at #{$now}"
+  db.sql_bugs_update_mult 1, "title", "a new title #{$num}", "description", "a new description #{$num}"
+  puts  "\n------------ one row -----\n "
+  db.select_where "bugs", "id", 1 do |r|
+    puts r.join(" | ")
+  end
+
+end # if