Merge pull request #103 from Guldoman/force_syntax

Add `force_syntax`

2 files changed, 133 insertions, 0 deletions

diff --git a/README.md b/README.md
index 779d1c0..971f8e6 100644
--- a/README.md
+++ b/README.md
@@ -36,6 +36,7 @@ Plugin | Description
 [`ephemeraldocviews`](plugins/ephemeraldocviews.lua?raw=1) | Preview tabs. Opening a doc will replace the contents of the preview tab. Marks tabs as non-preview on any change.
 [`fallbackfonts`](https://github.com/takase1121/lite-fallback-fonts)* | Adds support for fallback fonts *([gif](https://raw.githubusercontent.com/takase1121/lite-fallback-fonts/master/assets/Iw18fI57J0.gif))*
 [`fontconfig`](plugins/fontconfig.lua?raw=1) | Allows users to load fonts with [fontconfig](https://www.freedesktop.org/software/fontconfig/fontconfig-user.html).
+*[`force_syntax`](plugins/force_syntax.lua?raw=1)* | Change the syntax used for a file.
 [`formatter`](https://github.com/vincens2005/lite-formatters)* | formatters for various languages
 [`ghmarkdown`](plugins/ghmarkdown.lua?raw=1) | Opens a preview of the current markdown file in a browser window *([screenshot](https://user-images.githubusercontent.com/3920290/82754898-f7394600-9dc7-11ea-8278-2305363ed372.png))*
 *[`gitdiff_highlight`](https://github.com/vincens2005/lite-xl-gitdiff-highlight)** | highlight changed lines from git *([screenshot](https://raw.githubusercontent.com/vincens2005/lite-xl-gitdiff-highlight/master/screenshot.png))*
diff --git a/plugins/force_syntax.lua b/plugins/force_syntax.lua
new file mode 100644
index 0000000..dce4abc
--- /dev/null
+++ b/plugins/force_syntax.lua
@@ -0,0 +1,132 @@
+-- mod-version:2 -- lite-xl 2.0
+local core = require "core"
+local Doc = require "core.doc"
+local syntax = require "core.syntax"
+local command = require "core.command"
+local common = require "core.common"
+local style = require "core.style"
+local StatusView = require "core.statusview"
+local DocView = require "core.docview"
+
+local function doc()
+  if core.active_view and getmetatable(core.active_view) == DocView then return core.active_view.doc end
+  if core.last_active_view and getmetatable(core.last_active_view) == DocView then return core.last_active_view.doc end
+end
+
+-- Force plaintext syntax to have a name
+local plain_text_syntax = syntax.get("", "")
+plain_text_syntax.name = plain_text_syntax.name or "Plain Text"
+
+local doc_reset_syntax = Doc.reset_syntax
+function Doc:reset_syntax()
+  local syntax_get = syntax.get
+  if self.force_syntax then
+    syntax.get = function() return self.force_syntax end
+  end
+  doc_reset_syntax(self)
+  syntax.get = syntax_get
+end
+
+local function get_syntax_name(s)
+  if not s then return "Undefined" end
+  local name = s.name
+  if not name then
+    local exts = type(s.files) == "string" and { s.files } or s.files
+    if exts then
+      name = table.concat(exts, ", ")
+    end
+  end
+  return name or "Undefined"
+end
+
+local statusview_get_items = StatusView.get_items
+function StatusView:get_items()
+  local left, right = statusview_get_items(self)
+
+  local is_dv = core.active_view and getmetatable(core.active_view) == DocView
+  if not is_dv then return left, right end
+
+  local syntax_name = get_syntax_name(doc().syntax)
+
+  local ins = {
+    style.dim, 
+    self.separator2,
+    style.text,
+    syntax_name
+  }
+
+  if syntax_name then
+    for _,item in pairs(ins) do
+      table.insert(right, item)
+    end
+  end
+
+  return left, right
+end
+
+local function get_syntax_list()
+  local pt_name = plain_text_syntax.name
+  if doc().syntax == plain_text_syntax then
+    pt_name = "Current: "..pt_name
+  end
+  local list = { ["Auto detect"] = false,
+                 [pt_name] = plain_text_syntax }
+  local keylist = { "Auto detect", pt_name }
+
+  for _,s in pairs(syntax.items) do
+    local name = get_syntax_name(s)
+    local fullname = name
+    local i = 1
+    while list[fullname] do
+      i = i + 1
+      fullname = name.." ("..i..")"
+    end
+    if doc().syntax == s then
+      fullname = "Current: "..fullname
+    end
+    list[fullname] = s
+    table.insert(keylist, fullname)
+  end
+
+  return list, keylist
+end
+
+local function sorter(a, b)
+  -- Compare only syntax name
+  a = a:gsub("Current: ", "")
+  b = b:gsub("Current: ", "")
+  return string.upper(a) < string.upper(b)
+end
+
+local function bias_sorter(a, b)
+  -- Bias towards Current and Auto detect syntax
+  if a:match("Current: ") then return true end
+  if b:match("Current: ") then return false end
+  if a:match("Auto detect") then return true end
+  if b:match("Auto detect") then return false end
+  return sorter(a, b)
+end
+
+command.add("core.docview", {
+  ["force-syntax:select-file-syntax"] =
+    function()
+      core.command_view:enter(
+        "Set syntax for this file",
+        function(text, item) -- submit
+          local list, _ = get_syntax_list()
+          doc().force_syntax = list[item.text]
+          doc():reset_syntax()
+        end,
+        function(text) -- suggest
+          local _, keylist = get_syntax_list()
+          local res = common.fuzzy_match(keylist, text)
+          -- Force Current and Auto detect syntax to the bottom
+          -- if the text is empty
+          table.sort(res, #text == 0 and bias_sorter or sorter)
+          return res
+        end,
+        nil, -- cancel
+        nil -- validate
+      )
+    end
+})