treerful_scanner 1.0.1 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: bd67da624a014451be06adda6a3fef925d0466159b92820db3c3de3347e6c0d8
-  data.tar.gz: fdc4d06f0ee2b4402bc3a0d484942365402a82a66116b50f055e4d3a6f40027e
+  metadata.gz: 0a365a19a0eb1375fc78e84d6a2bea533452548b4bb56ee6843bd052ab3ac469
+  data.tar.gz: '08c18615a4d44f9d257010f40aace6ec423d2f4b61c59e43777bf7723aee7563'
 SHA512:
-  metadata.gz: 29c354e88d57ce826b8b3e098d6ec8d4a825bb7e862db128d608750bb3eea19a33ba7c9c069658aed2d038d13c8b100be9739692a715c9f7a406f85d61e70528
-  data.tar.gz: 56db138ec88c93517fae2d080827012b66794929e8d4e7e95a11c64cc8a2fbd9862f3d5c4c2db63ba99e93e5c0f55a1918879227d2e9c858a7cbc4b9a23ce478
+  metadata.gz: ff7a389844442ea8422095fb1f13b6e94d45b6429a0c331b8d1559173c1788509cfe6f8f7f02a231c077b1f8cb161a5cf98cf996e571a6909b0f191b81c053a9
+  data.tar.gz: 2f6cfbde0105d61b20929625c9e1a5319d4302bd0b28a9d8cc70aa45d3558e9124eff7134492c3d39e177d45c40d8f85020b23e647c211059b5b73daa8a324e4

data/bin/treerful_scanner

@@ -6,22 +6,57 @@ require 'date'
 require 'optparse'
 
 options = {
-  date: (Date.today + 1).to_s
+  date: (Date.today + 1).to_s,
+  places: [],
+  place_names: [],
+  format: 'terminal'
 }
 
 OptionParser.new do |opts|
   opts.banner = "Usage: #{File.basename __FILE__} [options]"
-  opts.on('-d DATE', '--date=DATE', 'yyyy-mm-dd') do |date|
+  opts.on('-d DATE', '--date DATE', 'yyyy-mm-dd') do |date|
     options[:date] = date
   end
+
+  opts.on('-n x,y,z', '--place-names x,y,z', Array, 'Filter by place names') do |place_names|
+    options[:place_names] = place_names
+  end
+
+  opts.on('-p x,y,z', '--places x,y,z', Array, 'Filter by place IDs') do |ids|
+    options[:places] = ids
+  end
+
+  opts.on('-f FORMAT', '--format FORMAT', '"terminal" (default), "html", "png"') do |format|
+    options[:format] = format
+  end
 end.parse!
 
 client = TreerfulScanner::Client.new
-printer = TreerfulScanner::Printer.new([], options[:date])
 
-count = 0
-client.search_by_date(options[:date]) do |table|
-  print printer.header if count % 10 == 0
-  print printer.row(table)
-  count += 1
+place_filter = -> (place) do
+  return true if options[:place_names].empty? && options[:places].empty?
+  return options[:place_names].any?{ |name| place.name.include?(name) } unless options[:place_names].empty?
+  return options[:places].include?(place.id) unless options[:places].empty?
+  false
+end
+
+case options[:format]
+when 'terminal'
+  printer = TreerfulScanner::Printer.new([], options[:date])
+  count = 0
+  client.search_by_date(options[:date], place_filter: place_filter) do |table|
+    print printer.header if count % 10 == 0
+    print printer.row(table)
+    count += 1
+  end
+when 'html'
+  time_tables = client.search_by_date(options[:date], place_filter: place_filter)
+  printer = TreerfulScanner::HtmlPrinter.new(time_tables, options[:date])
+  print printer.print
+when 'png'
+  time_tables = client.search_by_date(options[:date], place_filter: place_filter)
+  printer = TreerfulScanner::PngPrinter.new(time_tables, options[:date])
+  print printer.print
+else
+  raise "unknown format: #{options[:format]}"
 end

data/lib/treerful_scanner/client.rb

@@ -10,7 +10,7 @@ module TreerfulScanner
       @parser = Parser.new
     end
 
-    def search_by_date(date)
+    def search_by_date(date, place_filter: -> { true })
       date = date.to_s
       result = []
       EventMachine.run do
@@ -18,7 +18,7 @@ module TreerfulScanner
         client = conn.get path: '/space/result', keepalive: true
         client.callback do
           multi = EventMachine::MultiRequest.new
-          places = @parser.parse_places(client.response)
+          places = @parser.parse_places(client.response).select(&place_filter)
           places.each do |place|
             client2 = conn.get path: "/space/allowTimes?id=#{place.id}&date=#{date}", keepalive: true
             client2.callback do

data/lib/treerful_scanner/html_printer.rb

