From dc531951ff6ccb4a2deaaaebfa31f19a0d301f99 Mon Sep 17 00:00:00 2001
From: JuanZoran <1430359574@qq.com>
Date: Mon, 19 Dec 2022 22:23:12 +0800
Subject: [PATCH] fix: can set wrap by setup function

---
 README.md             | 8 ++++++++
 lua/Trans/conf.lua    | 1 +
 lua/Trans/display.lua | 9 ++++++---
 lua/Trans/setup.lua   | 2 ++
 4 files changed, 17 insertions(+), 3 deletions(-)
 create mode 100644 README.md

diff --git a/README.md b/README.md
new file mode 100644
index 0000000..980a2c0
--- /dev/null
+++ b/README.md
@@ -0,0 +1,8 @@
+# TODO
+- 关键字高亮
+- 多风格样式
+- 移动光标自动关闭窗口
+- etc
+
+# FIXME
+- cursor模式下会自动聚焦在新窗口
diff --git a/lua/Trans/conf.lua b/lua/Trans/conf.lua
index 9efeda9..8d4238f 100644
--- a/lua/Trans/conf.lua
+++ b/lua/Trans/conf.lua
@@ -12,6 +12,7 @@ return {
         exchange = true,
         Trans_en = true,
         Trans_zh = true,
+        wrap = false,
     },
     view = {
         -- TODO: style: buffer | cursor | window
diff --git a/lua/Trans/display.lua b/lua/Trans/display.lua
index 169cc65..72f61b0 100644
--- a/lua/Trans/display.lua
+++ b/lua/Trans/display.lua
@@ -25,7 +25,7 @@ vim.api.nvim_buf_set_option(buf, 'filetype', 'Trans')
 --  }
 
 local function show_win(width, height)
-    vim.api.nvim_open_win(buf, true, {
+    local win = vim.api.nvim_open_win(buf, true, {
         relative = 'cursor',
         title = 'Trans',
         title_pos = 'center',
@@ -35,8 +35,7 @@ local function show_win(width, height)
         border = 'rounded',
         focusable = false,
     })
-    -- vim.api.nvim_win_set_option(win, 'warp', true)
-
+    vim.api.nvim_win_set_option(win, 'wrap', display.wrap)
 end
 
 -- @return string array
@@ -138,4 +137,8 @@ function M.query()
     -- TODO:
 end
 
+function M.toggle()
+    -- TODO: wrap some function
+end
+
 return M
diff --git a/lua/Trans/setup.lua b/lua/Trans/setup.lua
index 129346f..f1adcdd 100644
--- a/lua/Trans/setup.lua
+++ b/lua/Trans/setup.lua
@@ -13,3 +13,5 @@ vim.api.nvim_create_autocmd('VimLeave', {
 })
 
 vim.api.nvim_create_user_command('TranslateCurosorWord', require("Trans").query_cursor, {})
+
+-- TODO: set command to close preview window automatically