codestock 0.1.0

data/lib/grenweb/query.rb

@@ -0,0 +1,82 @@
+# -*- coding: utf-8 -*-
+#
+# @file 
+# @brief  クエリーの解析
+# @author ongaeshi
+# @date   2010/10/21
+
+require 'rubygems'
+require 'rack'
+
+module Grenweb
+  class Query
+    include Rack::Utils
+    attr_reader :query_string
+  
+    OPTIONS = [
+               ['package',  'p'],
+               ['filepath', 'fpath', 'f'],
+               ['suffix',   's'],
+              ]
+
+    def initialize(request)
+      @query_string = unescape(request.path_info.gsub(/\A\/|\/\z/, ''))
+      init_hash
+      parse
+    end
+
+    def escape_html
+      Rack::Utils::escape_html(@query_string)
+    end
+
+    def empty?
+      keywords.size == 0 && packages.size == 0 && fpaths.size == 0 && suffixs.size == 0
+    end
+
+    def keywords
+      @hash['keywords']
+    end
+
+    def packages
+      calc_param(0)
+    end
+
+    def fpaths
+      calc_param(1)
+    end
+
+    def suffixs
+      calc_param(2)
+    end
+
+    private
+
+    def calc_param(index)
+      OPTIONS[index].inject([]){|result, item| result.concat @hash[item] }
+    end
+
+    def init_hash
+      @hash = {}
+      @hash['keywords'] = []
+
+      OPTIONS.flatten.each do |key|
+        @hash[key] = []
+      end
+    end
+
+    def parse
+      kp = OPTIONS.flatten.join('|')
+      parts = @query_string.scan(/(?:(#{kp}):)?(?:"(.+)"|(\S+))/)
+
+      parts.each do |key, quoted_value, value|
+        text = quoted_value || value
+        unless (key)
+          @hash['keywords'] << text
+        else
+          @hash[key] << text
+        end
+      end
+    end
+  end
+end
+

data/lib/grenweb/searcher.rb

@@ -0,0 +1,130 @@
+# -*- coding: utf-8 -*-
+#
+# @file 
+# @brief  検索処理本体
+# @author ongaeshi
+# @date   2010/10/13
+
+require 'rubygems'
+require 'rack'
+require File.join(File.dirname(__FILE__), 'database')
+require File.join(File.dirname(__FILE__), 'html_renderer')
+require File.join(File.dirname(__FILE__), 'query')
+
+module Grenweb
+  class Searcher
+    include Rack::Utils
+
+    def initialize
+    end
+    
+    def call(env)
+      @request = Rack::Request.new(env)
+      @query = Query.new(@request)
+
+      @response = Rack::Response.new
+      @response["Content-Type"] = "text/html; charset=UTF-8"
+
+      @nth = 3                  # マッチした行の前後何行を表示するか
+
+      @rendeler = HTMLRendeler.new(@request.script_name + '/..')
+
+      if @request.post? or @request['query']
+        post_request
+      else
+        search
+      end
+    end
+
+    private
+
+    def post_request
+      query = @request['query'] || ''
+      if query.empty?
+        @request.path_info = "/"
+      else
+        @request.path_info = "/#{escape(query)}/"
+      end
+      @response.redirect(@request.url.split(/\?/, 2)[0])
+      @response.to_a
+    end
+
+    def search
+      render_header
+      render_search_box
+      render_search_result
+      render_footer
+      @response.to_a
+    end
+
+    def render_header
+      @response.write @rendeler.header("gren : #{@query.escape_html}", "gren")
+    end
+
+    def render_search_box
+      @response.write @rendeler.search_box(@query.escape_html)
+    end
+
+    def render_search_result
+      if @query.empty?
+        @response.write @rendeler.empty_summary
+      else
+        records, total_records, elapsed = Database.instance.search(@query.keywords, @query.packages, @query.fpaths, @query.suffixs, calcPage, calcLimit)
+        render_search_summary(records, total_records, elapsed)
+        records.each { |record| @response.write(@rendeler.result_record(record, @query.keywords, @nth)) }
+        render_pagination(calcPage, total_records)
+      end
+    end
+
+    def render_search_summary(records, total_records, elapsed)
+      pageStart = calcPage * calcLimit
+      @response.write @rendeler.search_summary(@query.query_string,
+                                                  total_records,
+                                                  (total_records.zero? ? 0 : pageStart + 1)..(pageStart + records.size),
+                                                  elapsed)
+    end
+
+    def render_pagination(page, total_records)
+      return if @query.empty?
+      return if total_records < calcLimit
+
+      last_page = (total_records / calcLimit.to_f).ceil
+      @response.write("<div class='pagination'>\n")
+      if page > 0
+        @response.write(@rendeler.pagination_link(page - 1, "<<"))
+      end
+      last_page.times do |i|
+        if i == page
+          @response.write(@rendeler.pagination_span(i))
+        else
+          @response.write(@rendeler.pagination_link(i, i))
+        end
+      end
+      if page < (last_page - 1)
+        @response.write(@rendeler.pagination_link(page + 1, ">>"))
+      end
+      @response.write("</div>\n")
+    end
+
+    def render_footer
+      @response.write @rendeler.footer
+    end
+
+    private
+
+    # 1ページに表示する最大レコードを計算
+    def calcLimit
+      if @query.keywords.size == 0
+        100
+      else
+        20
+      end
+    end
+    
+    # 現在ページを計算
+    def calcPage
+      (@request['page'] || 0).to_i
+    end
+  end
+end
+

data/lib/grenweb/viewer.rb

@@ -0,0 +1,52 @@
+# -*- coding: utf-8 -*-
+#
+# @file 
+# @brief  ソースコードを表示する
+# @author ongaeshi
+# @date   2010/10/13
+
+require 'rubygems'
+require 'rack'
+require File.join(File.dirname(__FILE__), 'database')
+require File.join(File.dirname(__FILE__), 'html_renderer')
+
+module Grenweb
+  class Viewer
+    include Rack::Utils
+
+    def initialize
+    end
+    
+    def call(env)
+      @request = Rack::Request.new(env)
+      @response = Rack::Response.new
+      @response["Content-Type"] = "text/html; charset=UTF-8"
+
+      record, elapsed = Database.instance.record(req2query)
+
+      @rendeler = HTMLRendeler.new(@request.script_name + '/..')
+
+      if (record)
+        @response.write @rendeler.header("gren : #{record.shortpath}", "gren")
+        @response.write @rendeler.search_box("")
+        @response.write @rendeler.view_summary(record.shortpath, elapsed)
+        @response.write @rendeler.record_content(record)
+      else
+        @response.write @rendeler.header("gren : not found.", "gren")
+        @response.write @rendeler.search_box("")
+        @response.write @rendeler.empty_summary
+      end
+      @response.write @rendeler.footer
+      
+      @response.to_a
+    end
+
+    private
+
+    def req2query
+      p @request.path_info
+      unescape(@request.path_info.gsub(/\A\/|\/z/, ''))
+    end
+    
+  end
+end

data/lib/mkgrendb/cli.rb

@@ -0,0 +1,89 @@
+# -*- coding: utf-8 -*-
+require 'optparse'
+require File.join(File.dirname(__FILE__), 'mkgrendb')
+require File.join(File.dirname(__FILE__), '../common/dbdir.rb')
+include CodeStock
+
+module Mkgrendb
+  class CLI
+    def self.execute(stdout, arguments=[])
+      opt = OptionParser.new <<EOF
+#{File.basename($0)} COMMAND [ARGS]
+
+The most commonly used mkgrendb are:
+  init        Init db.
+  update      Update db.
+  add         Add contents. (ex. ~/Documents/gren, git://github.com/ongaeshi/gren.git)
+  remove      Remove contents.
+  list        List all contents. 
+  rebuild     Rebuild db.
+  dump        Dump database contents.
+EOF
+
+      subopt = Hash.new
+      
+      init_default = false
+      subopt['init'] = OptionParser.new("#{File.basename($0)} init")
+      subopt['init'].on('--default', 'Init db default path. (Maybe ~/.codestock)') { init_default = true }
+      
+      subopt['update'] = OptionParser.new("#{File.basename($0)} update")
+      subopt['add'] = OptionParser.new("#{File.basename($0)} add content1 [content2 ...]")
+      subopt['remove'] = OptionParser.new("#{File.basename($0)} remove content1 [content2 ...]")
+      subopt['list'] = OptionParser.new("#{File.basename($0)} list")
+      subopt['rebuild'] = OptionParser.new("#{File.basename($0)} rebuild")
+      subopt['dump'] = OptionParser.new("#{File.basename($0)} dump")
+
+      opt.order!(arguments)
+      subcommand = arguments.shift
+
+      if (subopt[subcommand])
+        subopt[subcommand].parse!(arguments) unless arguments.empty?
+
+        db_dir = select_dbdir(subcommand, init_default)
+        obj = Mkgrendb.new(stdout, db_dir)
+
+        case subcommand
+        when 'init'
+          FileUtils.mkdir_p db_dir if (init_default)
+          obj.init 
+        when 'update'
+          obj.update
+        when 'add'
+          obj.add *arguments
+        when 'remove'
+          obj.remove *arguments
+        when 'list'
+          obj.list
+        when 'rebuild'
+          obj.rebuild
+        when 'dump'
+          obj.dump
+        end
+      else
+        if subcommand
+          $stderr.puts "mkgrendb: '#{subcommand}' is not a mkgrendb command. See 'mkgrendb --help'"
+        else
+          stdout.puts opt.help
+        end
+      end
+    end
+
+    private
+
+    def self.select_dbdir(subcommand, init_default)
+      if (subcommand == 'init')
+        if (init_default)
+          db_default_dir
+        else
+          '.'
+        end
+      else
+        if (dbdir?('.') || !dbdir?(db_default_dir))
+          '.'
+        else
+          db_default_dir
+        end
+      end
+    end
+  end
+end

data/lib/mkgrendb/cli_old.rb

@@ -0,0 +1,49 @@
+# -*- coding: utf-8 -*-
+require 'optparse'
+require File.join(File.dirname(__FILE__), 'mkgrendb')
+
+module Mkgrendb
+  class CLI
+    def self.execute(stdout, arguments=[])
+      input_yamls = []
+      isDump = false
+      isFull = false
+      isDelete = false
+      isReport = false
+
+      opt = OptionParser.new "#{File.basename($0)} INPUT_YAML1 [INPUT_YAML2 ...]"
+      opt.on('--ddb', "--default-db", "Create or Update default DB. (Plase set ENV['GRENDB_DEFAULT_DB'])") {|v| input_yamls << ENV['GRENDB_DEFAULT_DB']}
+      opt.on('--full', "Full update DB. (Delete and create)") {|v| isFull = true }
+      opt.on('--delete', "Delete DB. (Not delete yaml)") {|v| isDelete = true }
+      opt.on('--dump', "Dump DB.") {|v| isDump = true }
+      opt.on('--report', "Database Report.") {|v| isReport = true }
+      opt.parse!(arguments)
+
+      input_yamls.concat arguments
+
+      if (input_yamls.size >= 1)
+        input_yamls.each do |input_yaml|
+          obj = Mkgrendb.new(input_yaml)
+          
+          if (isFull)
+            obj.full
+            stdout.puts
+          elsif (isDelete)
+            obj.delete
+            stdout.puts
+          elsif (isDump)
+            obj.dump
+          elsif (isReport)
+            obj.report
+          else
+            obj.update
+            stdout.puts
+          end
+
+        end
+      else
+        stdout.puts opt.help
+      end
+    end
+  end
+end

data/lib/mkgrendb/grendbyaml.rb

@@ -0,0 +1,76 @@
+# -*- coding: utf-8 -*-
+#
+# @file 
+# @brief
+# @author ongaeshi
+# @date   2011/02/20
+
+require 'yaml'
+require 'pathname'
+
+module Mkgrendb
+  class GrendbYAML
+    class YAMLAlreadyExist < RuntimeError
+    end
+    
+    class YAMLNotExist < RuntimeError
+    end
+
+    def self.create(path = ".")
+      yf = yaml_file(path)
+      raise YAMLAlreadyExist.new if FileTest.exist? yf
+      obj = GrendbYAML.new(yf, {'contents' => [], 'version' => 0.1})
+      obj.save
+      return obj
+    end
+
+    def self.load(path = ".")
+      yf = yaml_file(path)
+      raise YAMLNotExist.new unless FileTest.exist? yf
+      open(yf) do |f|
+        return GrendbYAML.new(yf, YAML.load(f.read()))
+      end
+    end
+
+    def add(*dirs)
+      contents.push(*dirs.map{|v|{"directory" => v}})
+      contents.uniq!
+    end
+
+    def remove(*dirs)
+      dirs.each {|f| contents.delete( {"directory" => f} ) }
+    end
+
+    def save
+      open(@yaml_file, "w") { |f| YAML.dump(@data, f) }
+    end
+
+    def contents
+      @data['contents']
+    end
+
+    def directorys
+      contents.map{|v|v["directory"]}
+    end
+
+    def version
+      @data['version']
+    end
+
+    def list
+      directorys
+    end
+
+    def self.yaml_file(path)
+      (Pathname.new(path) + 'grendb.yaml').to_s
+    end
+
+    private
+
+    def initialize(yaml_file, data)
+      @yaml_file = yaml_file
+      @data = data
+    end
+
+  end
+end