ecdict 1.2.alpha2 → 1.2.alpha5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 473f0731744d4a659730af33b6f716f0b84469b1ad00d209b84cadc5993113e4
-  data.tar.gz: 59dc212296d826b0b64f44b8167bed14024db904cc50fb58ed17efc49d1814ed
+  metadata.gz: 5d3b7af98d6ca07e5b9228c8ed515af5250f194b74ff1b5b8a15cc4a53418d48
+  data.tar.gz: 0ac17588afaa66aaddc89b65a870ccebf7d8318dfdc8526f1d2ba804d4624897
 SHA512:
-  metadata.gz: 1d049b3ced8252ca2cece8f46c797bcff109ea32c4762d21ff401aab071e2f0f71643bb1c22d0224dc6255440dda08ac0a0f317b0b45b05a67c1b368c0ef24b1
-  data.tar.gz: 235716b3c4589ae628ec798ef7b367d4a78aedd663182410b88be38872e564afb95bc1509b62e35e46f33f9509f0209c0ca3934961aa7ce70072e19e95ca09d1
+  metadata.gz: f8e715abe0412e94f79739a43cbabd063211d50977c4d7d8dbb49851cdedfd51f9ab4847b91d046a817c64c240baff591a37fd00d7ea45ee13c119c9afce61c8
+  data.tar.gz: ab5ec984d53c633ce72f5a557e5f7f7aab91f395cf24d8706562bbeef5a4efd9d3c653879a28170a0b6ea1dc1b05c819022656e29b97c8977e734a5508940d0e

data/bin/ecdict

@@ -15,16 +15,25 @@
 #   1. ecdict installer
 #   2. ecdict cmd
 #   3. ecdict repl
+#
+# Exit -1: 下载原始数据失败
+# Exit -2: 解压原始数据至CSV失败
+# EXit -3: 从CSV生成数据库失败
+# Exit  1: 无词典
+# Exit  2: 缺少参数
+#
+# 未搜索到结果并不认为是错误
 # ------------------------------------------------------
 
 require 'sqlite3'
 
-ECDICT_STORAGE = File.expand_path("~/.local/share/ecdict")
-ECDictRAWDATA  = File.join(ECDICT_STORAGE, 'ecdict-csv.7z')
+ECDICT_STORAGE   = File.expand_path("~/.local/share/ecdict")
+ECDICT_RAW_DATA  = File.join(ECDICT_STORAGE, 'ecdict-csv.7z')
+
 ECDictDB  = File.join(ECDICT_STORAGE, 'ecdict.db')
 ECDictCSV = File.join(ECDICT_STORAGE, 'ecdict.csv')
 
-ECDICT_GEM_VERSION  = "1.2.alpha2"
+ECDICT_GEM_VERSION  = "1.2.alpha5"
 
 # 这个版本是我从上游下载的CSV后自己压缩的
 # 这个文件的实际创建日期(上游作者创建)是2017/06/04
@@ -38,6 +47,16 @@ require 'fileutils'
 FileUtils.mkdir_p(ECDICT_STORAGE)
 $DB = SQLite3::Database.new ECDictDB
 
+def close_db
+  $DB.close if $DB
+end
+
+def close_db_and_exit(num = 0)
+  close_db
+  exit num
+end
+
+
 
 ####################
 # helper: for color
@@ -65,8 +84,8 @@ def cyan(str)       "\e[36m#{str}\e[0m" end
 
 def check_download_integrity
   require 'digest'
-  cont = File.read(ECDictRAWDATA)
-  sha256 = Digest::SHA256.hexdigest(cont)
+  return false if !File.exists?(ECDICT_RAW_DATA)
+  sha256 = Digest::SHA256.file(ECDICT_RAW_DATA).hexdigest
   if ECDICT_DICT_SHA256 == sha256
     true
   else
@@ -75,31 +94,47 @@ def check_download_integrity
 end
 
 
-def download_ecdict_data
+def download_ecdict_raw_data
 
-  if File.exists?(ECDictRAWDATA) && check_download_integrity
-    puts "=> 已存在原始数据文件#{ECDictRAWDATA}, 无需再次下载"
+  if File.exists?(ECDICT_RAW_DATA) && check_download_integrity
+    puts "=> 已存在原始数据文件#{ECDICT_RAW_DATA}, 无需再次下载"
   else
 
-    if File.exists?(ECDictRAWDATA)
-      puts "=> 删除已存在但不完整的数据"
-      FileUtils.rm ECDictRAWDATA
+    if File.exists?(ECDICT_RAW_DATA)
+      puts "=> 删除已存在但不完整的原始数据文件"
+      FileUtils.rm ECDICT_RAW_DATA
+    end
+
+    # 若已经有拉取的仓库，先检查里面的数据是否可用
+    raw_data_git_dir = File.join(ECDICT_STORAGE, 'ecdict-data')
+    raw_data_git_dir_data = File.join(raw_data_git_dir, 'ecdict-csv.7z')
+
+    if Dir.exists?(raw_data_git_dir)
+      if File.exists?(raw_data_git_dir_data)
+        FileUtils.cp(raw_data_git_dir_data, ECDICT_STORAGE)
+        if check_download_integrity
+          puts "=> 从已经Git pull到的仓库中获取原始数据文件"
+          return true
+        else
+          FileUtils.rm ECDICT_RAW_DATA
+        end
+      end
+      FileUtils.rm_rf(raw_data_git_dir)
     end
 
     begin
       puts "=> 使用Git从#{ECDICT_DICT_DOWNLOAD_LINK}获取原始数据库文件"
-      system("git -C #{ECDICT_STORAGE} clone #{ECDICT_DICT_DOWNLOAD_LINK} ")
-      FileUtils.cp(File.join(ECDICT_STORAGE, 'ecdict-data', 'ecdict-csv.7z'), ECDICT_STORAGE)
-      raise "Git拉取仓库数据不完整!" if !check_download_integrity
+      ret = system("git -C #{ECDICT_STORAGE} clone #{ECDICT_DICT_DOWNLOAD_LINK} ")
+      FileUtils.cp(File.join(ECDICT_STORAGE, 'ecdict-data', 'ecdict-csv.7z'), ECDICT_STORAGE) rescue nil
+      raise "Git拉取仓库过程失败，原始数据文件不完整!" if (ret != true  || !check_download_integrity)
     rescue StandardError => e
       puts "=> #{e.message}"
-      puts "=> Git获取原始数据库文件失败"
-      FileUtils.rm(ECDictRAWDATA)
+      FileUtils.rm(ECDICT_RAW_DATA) rescue nil
+      return false
     else
       puts "=> 下载完成!"
     end
   end
-
   true
 end
 
@@ -119,12 +154,12 @@ def decompress_7z_to_csv()
   end
 
 
-  File.open(ECDictRAWDATA, "rb") do |file|
-    puts "=> 正在解压ecdict-csv.7z(#{file.size/1024/1024}MB)"
+  File.open(ECDICT_RAW_DATA, "rb") do |file|
+    puts "=> 正在解压ecdict-csv.7z(#{ '%.2f' % (file.size/1024.0/1024) }MB)"
     SevenZipRuby::Reader.extract_all(file, ECDICT_STORAGE)
   end
 
-  puts "=> 已成功解压出ecdict.csv(#{File.size(ECDictCSV)/1024/1024}MB)"
+  puts "=> 已成功解压出ecdict.csv(#{ '%.2f' % (File.size(ECDictCSV)/1024.0/1024) }MB)"
   true
 end
 
@@ -153,9 +188,6 @@ end
 #   2. create table in the db
 #   3. insert data from CSV into the table
 #
-# exit 1 : create db failed
-# exit 2 : insert data failed
-#
 def generate_sqlite_db()
 
   # SQLite3::Database.new已经生成了该文件,所以需要提前判断到底是否存在
@@ -172,9 +204,6 @@ def generate_sqlite_db()
     end
   end
 
-  def db_close
-    $DB.close if $DB
-  end
 
   # 共15个字段,存的时候只需要14个
   # 1.id    	    主键，自增
@@ -224,9 +253,9 @@ EOF
   rescue Exception => e
     puts "=> #{e.message}"
     puts "=> 创建数据库文件#{ECDictDB}失败，或表`ecdict`生成失败"
-    db_close
+    close_db
     File.delete(ECDictDB)
-    exit 1
+    return false
   end
 
 
@@ -289,24 +318,28 @@ EOF
   rescue Exception => e
     puts "=> #{e.message}"
     puts "=> 数据库插入信息失败"
+    close_db
     File.delete(ECDictDB)
-    db_close
-    exit 2
+    return false
   end
   puts
   puts "=> 数据库#{ECDictDB}已完整生成"
-  db_close
+  close_db
+  return true
 end
 
 
 def download_and_install_ecdict_data
-  download_ecdict_data || ( puts("\e[31m=> 下载失败，安装取消！\e[0m") || exit(-1) )
+  download_ecdict_raw_data || ( puts("\e[31m=> 下载失败，安装取消！\e[0m") || exit(-1) )
   decompress_7z_to_csv || ( puts("\e[31m=> 解压失败，安装取消！\e[0m") || exit(-2) )
   generate_sqlite_db   || ( puts("\e[31m=> 生成数据库失败，安装取消!\e[0m") || exit(-3) )
   puts
-  puts "=> Hooray! 词典数据下载安装完成! 请尝试使用:"
-  puts "=>   1. ecdict word  查询单词 "
-  puts "=>   2. ecdict -r    进入交互式查询REPL"
+  puts "Hooray! 词典数据下载安装完成! 请尝试使用:\n\n"
+  puts "    1. ecdict word      查询单词 "
+  puts "    2. ecdict -r        进入交互式查询REPL"
+  puts "    3. ecdict -c  中文  搜索中文单词"
+  puts "    4. ecdict -cp 中文  搜索中文短语"
+  puts
 end
 
 
@@ -326,8 +359,8 @@ end
 def reply_once(query)
 
   if !check_db_integrity
-    puts "ecdict: 数据库不存在或数据库不完整，请使用ecdict -i来下载安装词典数据"
-    exit -1
+    puts "ecdict: 数据库不存在或数据库不完整，请使用`ecdict -i`来下载安装词典数据"
+    close_db_and_exit(1)
   end
 
   tables = []
@@ -346,8 +379,9 @@ def reply_once(query)
   end
 
   if rows.empty?
-    puts "ecdict: 未找到结果,请检查单词拼写是否正确",""
-    exit 2
+    puts "ecdict: 未找到结果,请检查单词拼写是否正确"
+    close_db
+    return
   else
     line = rows[0]
     phonetic,trans,definition,exchange = line[0],line[1],line[2],line[3]
@@ -393,11 +427,11 @@ def chinese_search(cn_word, support_phrase: false)
 
   if cn_word.empty?
     puts "ecdict: 请输入要查询的中文，或使用`ecdict -h`查看帮助"
-    exit 1
+    close_db_and_exit(2)
   end
   if !check_db_integrity
     puts "=> 数据库不存在或数据库不完整，请使用`ecdict -i`来下载安装词典数据"
-    exit -1
+    close_db_and_exit(1)
   end
 
   if support_phrase
@@ -418,7 +452,8 @@ def chinese_search(cn_word, support_phrase: false)
   end
   if rows.empty?
     puts "ecdict: 未找到相关英文,请不要输入中文短语或短句，仅支持词的搜索"
-    exit 2
+    close_db
+    return
   else
     # 有的时候并不是没有结果，而是被我们过滤掉了，对此确认一下
     found_a_word = false
@@ -471,7 +506,8 @@ def chinese_search(cn_word, support_phrase: false)
     if found_a_word
       puts
     else
-      puts "ecdict: 抱歉，为了防止显示内容过多，已将可能的答案过滤了",""
+      puts "ecdict: 抱歉，为了防止显示内容过多，已将可能的答案过滤了"
+      puts "ecdict: 您可尝试使用`ecdict -cp`来搜索短语，扩大搜索范围"
     end
   # end of else
   end
