codes21 0.1.2 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 59be62698687bfb3685a65c7249b1def4625f3c2
-  data.tar.gz: 8ef2fe9f0d837e7df644c9ce73c6ca287f3fe7c4
+  metadata.gz: 555c03e7f89ac0070403ba3dbf63feabe0b00517
+  data.tar.gz: f7c2a11c2519b46dab69d540e5b3ba87089c7736
 SHA512:
-  metadata.gz: 09edf309e6c207cd210fe73a71663c7191aa10665cae18aef1739e978cccea517abee37cdf50b0b6711da9bd26bd13f145aac4133a663a4e995b0a89b92c855e
-  data.tar.gz: 0185fdba1a7ac70b93d9b22ddc3061fb6aecb5be477eb21e02ec42cdbbce78c0fd8aaab1afca6d38f0d4d99fd7b9942ab8babaebcb719ab2ce84faad1c910606
+  metadata.gz: 859345aca9cdac4a6968cfb2e99eafb633d197d9ea041b4c5671d4d616d86c134b9097beef0e0232422a79a24a71867ccedfa232cc3563cef18f6fd3658fd203
+  data.tar.gz: ec14c060a36f175971f7db0f7d2c767eafb8e9336f45766825a62f730cd581a964aafcf992aed15a78d5111682288e8704f157bc7ae4f6ba6bf4d6e1e5464dd2

data/bin/codes21

@@ -1,10 +1,15 @@
 #!/usr/bin/env ruby
 
+require 'optparse'
 require_relative '../lib/codes21'
 
-if ARGV.count != 3
-  puts "With the given suffixes merge multiple files code into one file."
-  puts "usage: codes21 /path/to/project rb,js result.txt"
+help = <<-HELP
+With the given suffixes merge multiple files code into one file.
+usage: codes21 /path/to/project rb,js result.txt [-i UTF-8] [-o UTF-8]
+HELP
+
+if ARGV.count < 3
+  puts help
   exit
 end
 
@@ -12,4 +17,23 @@ path = ARGV[0]
 suffixes = ARGV[1]
 output = ARGV[2]
 
-Codes21.run path, suffixes, output
+input_coding = "UTF-8"
+output_coding = "UTF-8"
+
+OptionParser.new do |opts|
+  opts.banner = "usage: codes21 /path/to/project rb,js result.txt [-i=UTF-8] [-o=UTF-8]"
+  opts.on("-iENCODE", "Input file coding") do |coding|
+    input_coding = coding
+  end
+
+  opts.on("-oENCODE", "Output file coding") do |coding|
+    output_coding = coding
+  end
+
+  opts.on("-h") do
+    exit
+  end
+
+end.parse!
+
+Codes21.run path, suffixes, output, input_coding, output_coding

data/lib/codes21.rb

@@ -1,12 +1,12 @@
 class Codes21
-  def get_codes(path, suffixes)
+  def get_codes(path, suffixes, coding="UTF-8")
     @items ||= []
     Dir.glob("#{path}/**/*.{#{suffixes}}").each do |file|
-      title = file.split('/').last
+      title = file.split('/').last.force_encoding(coding)
       puts file
 
       puts '*' * 70
-      codes = File.read(file)
+      codes = File.read(file).force_encoding(coding)
       # sb的软件著作权要求不能有空行
       codes = codes.squeeze("\n")
       puts codes
@@ -17,12 +17,17 @@ class Codes21
     @items
   end
 
-  def write_result(result_file_path)
+  def write_result(result_file_path, coding="UTF-8")
     File.open(result_file_path, 'w') do |f|
       @items.each do |item|
-        f.puts "File:" + item[:title]
-        f.puts "Code:"
-        f.puts item[:codes]
+        title = "File:" + item[:title]
+        code_label = "Code:"
+        codes = item[:codes]
+
+        content = title + "\n" + code_label + "\n" + codes + "\n"
+        content = content.squeeze("\n")
+
+        f.print content.encode(coding)
       end
     end
 
@@ -30,10 +35,10 @@ class Codes21
     puts "codes files count:  #{@items.size}"
   end
 
-  def self.run(path, suffixes, result_file_path)
+  def self.run(path, suffixes, result_file_path, input_coding="UTF-8", output_coding="UTF-8")
     codes21 = Codes21.new
-    codes21.get_codes(path, suffixes)
-    codes21.write_result(result_file_path)
+    codes21.get_codes(path, suffixes, input_coding)
+    codes21.write_result(result_file_path, output_coding)
   end
 
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: codes21
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.2.0
 platform: ruby
 authors:
 - Aston Fu