snipper 0.0.1

data/bin/snipper

@@ -0,0 +1,32 @@
+#!/usr/bin/ruby
+
+require 'snipper'
+
+@options = {:command     => "new"}
+
+OptionParser.new do |opts|
+  opts.on("--lang LANGUAGE", "-l", "Snippet Language") do |v|
+    @options[:syntax] = v
+  end
+
+  opts.on("-L","List Languages") do
+    @options[:list] = true
+  end
+end.parse!
+
+exec "pygmentize -L lexer" if @options[:list]
+
+@options[:command] = ARGV[0] if ARGV.size >= 1
+
+case @options[:command]
+  when "search", "s"
+    Snipper::Command::Search.new(@options)
+  when "delete", "d", "rm"
+    Snipper::Command::Delete.new(@options)
+  when "view", "v"
+    Snipper::Command::View.new(@options)
+  when "edit", "e"
+    puts Snipper::Command::Edit.new(@options).url
+  else
+    puts Snipper::Command::New.new(@options).url
+end

data/lib/snipper.rb

@@ -0,0 +1,73 @@
+require 'rubygems'
+require 'pygments.rb'
+require 'optparse'
+require 'etc'
+require 'snipper/config'
+require 'snipper/snippet'
+require 'snipper/command/new'
+require 'snipper/command/edit'
+require 'snipper/command/delete'
+require 'snipper/command/search'
+require 'snipper/command/view'
+require 'snipper/version'
+require 'snipper/output/pygments_output.rb'
+require 'yaml'
+require 'fileutils'
+
+class Snipper
+  def initialize
+    config = File.join(Etc.getpwuid.dir, ".snipper", "config.yml")
+
+    if File.exist?(config)
+      Config.load(config)
+    else
+      Config.load({:config_file => config})
+    end
+
+    setup
+  end
+
+  def setup
+    FileUtils.mkdir_p(Config[:snippets_dir])
+
+    unless File.exist?(Config[:config_file])
+      File.open(Config[:config_file], "w") do |c|
+        c.print Config.to_yaml
+      end
+    end
+  end
+end
+
+class Array
+  def in_groups_of(chunk_size, padded_with=nil, &block)
+    arr = self.clone
+
+    # how many to add
+    padding = chunk_size - (arr.size % chunk_size)
+
+    # pad at the end
+    arr.concat([padded_with] * padding) unless padding == chunk_size
+
+    # how many chunks we'll make
+    count = arr.size / chunk_size
+
+    # make that many arrays
+    result = []
+    count.times {|s| result <<  arr[s * chunk_size, chunk_size]}
+
+    if block_given?
+      result.each_with_index do |a, i|
+        case block.arity
+          when 1
+            yield(a)
+          when 2
+            yield(a, (i == result.size - 1))
+          else
+            raise "Expected 1 or 2 arguments, got #{block.arity}"
+        end
+      end
+    else
+      result
+    end
+  end unless method_defined?(:in_groups_of)
+end

data/lib/snipper/command/delete.rb

@@ -0,0 +1,21 @@
+class Snipper
+  class Command
+    class Delete
+      attr_reader :url, :snippet
+
+      def initialize(options)
+        @options = options
+
+        snipper = Snipper.new
+
+        raise "Please specify a snippet number" unless ARGV[1] =~ /^\d+$/
+
+        options[:snippet] = ARGV[1]
+
+        snippet = Snippet.new(nil, @options)
+
+        snippet.delete!
+      end
+    end
+  end
+end

data/lib/snipper/command/edit.rb

@@ -0,0 +1,35 @@
+class Snipper
+  class Command
+    class Edit
+      attr_reader :url, :snippet
+
+      def initialize(options)
+        @options = options
+
+        snipper = Snipper.new
+
+        raise "Please specify a snippet number" unless ARGV[1] =~ /^\d+$/
+
+        options[:snippet] = ARGV[1]
+
+        snippet = Snippet.new(nil, @options)
+        @url = snippet.url_for_snippet
+
+        files = snippet.files.map {|file| File.join(snippet.snippet_path, file)}
+
+        editor = ENV["EDITOR"] || "vi"
+
+        system "%s %s" % [editor, files.join(" ")]
+
+        files.each do |file|
+          unless File.size?(file)
+            puts "Removing #{file} from the snippet"
+            File.unlink(file)
+          end
+        end
+
+        snippet.update_html
+      end
+    end
+  end
+end

