csexton-captured 0.1.1 → 0.1.3

data/bin/captured

@@ -1,9 +1,7 @@
 #!/usr/bin/env ruby
 require 'rubygems'
 require 'optparse'
-
-$:.unshift File.join(File.dirname(__FILE__), '..', 'lib')
-require 'captured'
+require "#{File.dirname(__FILE__)}/../lib/captured"
 
 options = {:config_file => "#{ENV['HOME']}/.captured.yml",
            :watch_path => "#{ENV['HOME']}/Desktop/",

data/lib/captured/file_tracker.rb

@@ -0,0 +1,37 @@
+class FileTracker
+  attr_accessor :tracked_files
+
+  def initialize(options)
+    @options = options
+    @tracked_files = {}
+  end
+
+  def scan(paths, state)
+    puts "Scanning #{paths}"
+    paths.each do |path|
+      Dir["#{path}#{@options[:watch_pattern]}"].each do |file|
+        self.add file, state
+      end
+    end
+  end
+
+  def add(file, state)
+    unless @tracked_files[file]
+      puts "Adding #{file} to tracked files as #{state}"
+      @tracked_files[file] = state
+    end
+  end
+
+  def mark_processed(file)
+    puts "Marking #{file} processed"
+    @tracked_files[file] = :processed
+  end
+
+  def each_pending
+    @tracked_files.each_pair do |key, value|
+      if(value == :pending)
+        yield(key)
+      end
+    end
+  end
+end

data/lib/captured/file_uploader.rb

@@ -0,0 +1,70 @@
+require 'digest/md5'
+require 'erb'
+
+class FileUploader
+  def self.upload(file, options)
+    self.new(options).process_upload(file)
+  end
+
+  def initialize(options)
+    @growl_path = options[:growl_path] || "/usr/local/bin/growlnotify"
+    @config = YAML.load_file(options[:config_file])
+    case @config['upload']['type']
+    when"eval"
+      @upload_proc = eval_proc
+    when"scp"
+      @upload_proc = scp_proc
+    when"ftp"
+    else
+      raise "Invalid Type"
+    end
+  end
+
+  def eval_proc
+    lambda do |file, remote_name|
+      remote_path = nil
+      unless eval @config['upload']['command']
+        raise "Upload failed: Bad Eval in config file"
+      end
+      # if the eval defines remote_path we will copy that to the clipboard
+      # otherwise we compute it ouselves
+      remote_path || "#{@config['upload']['url']}#{remote_name}"
+    end
+  end
+
+  def scp_proc
+    require 'net/scp'
+    require 'etc'
+    settings = @config['upload']
+    lambda do |file, remote_name|
+      puts "host = '#{settings['host']}'"
+      puts "user = '#{settings['user'] || Etc.getlogin}'"
+      puts "file = '#{file}'"
+      puts "path = '#{settings['path']+remote_name}'"
+      Net::SCP.upload!(settings['host'], 
+                       settings['user'] || Etc.getlogin,
+                       file,
+                       settings['path']+remote_name)
+      "#{@config['upload']['url']}#{remote_name}"
+    end
+  end
+
+  def process_upload(file)
+    remote_name = Digest::MD5.hexdigest(file+Time.now.to_i.to_s) +  File.extname(file)
+    remote_path = @upload_proc.call(file, remote_name)
+    puts "Uploaded '#{file}' to '#{remote_path}'"
+    raise "Copy Failed" unless system("echo '#{remote_path}' | pbcopy")
+    growl("Uploaded Image", "#{File.dirname(File.expand_path(__FILE__))}/../../resources/green_check.png")
+  rescue => e 
+    puts e
+    puts e.backtrace
+    growl(e)
+  end
+
+  def growl(msg, image = "#{File.dirname(File.expand_path(__FILE__))}/../../resources/red_x.png")
+    puts "grr: #{msg}"
+    if File.exists? @growl_path
+      raise "Growl Failed" unless system("#{@growl_path} -t 'Captured' -m '#{msg}' --image '#{image}'")
+    end
+  end
+end

data/lib/captured/fs_events.rb

@@ -0,0 +1,54 @@
+require 'osx/foundation'
+OSX.require_framework '/System/Library/Frameworks/CoreServices.framework/Frameworks/CarbonCore.framework'
+
+class FSEvents
+  def initialize(dir = "#{HOME}/Desktop")
+    @watch_dir = dir
+  end
+
+  def run
+    callback = proc do |stream, ctx, numEvents, paths, marks, eventIDs|
+
+      #p stream
+      #p ctx
+      #p numEvents
+      #p paths.methods
+      #p marks.methods
+      #p eventIDs
+
+      paths.regard_as('*')
+      rpaths = []
+      numEvents.times { |i| rpaths << paths[i] }
+
+      yield(*rpaths)
+    end
+
+    allocator = OSX::KCFAllocatorDefault
+    context   = nil
+    path      = [@watch_dir] #[Dir.pwd]
+    sinceWhen = OSX::KFSEventStreamEventIdSinceNow
+    latency   = 1.0
+    flags     = 0
+
+    stream   = OSX::FSEventStreamCreate(allocator, callback, context, path, sinceWhen, latency, flags)
+    unless stream
+      puts "Failed to create stream"
+      exit
+    end
+
+    OSX::FSEventStreamScheduleWithRunLoop(stream, OSX::CFRunLoopGetCurrent(), OSX::KCFRunLoopDefaultMode)
+    unless OSX::FSEventStreamStart(stream)
+      puts "Failed to start stream"
+      exit
+    end
+
+    puts "Watching #{path}"
+
+    OSX::CFRunLoopRun()
+  rescue Interrupt
+    OSX::FSEventStreamStop(stream)
+    OSX::FSEventStreamInvalidate(stream)
+    OSX::FSEventStreamRelease(stream)
+  end
+
+end

data/lib/captured.rb

@@ -0,0 +1,33 @@
+require 'captured/file_tracker'
+require 'captured/file_uploader'
+require 'captured/fs_events'
+
+class Captured
+  def self.config_file
+    "#{ENV['HOME']}/.captured.yml" 
+  end
+
+  def self.run_once!(options)
+    watch_path = options[:watch_path] || "#{ENV['HOME']}/Desktop/"
+    Dir["#{watch_path}#{options[:watch_pattern]}"].each do |file|
+      if (File.mtime(file).to_i > (Time.now.to_i-10))
+        puts "#{file} is new"
+        FileUploader.upload(file, options)
+      end
+    end
+  end
+
+  def self.run_and_watch!(options)
+    watch_path = options[:watch_path] || "#{ENV['HOME']}/Desktop/"
+    tracker = FileTracker.new(options)
+    tracker.scan([desktop_dir], :existing)
+    e = FSEvents.new(watch_path)
+    e.run do |paths|
+      tracker.scan paths, :pending
+      tracker.each_pending do |file|
+        FileUploader.upload(file, options)
+        tracker.mark_processed(file)
+      end
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: csexton-captured
 version: !ruby/object:Gem::Version 
-  version: 0.1.1
+  version: 0.1.3
 platform: ruby
 authors: 
 - Christopher Sexton
@@ -23,10 +23,22 @@ extra_rdoc_files:
 - LICENSE
 - README.markdown
 files: 
+- README.markdown
+- bin/captured
 - etc/captured.yml-example
 - etc/launchd.plist.erb
+- lib/captured.rb
+- lib/captured/file_tracker.rb
+- lib/captured/file_uploader.rb
+- lib/captured/fs_events.rb
+- resources/captured.png
+- resources/green_check.png
+- resources/red_star.png
+- resources/red_x.png
+- resources/ruby.png
+- spec/captured_spec.rb
+- spec/spec_helper.rb
 - LICENSE
-- README.markdown
 has_rdoc: false
 homepage: http://github.com/csexton/captured
 post_install_message: