lg 0.0.1

data/lib/gist_store/gist_store.rb

@@ -0,0 +1,62 @@
+require 'gist_store/gist'
+
+module Logbook
+ class GistStore
+    DATE_FMT = "%Y%m%d".freeze
+    attr_reader :error
+
+    def initialize
+      @valid = !!(ENV['GITHUB_USER'] && ENV['GITHUB_PASSWORD'])
+      @error = @valid ? '' : "Please set up GITHUB_USER and GITHUB_PASSWORD in your env."
+    end
+
+    def valid?
+      @valid
+    end
+    
+    def get(id, time)
+      gist = Gist.read_raw(id)
+      return nil unless gist
+      file = gist['files'][gist_file(time)]
+      return nil unless file
+      file['content']
+    end
+
+    def create(covertext)
+      u = URI.parse Gist.write(gist_data(covertext, "cover"), true)
+      u.path[1..-1]
+    end
+
+    def update(id, time, page)
+      Gist.update(id, gist_data(page, gist_file(time)))
+    end
+
+    def all(id)
+      gist = Gist.read_raw(id)
+      return nil unless gist
+
+      entries = gist['files'].reject{|k|  ["cover"].include? k }.map do |fname, v|
+        { :date    => DateTime.strptime(v['filename'], DATE_FMT),
+          :content => v['content'] }
+      end
+
+      { :id => id,
+        :cover => gist['files']['cover']['content'],
+        :entries => entries }
+    end
+
+    def destroy(id)
+      Gist.delete(id)
+    end
+
+  private
+    def gist_file(time)
+      time.strftime DATE_FMT
+    end
+
+    def gist_data(text, fname)
+      [{:input => text, :filename => fname, :extension => "txt"}]
+    end
+
+  end
+end

data/lib/gist_store/net_http_ext.rb

@@ -0,0 +1,56 @@
+require 'net/http'
+module Net
+  class HTTP    
+    
+  # Adding the patch method if it doesn't exist (rest-client issue: https://github.com/archiloque/rest-client/issues/79)
+  if !defined?(Net::HTTP::Patch)
+    # Code taken from this commit: https://github.com/ruby/ruby/commit/ab70e53ac3b5102d4ecbe8f38d4f76afad29d37d#lib/net/http.rb
+    class Protocol
+      # Sends a PATCH request to the +path+ and gets a response,
+      # as an HTTPResponse object.
+      def patch(path, data, initheader = nil, dest = nil, &block) # :yield: +body_segment+
+        send_entity(path, data, initheader, dest, Patch, &block)
+      end
+      
+      # Executes a request which uses a representation
+      # and returns its body.
+      def send_entity(path, data, initheader, dest, type, &block)
+        res = nil
+        request(type.new(path, initheader), data) {|r|
+          r.read_body dest, &block
+          res = r
+        }
+        unless @newimpl
+          res.value
+          return res, res.body
+        end
+        res
+      end
+    end
+    
+    class Patch < HTTPRequest
+      METHOD = 'PATCH'
+      REQUEST_HAS_BODY = true
+      RESPONSE_HAS_BODY = true
+    end
+  end
+
+    #
+    # Replace the request method in Net::HTTP to sniff the body type
+    # and set the stream if appropriate
+    #
+    # Taken from:	
+    # http://www.missiondata.com/blog/ruby/29/streaming-data-to-s3-with-ruby/
+
+    alias __request__ request
+
+    def request(req, body=nil, &block)
+      if body != nil && body.respond_to?(:read)
+        req.body_stream = body
+        return __request__(req, nil, &block)
+      else
+        return __request__(req, body, &block)
+      end
+    end
+  end
+end

data/lib/logbook.rb

@@ -0,0 +1,14 @@
+require 'gist_store/gist_store'
+
+module Logbook
+  def self.store
+    @store ||= GistStore.new
+  end
+  def self.store=(store)
+    @store = store
+  end
+end
+
+require "logbook/version"
+require 'logbook/book'
+

data/lib/logbook/book.rb