data/lib/snipper/command/new.rb

@@ -0,0 +1,30 @@
+class Snipper
+  class Command
+    class New
+      attr_reader :url, :snippet
+
+      def initialize(options)
+        @options = options
+
+        @snipper = Snipper.new
+
+        # if the first arg is a number and it isnt a file path
+        # then assume this is a snippet id we want to append to
+        if ARGV.first =~ /^\d+$/ && !File.exist?(ARGV.first)
+          options[:snippet] = ARGV.shift
+        end
+
+        if STDIN.tty?
+          abort "Please specify a filename or provide one on STDIN" if ARGV.empty?
+
+          txt = ARGV
+        else
+          txt = STDIN.read
+        end
+
+        @snippet = Snippet.new(txt, @options)
+        @url = @snippet.url_for_snippet
+      end
+    end
+  end
+end

data/lib/snipper/command/search.rb

@@ -0,0 +1,21 @@
+class Snipper
+  class Command
+    class Search
+      attr_reader :url, :snippet
+
+      def initialize(options)
+        @options = options
+
+        snipper = Snipper.new
+
+        raise "Please specify a search string" unless ARGV[1]
+
+        Dir.chdir(Config[:snippets_dir]) do
+          grep = Config[:grep].gsub("%Q%", ARGV[1].gsub("'", "\\'"))
+
+          system grep
+        end
+      end
+    end
+  end
+end

data/lib/snipper/command/view.rb

@@ -0,0 +1,25 @@
+class Snipper
+  class Command
+    class View
+      attr_reader :url, :snippet
+
+      def initialize(options)
+        @options = options
+
+        snipper = Snipper.new
+
+        raise "Please specify a snippet number" unless ARGV[1] =~ /^\d+$/
+
+        options[:snippet] = ARGV[1]
+
+        snippet = Snippet.new(nil, @options)
+
+        files = snippet.files.map {|file| File.join(snippet.snippet_path, file)}
+
+        pager = ENV["PAGER"] || "less"
+
+        system "cat %s|%s" % [files.join(" "), pager]
+      end
+    end
+  end
+end

data/lib/snipper/config.rb

@@ -0,0 +1,36 @@
+## description: shitty config class
+class Snipper
+  class Config
+    class << self
+      def load(source)
+        @config = {:renderer => "ultraviolet",
+                   :grep     => "grep --color=auto -riHn -C 2 '%Q%' *",
+                   :snipper_dir => File.join(Etc.getpwuid.dir, ".snipper"),
+                   :snippets_dir => File.join(Etc.getpwuid.dir, ".snipper", "snippets"),
+                   :config_file => File.join(Etc.getpwuid.dir, ".snipper", "config.yml")}
+
+        if source.is_a?(String)
+          raise "Config file #{source} not found" unless File.exist?(source)
+
+          config = YAML.load_file(source)
+          @config.merge! config if config
+
+        elsif source.is_a?(Hash)
+          @config.merge! source
+        end
+      end
+
+      def include?(key)
+        @config.include?(key)
+      end
+
+      def [](key)
+        @config[key]
+      end
+
+      def to_yaml
+        @config.to_yaml
+      end
+    end
+  end
+end

data/lib/snipper/output/pygments_output.rb

