coral_core 0.1.2

data/lib/coral_core/util/data.rb

@@ -0,0 +1,59 @@
+
+module Coral
+module Util
+class Data < Core
+   
+  #-----------------------------------------------------------------------------
+  # Utilities
+    
+  def self.merge(objects, merge_arrays = false)
+    value = nil
+    return value unless objects
+    
+    unless objects.is_a?(Array)
+      objects = [ objects ]
+    end
+            
+    objects.each do |object|
+      value = recursive_merge(value, object, merge_arrays)  
+    end
+    return value
+  end
+    
+  #---
+    
+  def self.recursive_merge(overrides, data, merge_arrays = false)
+    return data unless overrides
+    return overrides unless data
+            
+    if overrides.is_a?(Hash)
+      overrides.each do |name, override|
+        if data.is_a?(Hash)
+          if data[name]
+            data[name] = recursive_merge(override, data[name])
+          else
+            begin
+              item = override.dup
+            rescue TypeError
+              item = override
+            end
+            data[name] = recursive_merge(override, item)
+          end
+        else
+          data = overrides
+        end
+      end
+    elsif merge_arrays && overrides.is_a?(Array)
+      if data.is_a?(Array)
+        data = data | overrides
+      else
+        data = overrides
+      end
+    else
+      data = overrides
+    end
+    return data   
+  end
+end
+end
+end

data/lib/coral_core/util/disk.rb

@@ -0,0 +1,82 @@
+
+module Coral
+module Util
+class Disk < Core
+  
+  #-----------------------------------------------------------------------------
+  # Properties
+ 
+  @@files = {}
+  
+  #-----------------------------------------------------------------------------
+  # Utilities
+  
+  def self.open(file_name, options = {}, reset = false)
+    mode        = string(options[:mode])
+    separator   = ( options[:separator] ? options[:separator] : false )
+    description = ( options[:description] ? options[:description] : '' )
+    
+    if @@files.has_key?(file_name) && ! reset
+      reset = true if ! mode.empty? && mode != @@files[file_name][:mode]
+    end
+    
+    if ! @@files.has_key?(file_name) || ! @@files[file_name][:file] || reset
+      @@files[file_name][:file].close if @@files[file_name][:file]
+      unless mode.empty?
+        @@files[file_name] = {
+          :file => File.open(file_name, mode),
+          :mode => mode,
+        }
+      end
+    end
+    return @@files[file_name][:file]
+  end
+  
+  #---
+  
+  def self.read(file_name, options = {})
+    options[:mode] = ( options[:mode] ? options[:mode] : 'r' )
+    file           = open(file_name, options)
+    
+    if file
+      return file.read
+    end
+    return nil
+  end
+  
+  #---
+  
+  def self.write(file_name, data, options = {})
+    options[:mode] = ( options[:mode] ? options[:mode] : 'w' )
+    file           = open(file_name, options)
+    
+    if file
+      return file.write(data)
+    end
+    return nil
+  end
+  
+  #---
+  
+  def self.log(data, options = {})
+    reset = ( options[:file_name] || options[:mode] )
+    file  = open(( options[:file_name] ? options[:file_name] : 'log.txt' ), options, reset)    
+    if file      
+      file.write("--------------------------------------\n") if separator
+      file.write("#{description}\n") if description        
+      file.write("#{data}\n")
+    end
+  end
+  
+  #---
+  
+  def self.close(file_names = [])
+    file_names = @@files.keys unless file_names && ! file_names.empty?
+    array(file_names).each do |file_name|
+      @@files[file_name][:file].close if @@files[file_name][:file]
+      @@files.delete(file_name)
+    end
+  end
+end
+end
+end

data/lib/coral_core/util/git/base.rb

@@ -0,0 +1,58 @@
+
+module Git
+
+#*******************************************************************************
+# Errors
+  
+class GitDirectoryError < StandardError 
+end
+
+#*******************************************************************************
+# Base Git definition
+
+class Base
+ 
+  #-----------------------------------------------------------------------------
+  # Constructor / Destructor
+  
+  def initialize(options = {})
+    if working_dir = options[:working_directory]
+      options[:repository] ||= File.join(working_dir, '.git')
+      
+      if File.file?(options[:repository])
+        File.read(options[:repository]).each_line do |line|
+          matches = line.match(/^\s*gitdir:\s*(.+)\s*/)
+          if matches.length && matches[1]
+            options[:repository] = matches[1]
+            break
+          end
+        end        
+      end
+      
+      if File.directory?(options[:repository])
+        options[:index] ||= File.join(options[:repository], 'index')
+      else
+        raise GitDirectoryError.new("Git repository directory #{options[:repository]} not found for #{working_dir}")
+      end
+    end
+    
+    if options[:log]
+      @logger = options[:log]
+      @logger.info("Starting Git")
+    else
+      @logger = nil
+    end
+     
+    @working_directory = options[:working_directory] ? Git::WorkingDirectory.new(options[:working_directory]) : nil
+    @repository        = options[:repository] ? Git::Repository.new(options[:repository]) : nil 
+    @index             = options[:index] ? Git::Index.new(options[:index], false) : nil
+  end
+    
+  #-----------------------------------------------------------------------------
+  # Commit extensions
+
+  def add(path = '.', opts = {})
+    self.lib.add(path, opts)
+  end
+end
+end