@@ -541,12 +577,7 @@ def start_ecrepl
 
   if !check_db_integrity
     puts "ecdict: 数据库不存在或数据库不完整，请使用`ecdict -i`来下载安装词典数据"
-    exit -1
-  end
-
-  def db_close_and_exit
-    $DB.close if $DB
-    exit 0
+    close_db_and_exit(1)
   end
 
   begin
@@ -561,7 +592,7 @@ def start_ecrepl
   require 'reline'
   Reline.completion_proc = lambda do |word|
     if word.strip.empty?
-      return %w{1.输入单词查询 2.输入exit或按Ctrl-C退出}
+      return %w{1.输入单词并回车查询含义 2.输入单词时按一次tab键反馈搜索建议 3.输入exit或按Ctrl-C或Ctrl-D退出}
     end
 
     max_len = word.length + 4
@@ -604,7 +635,8 @@ def start_ecrepl
     while line = Reline.readline("\e[32mecdict> \e[0m", true)
       case word = line.chomp
       when 'exit'
-        db_close_and_exit
+        close_db
+        return
       when ''
         # NOOP
       else
@@ -612,9 +644,10 @@ def start_ecrepl
       end
     end
   rescue Interrupt
-    puts '^C'
+    # puts '^C'
     `stty #{stty_save}` if stty_save
-    db_close_and_exit
+    close_db
+    return
   end
 
 end
@@ -624,14 +657,14 @@ end
 # remove everything in #{ECDICT_STORAGE}
 #
 def delete_cache
-
+  close_db
   begin
     FileUtils.rm_rf(ECDICT_STORAGE)
   rescue => e
-    puts "=> #{e.message}"
-    puts "=> 清空词典数据失败"
+    puts "ecdict: #{e.message}"
+    puts "ecdict: 清空词典数据失败"
   else
-    puts "=> 清空词典数据完成"
+    puts "ecdict: 清空词典数据完成"
   end
   puts
 end
@@ -644,8 +677,6 @@ end
 def print_version
   puts <<EOH
 ecdict (v#{ECDICT_GEM_VERSION}): A courteous cli translator.
-
-
 EOH
 
 end
@@ -657,12 +688,12 @@ ecdict (v#{ECDICT_GEM_VERSION}): A courteous cli translator.
 
 usage:
 
-  ecdict word      => 查询单词 word
-  ecdict -r        => 启动ecdict repl交互式查询，输入exit或Ctrl-C退出
-  ecdict -c  中文  => 搜索中文对应的英文单词
-  ecdict -cp 中文  => 搜索中文对应的英文单词以及短语(结果可能很多)
+  ecdict word      => 查询单词word
+  ecdict -r        => 启动ecdict repl交互式查询，输入exit或Ctrl-C/D退出
+  ecdict -c  中文  => 搜索中文单词对应的英文单词
+  ecdict -cp 中文  => 搜索中文短语对应的英文短语
   ecdict -i        => 下载ecdict词典数据并安装
-  ecdict -v        => 打印此Gem与词典版本号
+  ecdict -v        => 打印此Gem版本号
   ecdict -h        => 打印此帮助
   ecdict -d        => 清空词典数据
 
@@ -678,10 +709,10 @@ end
 if ARGV.empty?
   if !check_db_integrity
     puts "ecdict: 数据库不存在或数据库不完整，请使用`ecdict -i`来下载安装词典数据"
-    exit -1
+    close_db_and_exit(1)
   end
   puts "ecdict: 请输入要查询的词/词组 或使用`ecdict -h`查看帮助"
-  exit 1
+  close_db_and_exit(2)
 end
 
 query = ARGV.shift
@@ -697,3 +728,5 @@ else
   reply_once(query)
 end
 
+# ensure close db
+close_db

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ecdict
 version: !ruby/object:Gem::Version
-  version: 1.2.alpha2
+  version: 1.2.alpha5
 platform: ruby
 authors:
 - ccmywish