@@ -0,0 +1,75 @@
+class Snipper
+  class Output
+    class Pygments_Output
+      def initialize(snippet)
+        @snippet = snippet
+      end
+
+      def self.list_langs
+        Pygments.lexers.map{|l| l[1][:aliases]}.flatten.sort
+      end
+
+      def save
+        theme = Config[:theme] || "default"
+
+        raise "Please set the :public_target_dir setting" unless Config[:public_target_dir]
+
+        path = @snippet.snippet_html_path
+
+        FileUtils.mkdir_p(path)
+
+        File.open(File.join(path, "index.html"), "w") do |html|
+          html.puts "<head>"
+          html.puts "<link rel='stylesheet' type='text/css' href='../css/#{theme}.css' media='all' />"
+          html.puts "<style type='text/css'>"
+          html.puts ".highlighttable { background-color: #253B76; width: 100%;}"
+          html.puts ".content { width: 100%;}"
+          html.puts ".linenos { text-align: right; color: white; width: 30px; }"
+
+          if Config[:dark_theme]
+            html.puts ".highlight { background-color: #0C1000 }"
+          else
+            html.puts ".highlight { background-color: white }"
+          end
+
+          html.puts "</style>"
+          html.puts "</head>"
+          html.puts "<body>"
+
+          html.puts "<table class='content'>"
+          html.puts "<tr><td>"
+
+          @snippet.each_file do |file, text, headers|
+            File.open(File.join(path, "#{File.basename(file)}.raw"), "w") do |f|
+              f.puts text
+            end
+
+            syntax = @snippet.options[:syntax]
+
+            unless syntax
+              lexer = Pygments::Lexer.find_by_extname(File.extname(file))
+
+              if lexer
+                syntax = lexer.name
+              else
+                syntax = Config[:default_syntax] || "ruby"
+              end
+            end
+
+            raise "Unknown syntax #{syntax}" unless self.class.list_langs.include?(syntax)
+
+            html.puts "<H2>%s</H2>" % [ headers["description"] ] if headers["description"]
+            html.puts Pygments.highlight(text, :lexer => syntax, :options => {:linenos => "table", :style => theme})
+            html.puts "syntax: %s download: <a href='%s.raw'>raw</a>" % [ syntax, File.basename(file) ]
+            html.puts "<br /><br />"
+          end
+
+          html.puts "</td></tr>"
+          html.puts "<tr align='right'><td><a href='%s'>Snipper</a></td></tr>" % [ "http://github.com/ripienaar/snipper" ]
+          html.puts "</table>"
+          html.puts "</body>"
+        end
+      end
+    end
+  end
+end

data/lib/snipper/snippet.rb

