sbfaulkner-pygmalion 0.9.0

data/MIT-LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2008 unwwwired.net
+ 
+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/Rakefile

@@ -0,0 +1,10 @@
+require 'rubygems'
+Gem::manage_gems
+require 'rake/gempackagetask'
+
+load File.join(File.dirname(__FILE__),'pygmalion.gemspec')
+
+Rake::GemPackageTask.new(SPEC) do |pkg|
+end
+
+task :default => :package

data/bin/pygmalion

@@ -0,0 +1,40 @@
+#!/usr/bin/ruby
+
+require 'rubygems'
+require 'getoptlong'
+
+opts = GetoptLong.new(
+  [ "--help", "-h", GetoptLong::NO_ARGUMENT ],
+  [ "--language", "-l", GetoptLong::REQUIRED_ARGUMENT ],
+  [ "--line-numbers", "-n", GetoptLong::NO_ARGUMENT ]
+)
+
+OPTIONS = {
+  :code => nil,
+  :type => nil,
+  :css => 'n',
+  :linenos => 'off'
+}
+
+opts.each do |opt, arg|
+  case opt
+  when '--help'
+    puts "Usage: #{File.basename(ARGV[0])} [options] [file ...]"
+    puts "\nOptions:"
+    puts "    -l, --language=name     The language for the code to be highlighted."
+    puts "    -n, --line-numbers      Include line numbers."
+    exit 1
+  when '--language'
+    OPTIONS[:type] = arg
+  when '--line-numbers'
+    OPTIONS[:linenos] = 'on'
+  end
+end
+
+if ARGV.empty?
+  puts STDIN.read.highlight
+else
+  ARGV.each_with_index do |file,id|
+    puts File.read(file).highlight
+  end
+end

data/lib/pygmalion.rb

@@ -0,0 +1,80 @@
+require 'net/http'
+require 'uri'
+require 'active_support'
+
+module Pygmalion
+  PYGMENTS_HOST = 'pygments.com'
+  
+  JSON_PATH = "/json/"
+  PUSH_PATH = "/push_chunk/"
+
+  CHUNK_SIZE = 4000
+  CHUNK_REGEX = Regexp.new(".{1,#{CHUNK_SIZE}}", Regexp::MULTILINE)
+  
+  DEFAULT_OPTIONS = {
+    :css => 'n',
+    :id => 'pygmalion',
+    :linenos => 'off'
+  }
+
+  def self.included(base)
+    base.extend ClassMethods
+  end
+  
+  module ClassMethods
+    #
+    # build query string for pygments.com
+    #
+    def pygments_query(parameters = {}, code = nil, chunkid = nil)
+      parameters[:code] = code
+      parameters[:chunkid] = chunkid if chunkid
+      parameters = parameters.collect { |k,v| "#{k}=#{URI.escape(v.to_s)}" }.join('&')
+      query = [ chunkid ? PUSH_PATH : JSON_PATH ]
+      query << parameters unless parameters.empty?
+      query.join('?')
+    end
+  end
+  
+  #
+  # return a syntax-highlighted version of the string
+  #
+  # options:
+  #
+  # **line_numbers**:   boolean indicating whether or not to include line numbers
+  # **language**:       language to use for highlighting
+  #
+  def highlight(options = {})
+    # strip out options needing translation
+    line_numbers = options.delete(:line_numbers)
+    language = options.delete(:language)
+
+    # merge provided options into the defaults
+    options = DEFAULT_OPTIONS.merge(options)
+
+    # translate the aforementioned options
+    options[:linenos] = line_numbers ? 'on' : 'off'
+    options[:type] = language
+    
+    res = Net::HTTP::start(PYGMENTS_HOST) do |http|
+      if self.size <= CHUNK_SIZE
+        # if code is small enough, use a single request
+        http.get self.class.pygments_query(options, self)
+      else
+        # otherwise break into chunks and push to the server
+        headers = {}
+        self.scan(CHUNK_REGEX).each_with_index do |chunk, chunkid|
+          # TODO: try threads for potential speedup with multiple simultaneous requests if this proves too slow
+          response = http.get self.class.pygments_query(options, chunk, chunkid), headers
+          # the server uses a cookie-based session to deal with chunked requests
+          headers['cookie'] = response['set-cookie'] if response['set-cookie']
+        end
+        # get the final response once all chunks are processed
+        http.get self.class.pygments_query(options), headers
+      end
+    end
+    
+    ActiveSupport::JSON.decode(res.body[/\{.*\}/])['output']
+  end
+end
+
+String.send :include, Pygmalion

data/pygmalion.gemspec

@@ -0,0 +1,20 @@
+SPEC = Gem::Specification.new do |s| 
+  # identify the gem
+  s.name = "pygmalion" 
+  s.version = "0.9.0" 
+  s.author = "S. Brent Faulkner" 
+  s.email = "brentf@unwwwired.net" 
+  s.homepage = "http://www.unwwwired.net" 
+  # platform of choice
+  s.platform = Gem::Platform::RUBY 
+  # description of gem
+  s.summary = "A ruby implementation of an interactive SQL command-line for ODBC data sources" 
+  s.files = %w(bin/pygmalion lib/pygmalion.rb MIT-LICENSE Rakefile README.markdown pygmalion.gemspec)
+  s.require_path = "lib" 
+  s.autorequire = "pygmalion" 
+  # s.test_file = "test/pygmalion.rb" 
+  s.has_rdoc = true 
+  s.extra_rdoc_files = ["README.markdown"] 
+  s.add_dependency("fastercsv", ">= 1.2.3") 
+  s.executables = ["pygmalion"]
+end 

metadata

@@ -0,0 +1,66 @@
+--- !ruby/object:Gem::Specification 
+name: sbfaulkner-pygmalion
+version: !ruby/object:Gem::Version 
+  version: 0.9.0
+platform: ruby
+authors: 
+- S. Brent Faulkner
+autorequire: pygmalion
+bindir: bin
+cert_chain: []
+
+date: 2008-08-13 00:00:00 -07:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: fastercsv
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 1.2.3
+    version: 
+description: 
+email: brentf@unwwwired.net
+executables: 
+- pygmalion
+extensions: []
+
+extra_rdoc_files: 
+- README.markdown
+files: 
+- bin/pygmalion
+- lib/pygmalion.rb
+- MIT-LICENSE
+- Rakefile
+- README.markdown
+- pygmalion.gemspec
+has_rdoc: true
+homepage: http://www.unwwwired.net
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.2.0
+signing_key: 
+specification_version: 2
+summary: A ruby implementation of an interactive SQL command-line for ODBC data sources
+test_files: []
+