@@ -0,0 +1,39 @@
+# frozen_string_literal: true
+
+require 'treerful_scanner'
+require 'erb'
+
+module TreerfulScanner
+  class HtmlPrinter < Printer
+    attr_reader :date
+
+    def initialize(*)
+      super
+      @template = IO.read("#{__dir__}/html_printer/template.erb")
+    end
+
+    def print
+      erb = ERB.new(@template)
+      erb.result(binding)
+    end
+
+    private
+
+    def tbody
+      "<tbody>#{@time_tables.map(&method(:to_tr)).join}</tbody>"
+    end
+
+    def to_tr(time_table)
+      result = String.new
+      result << "<tr><td><a href=\"https://www.treerful.com/space/#{time_table.place.id}\" rel=\"noopener noreferrer\" target=\"_blank\">#{time_table.place.name}</a></td><td>#{time_table.place.volume}</td>"
+      result <<
+        binary_form(time_table).each_char.map do |c|
+          case c
+          when ' ' then '<td></td>'
+          when 'O' then '<td class="available"></td>'
+          end
+        end.join
+      result << '</tr>'
+    end
+  end
+end

data/lib/treerful_scanner/png_printer.rb

@@ -0,0 +1,42 @@
+# frozen_string_literal: true
+
+require 'treerful_scanner'
+require 'tmpdir'
+require 'English'
+
+module TreerfulScanner
+  class PngPrinter < Printer
+    CHROME_BINS = [
+      'chromium-browser',
+      '/Applications/Google Chrome.app/Contents/MacOS/Google Chrome'
+    ].freeze
+    WINDOW_HEIGHT_OFFSET = 200
+    WINDOW_CELL_HEIGHT = 24.5
+
+    def print
+      html_printer = HtmlPrinter.new(@time_tables, @date)
+      Dir.mktmpdir do |dir|
+        html_path = "#{dir}/output.html"
+        IO.write(html_path, html_printer.print)
+        chrome_bin = CHROME_BINS.find(&File.method(:exist?)) || ENV['CHROME_BIN']
+        raise 'ENV["CHROME_BIN"] is nil' unless chrome_bin
+        window_height = WINDOW_HEIGHT_OFFSET + WINDOW_CELL_HEIGHT * @time_tables.length
+        Dir.chdir(dir) do
+          raise "chrome exited with status #{$CHILD_STATUS}" unless system(
+            chrome_bin,
+            '--headless',
+            '--disable-gpu',
+            '--disable-software-rasterizer',
+            '--disable-dev-shm-usage',
+            '--no-sandbox',
+            '--screenshot',
+            '--hide-scrollbars',
+            "--window-size=1280,#{window_height}",
+            html_path
+          )
+          IO.read('screenshot.png', mode: 'rb')
+        end
+      end
+    end
+  end
+end

data/lib/treerful_scanner/printer.rb

@@ -17,18 +17,21 @@ module TreerfulScanner
     end
 
     def row(time_table)
-      tmp = String.new
-      tmp << ' ' * (BEGIN_HOUR..END_HOUR).size * 2
+      fixed_string(time_table.place.name, TITLE_WIDTH) +
+        binary_form(time_table).each_char.map { |i| format('%-2s', i) }.join +
+        "\n"
+    end
+
+    def binary_form(time_table)
+      result = String.new
+      result << ' ' * (BEGIN_HOUR..END_HOUR).size * 2
       offset = BEGIN_HOUR * 60
       time_table.durations.each do |duration|
         start = (duration.from - offset) / 30
         finish = (duration.to - offset) / 30
-        tmp[start..finish] = 'O' * (finish - start + 1)
+        result[start...finish] = 'O' * (finish - start)
       end
-
-      fixied_string(time_table.place.name, TITLE_WIDTH) +
-        tmp.each_char.map { |i| format('%-2s', i) }.join +
-        "\n"
+      result
     end
 
     def print
@@ -42,7 +45,7 @@ module TreerfulScanner
 
     private
 
-    def fixied_string(string, width)
+    def fixed_string(string, width)
       length = string.length + string.scan(/\p{Han}|\p{Katakana}|\p{Hiragana}\p{Hangul}/).length
       diff = width - length
       return string + (' ' * diff) if diff > 0

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: treerful_scanner
 version: !ruby/object:Gem::Version
-  version: 1.0.1
+  version: 1.1.0
 platform: ruby
 authors:
 - Jian Weihang
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2018-09-18 00:00:00.000000000 Z
+date: 2018-09-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: nokogiri
@@ -91,8 +91,10 @@ files:
 - lib/treerful_scanner.rb
 - lib/treerful_scanner/client.rb
 - lib/treerful_scanner/duration.rb
+- lib/treerful_scanner/html_printer.rb
 - lib/treerful_scanner/parser.rb
 - lib/treerful_scanner/place.rb
+- lib/treerful_scanner/png_printer.rb
 - lib/treerful_scanner/printer.rb
 - lib/treerful_scanner/time_table.rb
 homepage: https://github.com/tonytonyjan/treerful_scanner