@@ -0,0 +1,158 @@
+class Snipper
+  class Snippet
+    attr_reader :options
+
+    def initialize(snippet, options = {:snippet => nil})
+      @options = options
+
+      if options[:snippet]
+        raise "Snippets ids must be numerical" unless options[:snippet] == options[:snippet].to_i.to_s
+        raise "Unknown snippet %s" % [ options[:snippet] ] unless File.exist?(snippet_path)
+      end
+
+      if snippet
+        # new snippet
+        begin
+          unless @options[:snippet]
+            @options[:snippet] = next_snippet_id
+
+            create_snippet_dir
+
+            append_snippet(snippet)
+
+            update_html
+          else # add to existing snippet
+            append_snippet(snippet)
+
+            update_html
+          end
+        rescue Exception => e
+          STDERR.puts "Could not save snippet: #{e.class}: #{e}"
+          # TODO: clear up half baked snippets
+
+          raise
+        end
+      end
+    end
+
+    def delete!
+      raise "Refusing to remove all snippets" if File.expand_path(snippet_path) == File.expand_path(Config[:public_target_dir])
+      raise "Snippet path #{snippet_path} looks suspect, refusing to delete" unless snippet_path.match(/\d+$/)
+
+      raise "Refusing to remove all snippets" if File.expand_path(snippet_html_path) == File.expand_path(Config[:public_target_dir])
+      raise "Snippet path #{snippet_html_path} looks suspect, refusing to delete" unless snippet_html_path.match(/\d+$/)
+
+      if File.directory?(snippet_path)
+        puts "Removing #{snippet_path}"
+        FileUtils.remove_entry_secure(snippet_path)
+      end
+
+      if File.directory?(snippet_html_path)
+        puts "Removing #{snippet_html_path}"
+        FileUtils.remove_entry_secure(snippet_html_path)
+      end
+    end
+
+    def append_snippet(snippet)
+      if snippet.is_a?(Array)
+        [snippet].flatten.each do |file|
+          save_next_file(snippet_file(File.read(file)))
+        end
+      else
+        save_next_file(snippet_file(snippet))
+      end
+    end
+
+    def update_html
+      Output::Pygments_Output.new(self).save
+    end
+
+    def url_for_snippet
+      "%s/%s" % [ Config[:baseurl], @options[:snippet] ]
+    end
+
+    def save_next_file(text)
+      file_name = File.join(snippet_path, next_file_id.to_s)
+
+      File.open(file_name, "w") do |file|
+        file.puts text
+      end
+
+      return file_name
+    end
+
+    def files
+      Dir.entries(snippet_path).grep(/^\d+$/).sort_by{|i| Integer(i)}
+    end
+
+    def each_file
+      files.each do |file|
+        headers, text = parse_file(File.read(File.join(snippet_path, file)))
+
+        yield File.join(snippet_path, file), text, headers
+      end
+    end
+
+    def parse_file(file)
+      headers = {}
+      text = ""
+      inheader = true
+
+      file.each_line do |line|
+        # skips the first blank line
+        if line == "\n" && inheader
+          inheader = false
+          next
+        end
+
+        if line.start_with?("##") && inheader
+          if line =~ /^## (\S+): (.+)/
+            headers[$1] = $2
+          elsif line =~ /## (.+)/
+            headers["description"] = $1
+          end
+        else
+          inheader = false
+          text += line
+        end
+      end
+
+      return headers, text
+    end
+
+    def snippet_html_path
+      File.expand_path(File.join(Config[:public_target_dir], @options[:snippet].to_s))
+    end
+
+    def snippet_path
+      File.join(Config[:snippets_dir], @options[:snippet].to_s)
+    end
+
+    def create_snippet_dir
+      FileUtils.mkdir(snippet_path)
+    end
+
+    def next_file_id
+      Integer(files[-1]) + 1
+    end
+
+    def next_snippet_id
+      snippets = Dir.entries(Config[:snippets_dir]).grep(/^\d+$/)
+      Integer(snippets.map{|s| s.to_i}.sort.max) + 1
+    end
+
+    def snippet_file(text)
+      snippet = StringIO.new
+
+      unless text.start_with?("##")
+        snippet.puts "## description: %s" % [ @options[:description] ] if @options[:description]
+        snippet.puts "## syntax: %s" % [ @options[:syntax] ] if @options[:syntax]
+        snippet.puts
+      end
+
+      snippet.puts text
+
+      snippet.string
+    end
+  end
+end

data/lib/snipper/version.rb

@@ -0,0 +1,3 @@
+class Snipper
+  VERSION="0.0.1"
+end

metadata

@@ -0,0 +1,119 @@
+--- !ruby/object:Gem::Specification 
+name: snipper
+version: !ruby/object:Gem::Version 
+  hash: 29
+  prerelease: false
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- R.I.Pienaar
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2012-04-26 00:00:00 +01:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: rake
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: rdoc
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id002
+- !ruby/object:Gem::Dependency 
+  name: pygments.rb
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id003
+description: A Unix CLI centric snippet manager that produces static files
+email: rip@devco.net
+executables: 
+- snipper
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- bin/snipper
+- lib/snipper/snippet.rb
+- lib/snipper/config.rb
+- lib/snipper/output/pygments_output.rb
+- lib/snipper/version.rb
+- lib/snipper/command/delete.rb
+- lib/snipper/command/edit.rb
+- lib/snipper/command/new.rb
+- lib/snipper/command/search.rb
+- lib/snipper/command/view.rb
+- lib/snipper.rb
+has_rdoc: true
+homepage: http://devco.net/
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: Snipper
+test_files: []
+