From 7772ce55a3cdd5aaf4649f0ad7609fa32162b909 Mon Sep 17 00:00:00 2001
From: JuanZoran <1430359574@qq.com>
Date: Thu, 22 Dec 2022 11:06:49 +0800
Subject: [PATCH] fix: improve install.sh and init.lua

---
 README.md              |  2 +-
 install.sh             |  9 +++++++++
 lua/Trans/database.lua |  2 +-
 lua/Trans/init.lua     | 13 +++++++++++--
 4 files changed, 22 insertions(+), 4 deletions(-)

diff --git a/README.md b/README.md
index 0836661..f8d7950 100644
--- a/README.md
+++ b/README.md
@@ -59,7 +59,7 @@ use {
   - install.sh 下载后会自动将词库解压, 并移动到 `$HOME/.vim/dict`文件夹下
   - 目前仅在 `Ubuntu22.04`的环境下测试通过
     > 如果上述条件不符合, 请删掉 `run = 'install.sh'`部分, 考虑手动安装词库
-    > 如果上述条件满足, 仍出现问题, 欢迎在issure里向我反馈,我会及时尝试解决
+    > 如果上述条件满足, 仍出现问题, 欢迎在issue里向我反馈,我会及时尝试解决
 
 - 下载词典的过程中, 需要能够 `流畅的访问github下载`
     > 词库文件压缩包大小为: **281M**
diff --git a/install.sh b/install.sh
index 82e79d8..a939dce 100755
--- a/install.sh
+++ b/install.sh
@@ -1,6 +1,15 @@
 #!/bin/bash
 set -e
 
+if [[ -d "$HOME/.vim" ]]; then
+    if [[ -d "$HOME/.vim/dict" && -f "$HOME/.vim/dict/ultimate.db" ]]; then
+        echo 'db has been installed'
+        exit 0
+    fi
+else
+    echo '$HOME/.vim not exist'
+    exit 1
+fi
 wget https://github.com/skywind3000/ECDICT-ultimate/releases/download/1.0.0/ecdict-ultimate-sqlite.zip -O /tmp/dict.zip
 
 unzip /tmp/dict.zip -d $HOME/.vim/dict
diff --git a/lua/Trans/database.lua b/lua/Trans/database.lua
index 092e451..4b3984f 100644
--- a/lua/Trans/database.lua
+++ b/lua/Trans/database.lua
@@ -1,5 +1,5 @@
 local M = {}
-local dict = require("Trans").dict
+local dict = require("Trans").db:open(M.conf.db_path)
 
 function M.query(arg)
     -- TODO: return type: a result table:
diff --git a/lua/Trans/init.lua b/lua/Trans/init.lua
index b1117e6..79e2c73 100644
--- a/lua/Trans/init.lua
+++ b/lua/Trans/init.lua
@@ -15,7 +15,16 @@ function M.setup(conf)
     require("Trans.setup")
 end
 
-M.db = require("sqlite.db")
-M.dict = M.db:open(M.conf.db_path)
+local res = vim.fn.executable('sqlite3')
+if res ~= 1 then
+    error('Please check out sqlite3')
+end
+
+local status, db = pcall(require, 'sqlite.db')
+if not status then
+    error('Please check out sqlite.lua')
+end
+
+M.db = db
 
 return M