@@ -0,0 +1,79 @@
+
+require 'date'
+require 'chronic'
+require 'gist_store/gist_store'
+
+module Logbook
+  class Book
+    attr_reader :id
+    attr_accessor :store
+
+
+    def initialize(id=nil)
+      @id = id
+      @store = Logbook.store
+    end
+
+    def create(covertext=nil)
+      store_available
+
+      cover = covertext || "#{ENV['USER'] || 'Captain'}'s log."
+      @id = store.create cover
+    end
+
+    def add_temporal(text)
+      temporal = false
+
+      time = nil
+      if text =~ /(.+?)\s*:\s*(.*)/
+        time = Chronic.parse($1)
+        text = $2 if time
+        temporal = true if time
+      end
+
+      time ||= Time.now
+
+      add time, text
+    end
+
+    def add(time, text)
+      store_available
+      must_exist
+
+      page = get(time)
+      store.update(id, time, "#{page ? "#{page}\n" : ""}#{text}")
+      time
+    end
+
+    def get(time)
+      store_available
+      must_exist
+
+      page = store.get(id, time)
+    end
+
+
+    def all
+      store_available
+      must_exist
+      
+      store.all(id)
+    end
+  
+    def destroy
+      store_available
+      must_exist
+
+      store.destroy(id)
+    end
+
+    def store_available
+      raise store.error unless store.valid?
+    end
+
+    def must_exist
+      raise "create a book first" unless @id
+    end
+  end
+
+end

data/lib/logbook/cli.rb

@@ -0,0 +1,127 @@
+
+require 'logbook'
+require 'thor'
+require 'user_config'
+
+
+
+class Logbook::CLI < Thor
+
+  desc "book?", "which book are we on?"
+  def book?
+    j = current_book
+    return unless j
+    say config[:books][j.id]
+  end
+
+  desc "book [ID]", "open a book"
+  def book(*params)
+    text = params.join ' '
+    config[:books] ||= {}
+
+    # no argument given: prompt user to switch books from a menu
+    if text.empty? && !config[:books].empty?
+      choices = config[:books].to_a
+      choices = choices.map.with_index{ |a, i| [i+1, *a]}
+      print_table choices
+      selection = ask("Pick one:").to_i
+      if selection == 0 || choices.size < selection
+        error "No such option"
+        return
+      end
+
+      selected_id = choices[selection-1][1]
+      self.current_book = selected_id
+      ok "selected"
+      return
+    end
+
+    # we have an argument, switch to, or create a new one.
+    if config[:books].has_key? text
+      self.current_book = text
+      ok "switched"
+    else
+      if yes? "Create #{em(text)}?"
+        j = Logbook::Book.new
+        begin
+          id = j.create text
+          self.current_book = id
+
+          config[:books][id] = text
+          config.save 
+          ok "saved '#{text}' as #{em(id)}"
+        rescue
+          error $!
+        end
+      end
+    end
+  end
+
+  desc "add MEMORY", "add a new memory"
+  def add(*memory)
+    text = memory.join ' '
+    j = current_book
+    return unless j
+    begin
+      time = j.add_temporal text
+      ok("at #{em(time.to_s)}")
+    rescue
+      error $!
+    end
+  end
+
+  desc "all", "list the entire book"
+  def all
+    j = current_book
+    begin
+      data = j.all
+      say "[#{em data[:cover]}]\n"
+      data[:entries].group_by{|g| g[:date]}.each do |key, entries|
+        say "#{em key.strftime("%A, %d-%b-%Y")}"
+        entries.each do |entry|
+          say "#{entry[:content]}"
+        end
+      end
+    rescue
+      error $!
+    end
+  end
+
+
+private 
+  def current_book
+    id = config[:current_book]
+    unless id
+      error "No book is set. Create one with $ lg book #{em 'My Stories Unfold'}."
+      return nil
+    end
+    
+    Logbook::Book.new id
+  end
+
+  def current_book=(id)
+    config[:current_book] = id
+    config.save
+  end
+
+  def config
+    @uconf ||= UserConfig.new(".logbook")
+    @uconf["logbook.yaml"]
+  end
+
+  def em(text)
+    shell.set_color(text, nil, true)
+  end
+
+  def ok(detail=nil)
+    text = detail ? "OK, #{detail}." : "OK."
+    say text, :green
+  end
+
+  def error(detail)
+    say detail, :red
+  end
+
+
+end
+

data/lib/logbook/version.rb

@@ -0,0 +1,3 @@
+module Logbook
+  VERSION = "0.0.1"
+end

data/logbook.gemspec

@@ -0,0 +1,29 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "logbook/version"
+
+Gem::Specification.new do |s|
+  s.name        = "lg"
+  s.version     = Logbook::VERSION
+  s.authors     = ["Dotan Nahum"]
+  s.email       = ["jondotan@gmail.com"]
+  s.homepage    = ""
+  s.summary     = %q{log your memories onto virtual logbooks made of Gists}
+  s.description = %q{log your memories onto virtual logbooks made of Gists}
+
+  s.rubyforge_project = "lg"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  # specify any dependencies here; for example:
+  s.add_development_dependency "rr"
+  s.add_development_dependency "fakefs"
+  s.add_development_dependency "guard-minitest"
+  s.add_runtime_dependency "user_config"
+  s.add_runtime_dependency "thor"
+  s.add_runtime_dependency "chronic"
+
+end

data/spec/gist_store/gist_store_spec.rb

@@ -0,0 +1,86 @@
+require 'spec_helper'
+require 'gist_store/gist_store'
+
+def stamp
+  @stamp ||= DateTime.new(1981,9,8)
+end
+
+describe Logbook::GistStore do
+  it "should get content" do
+    mock(Gist).read_raw('bid') {
+      {
+        'files' => {
+          '19810908' => {
+            'content' => "hello world"
+          }
+        }
+      }
+    }
+
+    g = Logbook::GistStore.new
+    g.get('bid', stamp).must_equal "hello world"
+  end
+
+  it "should create" do
+    mock(Gist).write(
+      [{
+        :input => "hello world",
+        :filename => "cover",
+        :extension => "txt"
+      }],
+        true
+    ){
+      "http://example.com/id"
+    }
+
+    g = Logbook::GistStore.new
+    g.create("hello world")
+  end
+
+  it "should list all" do
+    mock(Gist).read_raw('bid'){
+      {
+        'files' => {
+          '19810908' => {
+            'filename' => '19810908',
+            'content' => "hello world"
+          },
+          'cover' => {
+            'filename' => 'cover',
+            'content' => "my book"
+          }
+        }
+      }
+    }
+
+    g = Logbook::GistStore.new
+    book = g.all('bid')
+
+    book.must_equal(
+      {
+        :id => 'bid',
+        :cover => 'my book',
+        :entries => [{:date => DateTime.strptime("19810908", "%Y%m%d"), :content => 'hello world'}]
+      }
+    )
+  end
+
+  it "should be invalid when env variables are not set" do
+    ENV['GITHUB_USER'] = nil
+    ENV['GITHUB_PASSWORD'] = nil
+
+    g = Logbook::GistStore.new
+    g.valid?.must_equal false
+    g.error.must_match /Please set up/
+  end
+
+    it "should be valid when env variables are set" do
+    ENV['GITHUB_USER'] = 'ghuser'
+    ENV['GITHUB_PASSWORD'] = 'ghpwd'
+
+    g = Logbook::GistStore.new
+    g.valid?.must_equal true
+  end
+
+end
+