data/lib/coral_core/util/git/lib.rb

@@ -0,0 +1,82 @@
+
+module Git
+class Lib
+
+  #-----------------------------------------------------------------------------
+  # Commit extensions
+  
+  def add(path = '.', opts = {})
+    arr_opts = []
+    arr_opts << '-u' if opts[:update]
+    if path.is_a?(Array)
+      arr_opts += path
+    else
+      arr_opts << path
+    end
+    command('add', arr_opts)
+  end
+  
+  #---
+  
+  def commit(message, opts = {})
+    arr_opts = ['-m', message]
+    arr_opts << "--author=\'#{opts[:author]}\'" unless opts[:author] && opts[:author].empty?
+    arr_opts << '-a' if opts[:add_all]
+    arr_opts << '--allow-empty' if opts[:allow_empty]      
+    command('commit', arr_opts)
+  end
+  
+  #-----------------------------------------------------------------------------
+  # Remote extensions
+  
+  def remote_add(name, url, opts = {})
+    arr_opts = ['add']
+    arr_opts << '-f' if opts[:with_fetch]
+    arr_opts << name
+    arr_opts << url
+      
+    command('remote', arr_opts)
+  end
+  
+  #---
+
+  def remote_set_url(name, url, opts = {})
+    arr_opts = ['set-url']
+      
+    if opts[:add]
+      arr_opts << '--add' if opts[:add]
+    end
+      
+    if opts[:delete]
+      arr_opts << '--delete' if opts[:delete]
+    end
+      
+    if opts[:push]
+      arr_opts << '--push' if opts[:push]
+    end
+        
+    arr_opts << name
+    arr_opts << url
+      
+    command('remote', arr_opts)
+  end
+  
+  #---
+      
+  def remote_remove(name)
+    command('remote', ['rm', name])
+  end
+  
+  #-----------------------------------------------------------------------------
+  # Utilities
+  
+  def escape(s)
+    escaped = s.to_s.gsub('"', '\'')
+    if escaped =~ /^\-+/
+      escaped  
+    else
+      %Q{"#{escaped}"}
+    end
+  end
+end
+end

data/lib/coral_core/util/git/remote.rb

@@ -0,0 +1,12 @@
+
+module Git
+class Remote
+  
+  #-----------------------------------------------------------------------------
+  # Remote endpoints
+
+  def set_url(url, opts = {})
+    @base.lib.remote_set_url(@name, url, opts)  
+  end
+end
+end

data/lib/coral_core/util/git.rb

@@ -0,0 +1,15 @@
+
+require 'git'
+
+module Git
+
+  #-----------------------------------------------------------------------------
+  # Utilities
+
+  def self.url(host, repo, options = {})
+    options[:user] = ( options[:user] ? options[:user] : 'git' )
+    options[:auth] = ( options[:auth] ? options[:auth] : true )
+    
+    return options[:user] + ( options[:auth] ? '@' : '://' ) + host + ( options[:auth] ? ':' : '/' ) + repo
+  end
+end

data/lib/coral_core/util/shell.rb

