kor-input-ltsv 0.0.2 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: c5ee7e82e979bfad7c0d4ea68bd375295cd8b2d9
-  data.tar.gz: 6233224fe379142344d51e6a3535f2bae1c1029a
+  metadata.gz: b6212739aa66563f7e318f2cef919c8f83ac58a2
+  data.tar.gz: 45c50fc227d75b80483978e13d624d2feab185ce
 SHA512:
-  metadata.gz: cc805dc7e20c889a11bffbf063dbf9816b8d88e128357e6641514d1456a1a30561ada18313bfd5ff3f4687ba8cbc90b83f2dd6148d02ac0c5bf66ba80755a121
-  data.tar.gz: d4f56feab73a50527113854c595378b5fa8c75fc2215dbd3c004aa6b7681b3f547ac9e4d47421a881f0fbab80c76e8ac8f329cbc94f097beb49beeec80b656c4
+  metadata.gz: f9faba9eb923cbd5ab08252457aa36e0464ed31be67a9a362481279453caa6f05dd15e6c9695ea145aae115bd03f10c4bc4e0b9bc409962edef21c8581399e1c
+  data.tar.gz: 36ea2d4db59ee0339616c03cb796523727a6f41af7c62005ae539ca611f6bbb2f11569f3bb9b7173967cd04548b872519bbcafb4b44160bffe5ee3a7b0b2afad

data/kor-input-ltsv.gemspec

@@ -18,6 +18,7 @@ Gem::Specification.new do |spec|
   spec.require_paths = ["lib"]
 
   spec.add_runtime_dependency "kor"
+  spec.add_runtime_dependency "ltsv"
   spec.add_development_dependency "bundler", "~> 1.10"
   spec.add_development_dependency "rake", "~> 10.0"
   spec.add_development_dependency "rgot"

data/lib/kor/input/ltsv.rb

@@ -1,22 +1,19 @@
 require 'kor'
+require 'ltsv'
 
 module Kor
   module Input
     class Ltsv
-      class LTSV < Hash
-        DELIM = "\t".freeze
-        SPLIT = ":".freeze
-
-        def self.parse(line)
-          LTSV[line.split(DELIM).map{ |i| i.split(SPLIT, 2) }]
-        end
-      end
+      DEFAULT_GUESS_TIME = 5
 
       def initialize(io)
         @io = io
         @keys = []
         @prekeys = nil
         @ltsvs = []
+        @guess = true
+        @guess_time = DEFAULT_GUESS_TIME
+        @count = 0
         @fiber = Fiber.new do
           @ltsvs.each do |ltsv|
             Fiber.yield @keys.map{ |k| ltsv[k] }
@@ -30,6 +27,9 @@ module Kor
         opt.on("--key=KEY", "define keys preset (e.g. foo,bar,baz) (default auto)") do |arg|
           @prekeys = arg
         end
+        opt.on("--guess-time=NUM", "load lines this time for guess. no guess if under 0 (default #{DEFAULT_GUESS_TIME})") do |arg|
+          @guess_time = arg.to_i
+        end
       end
 
       def head
@@ -38,7 +38,10 @@ module Kor
         else
           while line = @io.gets
             line.strip!
-            @ltsvs << LTSV.parse(line)
+            @ltsvs << parse_line(line)
+            if 0 < @guess_time && @guess_time <= @ltsvs.length
+              break
+            end
           end
           @keys = @ltsvs.map { |ltsv| ltsv.keys }
           @keys.flatten!.uniq!
@@ -50,19 +53,35 @@ module Kor
         if @prekeys
           if line = @io.gets
             line.strip!
-            ltsv = LTSV.parse(line)
+            ltsv = parse_line(line)
             @keys.map { |k| ltsv[k] }
           else
             nil
           end
-        else
-          begin
-            @fiber.resume
-          rescue FiberError
-            nil
+        elsif 0 < @guess_time
+          if @count < @guess_time
+            @count += 1
+            return resume
           end
+          if line = @io.gets
+            line.strip!
+            ltsv = parse_line(line)
+            @keys.map { |k| ltsv[k] }
+          end
+        else
+          resume
         end
       end
+
+      def parse_line(line)
+        LTSV.parse(line, symbolize_keys: false).first
+      end
+
+      def resume
+        @fiber.resume
+      rescue FiberError
+        nil
+      end
     end
 
     require "kor/input/ltsv/version"

data/lib/kor/input/ltsv/version.rb

@@ -1,7 +1,7 @@
 module Kor
   module Input
     class Ltsv
-      VERSION = "0.0.2"
+      VERSION = "0.1.0"
     end
   end
 end

data/lib/kor/input/ltsv_test.rb

@@ -5,6 +5,10 @@ module KorInputLtsvTest
     ltsv = <<-LTSV
 foo:100\tbar:200\tbaz:qux:300
 \tbar:500\tfoo:600
+foo:110\tbar:120\tbaz:130
+foo:140\tbar:150\tbaz:160
+foo:170\tbar:180\tbaz:190
+foo:200\tbar:210\tbaz:220
 LTSV
 
     actual = `echo '#{ltsv.chomp}' | kor ltsv csv`
@@ -12,16 +16,34 @@ LTSV
 foo,bar,baz
 100,200,qux:300
 600,500,
+110,120,130
+140,150,160
+170,180,190
+200,210,220
 CSV
     if actual != expect
       t.error("expect #{expect} got #{actual}")
     end
 
-    actual = `echo '#{ltsv.chomp}' | kor ltsv --key=bar,foo csv`
+    actual = `echo '#{ltsv.chomp}' | kor ltsv --guess-time=100 csv`
+    if actual != expect
+      t.error("expect #{expect} got #{actual}")
+    end
+
+    actual = `echo '#{ltsv.chomp}' | kor ltsv --guess-time=0 csv`
+    if actual != expect
+      t.error("expect #{expect} got #{actual}")
+    end
+
+    actual = `echo '#{ltsv.chomp}' | kor ltsv --key=bar,foo,non csv`
     expect = <<-CSV
-bar,foo
-200,100
-500,600
+bar,foo,non
+200,100,
+500,600,
+120,110,
+150,140,
+180,170,
+210,200,
 CSV
     if actual != expect
       t.error("expect #{expect} got #{actual}")

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: kor-input-ltsv
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.1.0
 platform: ruby
 authors:
 - ksss
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-10-09 00:00:00.000000000 Z
+date: 2015-10-10 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: kor
@@ -24,6 +24,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: ltsv
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement