Merge pull request #78 from jdhao/theme

refactor: theme.vim --> theme.lua

2 files changed, 123 insertions, 99 deletions

diff --git a/core/themes.lua b/core/themes.lua
new file mode 100644
index 0000000..1a42828
--- /dev/null
+++ b/core/themes.lua
@@ -0,0 +1,123 @@
+local utils = require('utils')
+
+local M = {}
+
+-- Theme to directory name mapping, because theme repo name is not necessarily
+-- the same as the theme name itself.
+M.theme2dir = {
+  gruvbox8 = "vim-gruvbox8",
+  onedark = 'onedark.nvim',
+  edge = 'edge',
+  sonokai = 'sonokai',
+  gruvbox_material = 'gruvbox-material',
+  nord = 'nord.nvim',
+  doom_one = 'doom-one.nvim',
+  everforest = 'everforest',
+  nightfox = 'nightfox.nvim',
+  kanagawa = 'kanagawa.nvim',
+  catppuccin = 'catppuccin'
+}
+
+M.gruvbox8 = function()
+  -- Italic options should be put before colorscheme setting,
+  -- see https://github.com/morhetz/gruvbox/wiki/Terminal-specific#1-italics-is-disabled
+  vim.g.gruvbox_italics = 1
+  vim.g.gruvbox_italicize_strings = 1
+  vim.g.gruvbox_filetype_hi_groups = 1
+  vim.g.gruvbox_plugin_hi_groups = 1
+
+  vim.cmd [[colorscheme gruvbox8_hard]]
+end
+
+M.onedark = function()
+  vim.cmd [[colorscheme onedark]]
+end
+
+M.edge = function()
+  vim.g.edge_enable_italic = 1
+  vim.g.edge_better_performance = 1
+
+  vim.cmd [[colorscheme edge]]
+end
+
+M.sonokai = function()
+  vim.g.sonokai_enable_italic = 1
+  vim.g.sonokai_better_performance = 1
+
+  vim.cmd [[colorscheme sonokai]]
+end
+
+M.gruvbox_material = function()
+  vim.g.gruvbox_material_enable_italic = 1
+  vim.g.gruvbox_material_better_performance = 1
+
+  vim.cmd [[colorscheme gruvbox-material]]
+end
+
+M.nord = function()
+  vim.cmd [[colorscheme nord]]
+end
+
+M.doom_one = function()
+
+  vim.cmd [[colorscheme doom-one]]
+end
+
+M.everforest = function()
+  vim.g.everforest_enable_italic = 1
+  vim.g.everforest_better_performance = 1
+
+  vim.cmd [[colorscheme everforest]]
+end
+
+M.nightfox = function()
+  vim.cmd [[colorscheme nordfox]]
+end
+
+
+M.kanagawa = function()
+  vim.cmd [[colorscheme kanagawa]]
+end
+
+M.catppuccin = function()
+  -- available option: latte, frappe, macchiato, mocha
+  vim.g.catppuccin_flavour = "frappe"
+
+  require("catppuccin").setup()
+
+  vim.cmd [[colorscheme catppuccin]]
+end
+
+
+--- Use a random theme from the pre-defined list of themes.
+M.rand_theme = function ()
+  local theme = utils.rand_element(vim.tbl_keys(M.theme2dir))
+
+  if not vim.tbl_contains(vim.tbl_keys(M), theme) then
+    local msg = "Invalid theme: " .. theme
+    vim.notify(msg, vim.log.levels.ERROR, { title = 'nvim-config' })
+
+    return
+  end
+
+  -- Load the theme, because all the themes are declared as opt plugins, the theme isn't loaded yet.
+  local status = utils.add_pack(M.theme2dir[theme])
+
+  if not status then
+    local msg = string.format("Theme %s is not installed. Run PackerSync to install.", theme)
+    vim.notify(msg, vim.log.levels.ERROR, { title = 'nvim-config' })
+
+    return
+  end
+
+  -- Set the theme.
+  M[theme]()
+
+  if vim.g.logging_level == 'debug' then
+    local msg = "Colorscheme: " .. theme
+
+    vim.notify(msg, vim.log.levels.DEBUG, { title = 'nvim-config' })
+  end
+end
+
+M.rand_theme()
diff --git a/core/themes.vim b/core/themes.vim
deleted file mode 100644
index 17b9ca2..0000000
--- a/core/themes.vim
+++ /dev/null
@@ -1,99 +0,0 @@
-let s:theme_setup_dict = {}
-
-function! s:theme_setup_dict.gruvbox8() dict abort
-  " Italic options should be put before colorscheme setting,
-  " see https://github.com/morhetz/gruvbox/wiki/Terminal-specific#1-italics-is-disabled
-  let g:gruvbox_italics=1
-  let g:gruvbox_italicize_strings=1
-  let g:gruvbox_filetype_hi_groups = 1
-  let g:gruvbox_plugin_hi_groups = 1
-  colorscheme gruvbox8_hard
-endfunction
-
-function! s:theme_setup_dict.onedark() dict abort
-  colorscheme onedark
-endfunction
-
-function! s:theme_setup_dict.edge() dict abort
-  let g:edge_enable_italic = 1
-  let g:edge_better_performance = 1
-  colorscheme edge
-endfunction
-
-function! s:theme_setup_dict.sonokai() dict abort
-  let g:sonokai_enable_italic = 1
-  let g:sonokai_better_performance = 1
-  colorscheme sonokai
-endfunction
-
-function! s:theme_setup_dict.gruvbox_material() dict abort
-  let g:gruvbox_material_enable_italic = 1
-  let g:gruvbox_material_better_performance = 1
-  colorscheme gruvbox-material
-endfunction
-
-function! s:theme_setup_dict.nord() dict abort
-  colorscheme nord
-endfunction
-
-function! s:theme_setup_dict.doom_one() dict abort
-  colorscheme doom-one
-endfunction
-
-function! s:theme_setup_dict.everforest() dict abort
-  let g:everforest_enable_italic = 1
-  let g:everforest_better_performance = 1
-  colorscheme everforest
-endfunction
-
-function! s:theme_setup_dict.nightfox() dict abort
-  colorscheme nordfox
-endfunction
-
-function! s:theme_setup_dict.kanagawa() dict abort
-  colorscheme kanagawa
-endfunction
-
-function! s:theme_setup_dict.catppuccin() dict abort
-  let g:catppuccin_flavour = "frappe" " latte, frappe, macchiato, mocha
-
-  lua require("catppuccin").setup()
-  colorscheme catppuccin
-endfunction
-
-" Theme to directory name mapping, because theme repo name is not necessarily
-" the same as the theme name itself.
-let s:theme2dir = {
-      \ 'gruvbox8' : 'vim-gruvbox8',
-      \ 'onedark': 'onedark.nvim',
-      \ 'edge' : 'edge',
-      \ 'sonokai': 'sonokai',
-      \ 'gruvbox_material': 'gruvbox-material',
-      \ 'nord': 'nord.nvim',
-      \ 'doom_one': 'doom-one.nvim',
-      \ 'everforest' :'everforest',
-      \ 'nightfox': 'nightfox.nvim',
-      \ 'kanagawa': 'kanagawa.nvim',
-      \ 'catppuccin': 'catppuccin'
-      \ }
-
-let s:theme = utils#RandElement(keys(s:theme2dir))
-let s:colorscheme_func = printf('s:theme_setup_dict.%s()', s:theme)
-
-if !has_key(s:theme_setup_dict, s:theme)
-  let s:msg = "Invalid colorscheme function: " . s:colorscheme_func
-  call v:lua.vim.notify(s:msg, 'error', {'title': 'nvim-config'})
-  finish
-endif
-
-let s:status = utils#add_pack(s:theme2dir[s:theme])
-if !s:status
-  echomsg printf("Theme %s not installed. Run PackerSync to install.", s:theme)
-  finish
-endif
-
-execute 'call ' . s:colorscheme_func
-if g:logging_level == 'debug'
-  let s:msg1 = "Colorscheme: " . s:theme
-  call v:lua.vim.notify(s:msg1, 'info', {'title': 'nvim-config'})
-endif