@@ -0,0 +1,178 @@
+
+module Coral
+module Util
+class Shell < Core
+  
+  #-----------------------------------------------------------------------------
+  # Utilities
+  
+  def self.exec!(command, options = {})
+    min          = ( options[:min] ? options[:min].to_i : 1 )
+    tries        = ( options[:tries] ? options[:tries].to_i : min )
+    tries        = ( min > tries ? min : tries )
+    
+    info_prefix  = ( options[:info_prefix] ? options[:info_prefix] : '' )
+    info_suffix  = ( options[:info_suffix] ? options[:info_suffix] : '' )
+    error_prefix = ( options[:error_prefix] ? options[:error_prefix] : '' )
+    error_suffix = ( options[:error_suffix] ? options[:error_suffix] : '' )
+    
+    conditions   = Coral::Event.instance(options[:exit], true)
+    
+    $stdout.sync = true
+    $stderr.sync = true  
+    
+    for i in tries.downto(1)
+      ui.info(">> running: #{command}")
+      
+      begin
+        t1, output_new, output_orig = pipe_exec_stream!($stdout, conditions, { 
+          :prefix => info_prefix, 
+          :suffix => info_suffix, 
+        }) do |line|
+          block_given? ? yield(line) : true
+        end
+      
+        t2, error_new, error_orig = pipe_exec_stream!($stderr, conditions, { 
+          :prefix => error_prefix, 
+          :suffix => error_suffix, 
+        }) do |line|
+          block_given? ? yield(line) : true
+        end
+      
+        system_success = system(command)
+      
+      ensure
+        output_success = close_exec_pipe(t1, $stdout, output_orig, output_new)
+        error_success  = close_exec_pipe(t2, $stderr, error_orig, error_new)
+      end
+      ui.info('')
+      
+      success = ( system_success && output_success && error_success )
+      
+      min -= 1
+      break if success && min <= 0 && conditions.empty?
+    end
+    unless conditions.empty?
+      success = false
+    end
+    
+    return success   
+  end
+  
+  #---
+  
+  def self.exec(command, options = {})
+    return exec!(command, options)
+  end
+  
+  #---
+  
+  def self.pipe_exec_stream!(output, conditions, options)
+    original     = output.dup
+    read, write  = IO.pipe
+    
+    match_prefix = ( options[:match_prefix] ? options[:match_prefix] : 'EXIT' )
+            
+    thread = process_stream!(read, original, options) do |line|
+      check_conditions!(line, conditions, match_prefix) do
+        block_given? ? yield(line) : true
+      end
+    end
+    
+    thread.abort_on_exception = false
+    
+    output.reopen(write)    
+    return thread, write, original
+  end
+  
+  #---
+  
+  def self.close_exec_pipe(thread, output, original, write)
+    output.reopen(original)
+    
+    write.close
+    success = thread.value
+    Thread.kill(thread) # Die bastard, die!
+    
+    original.close
+    return success
+  end
+  
+  #---
+  
+  def self.check_conditions!(line, conditions, match_prefix = '')
+    prefix = ''
+    
+    unless ! conditions || conditions.empty?
+      conditions.each do |key, event|
+        if event.check(line)
+          prefix = match_prefix
+          conditions.delete(key)
+        end
+      end
+    end
+    
+    result = true
+    if block_given?
+      result = yield
+      
+      unless prefix.empty?
+        case result
+        when Hash
+          result[:prefix] = prefix
+        else
+          result = { :success => result, :prefix => prefix }
+        end
+      end
+    end
+    return result
+  end
+  
+  #---
+  
+  def self.process_stream!(input, output, options)
+    return Thread.new do
+      success        = true      
+      default_prefix = ( options[:prefix] ? options[:prefix] : '' )
+      default_suffix = ( options[:suffix] ? options[:suffix] : '' )
+      
+      begin
+        while ( data = input.readpartial(1024) )
+          message = data.strip
+          newline = ( data[-1,1].match(/\n/) ? true : false )
+                                 
+          unless message.empty?
+            lines = message.split(/\n/)
+            lines.each_with_index do |line, index|
+              prefix  = default_prefix
+              suffix  = default_suffix
+              
+              if block_given?
+                result = yield(line)
+                                          
+                if result && result.is_a?(Hash)
+                  prefix = result[:prefix]
+                  suffix = result[:suffix]
+                  result = result[:success]                 
+                end
+                success = result if success
+              end
+            
+              prefix = ( prefix && ! prefix.empty? ? "#{prefix}: " : '' )
+              suffix = ( suffix && ! suffix.empty? ? suffix : '' )            
+              eol    = ( index < lines.length - 1 || newline ? "\n" : ' ' )
+            
+              output.write(prefix.lstrip + line + suffix.rstrip + eol)
+            end
+          end
+        end
+      rescue EOFError
+      end
+      
+      input.close
+      success
+    end
+  end
+end
+end
+end

data/lib/coral_core.rb

@@ -0,0 +1,49 @@
+
+home = File.dirname(__FILE__)
+
+$:.unshift(home) unless
+  $:.include?(home) || $:.include?(File.expand_path(home))
+  
+#-------------------------------------------------------------------------------
+  
+require 'rubygems'
+
+#---
+
+# Include core
+[ :interface, :core ].each do |name| 
+  require File.join('coral_core', name.to_s + ".rb") 
+end
+
+# Include utilities
+[ :git, :data, :disk, :shell ].each do |name| 
+  require File.join('coral_core', 'util', name.to_s + ".rb") 
+end
+
+# Include Git overrides
+Dir.glob(File.join(home, 'coral_core', 'util', 'git', '*.rb')).each do |file|
+  require file
+end
+
+# Include data model
+[ :event, :command, :repository, :memory ].each do |name| 
+  require File.join('coral_core', name.to_s + ".rb") 
+end
+
+# Include specialized events
+Dir.glob(File.join(home, 'coral_core', 'event', '*.rb')).each do |file|
+  require file
+end
+
+#*******************************************************************************
+# Coral Core Library
+#
+# This provides core data elements and utilities used in the Coral gems.
+#
+# Author::    Adrian Webb (mailto:adrian.webb@coraltech.net)
+# License::   GPLv3
+module Coral
+  
+  VERSION = File.read(File.join(File.dirname(__FILE__), '..', 'VERSION'))
+ 
+end