From 5beedc0a76eafd7d75b94b976376c4e25f71bc48 Mon Sep 17 00:00:00 2001
From: mvllow <mvllow@icloud.com>
Date: Tue, 12 Oct 2021 12:14:41 -0500
Subject: [PATCH] feat: allow colour overrides

closes #10
---
 lua/rose-pine/theme.lua | 61 +++++++++++++++++++++++------------------
 readme.md               | 13 +++++++--
 2 files changed, 46 insertions(+), 28 deletions(-)

diff --git a/lua/rose-pine/theme.lua b/lua/rose-pine/theme.lua
index 6f00830..502ca57 100644
--- a/lua/rose-pine/theme.lua
+++ b/lua/rose-pine/theme.lua
@@ -1,5 +1,14 @@
 local p = require('rose-pine.palette')
 
+local user_colors = vim.g.rose_pine_colors or {}
+local groups = {
+	comment = user_colors.comment or p.subtle,
+	hint = user_colors.hint or p.iris,
+	info = user_colors.info or p.foam,
+	warn = user_colors.warn or p.gold,
+	error = user_colors.error or p.love,
+}
+
 local theme = {}
 local maybe_base = p.base
 local maybe_italic = 'italic'
@@ -76,7 +85,7 @@ theme.base = {
 	Boolean = { fg = p.gold },
 	Character = { fg = p.gold },
 	Comment = {
-		fg = p.subtle,
+		fg = groups.comment,
 		style = maybe_italic,
 	},
 	Conditional = { fg = p.pine },
@@ -119,35 +128,35 @@ theme.base = {
 	htmlTagN = { fg = p.text },
 	htmlTagName = { fg = p.foam },
 
-	DiagnosticHint = { fg = p.iris },
-	DiagnosticInfo = { fg = p.foam },
-	DiagnosticWarn = { fg = p.gold },
-	DiagnosticError = { fg = p.love },
+	DiagnosticHint = { fg = groups.hint },
+	DiagnosticInfo = { fg = groups.info },
+	DiagnosticWarn = { fg = groups.warn },
+	DiagnosticError = { fg = groups.error },
 
-	DiagnosticDefaultHint = { fg = p.iris },
-	DiagnosticDefaultInfo = { fg = p.foam },
-	DiagnosticDefaultWarn = { fg = p.gold },
-	DiagnosticDefaultError = { fg = p.love },
+	DiagnosticDefaultHint = { fg = groups.hint },
+	DiagnosticDefaultInfo = { fg = groups.info },
+	DiagnosticDefaultWarn = { fg = groups.warn },
+	DiagnosticDefaultError = { fg = groups.error },
 
-	DiagnosticFloatingHint = { fg = p.iris },
-	DiagnosticFloatingInfo = { fg = p.foam },
-	DiagnosticFloatingWarn = { fg = p.gold },
-	DiagnosticFloatingError = { fg = p.love },
+	DiagnosticFloatingHint = { fg = groups.hint },
+	DiagnosticFloatingInfo = { fg = groups.info },
+	DiagnosticFloatingWarn = { fg = groups.warn },
+	DiagnosticFloatingError = { fg = groups.error },
 
-	DiagnosticSignHint = { fg = p.iris },
-	DiagnosticSignInfo = { fg = p.foam },
-	DiagnosticSignWarn = { fg = p.gold },
-	DiagnosticSignError = { fg = p.love },
+	DiagnosticSignHint = { fg = groups.hint },
+	DiagnosticSignInfo = { fg = groups.info },
+	DiagnosticSignWarn = { fg = groups.warn },
+	DiagnosticSignError = { fg = groups.error },
 
-	DiagnosticUnderlineHint = { style = 'undercurl', sp = p.iris },
-	DiagnosticUnderlineInfo = { style = 'undercurl', sp = p.foam },
-	DiagnosticUnderlineWarn = { style = 'undercurl', sp = p.gold },
-	DiagnosticUnderlineError = { style = 'undercurl', sp = p.love },
+	DiagnosticUnderlineHint = { style = 'undercurl', sp = groups.hint },
+	DiagnosticUnderlineInfo = { style = 'undercurl', sp = groups.info },
+	DiagnosticUnderlineWarn = { style = 'undercurl', sp = groups.warn },
+	DiagnosticUnderlineError = { style = 'undercurl', sp = groups.error },
 
-	DiagnosticVirtualTextHint = { fg = p.iris },
-	DiagnosticVirtualTextInfo = { fg = p.foam },
-	DiagnosticVirtualTextWarn = { fg = p.gold },
-	DiagnosticVirtualTextError = { fg = p.love },
+	DiagnosticVirtualTextHint = { fg = groups.hint },
+	DiagnosticVirtualTextInfo = { fg = groups.info },
+	DiagnosticVirtualTextWarn = { fg = groups.warn },
+	DiagnosticVirtualTextError = { fg = groups.error },
 
 	LspReferenceText = { fg = p.rose, bg = p.highlight },
 	LspReferenceRead = { fg = p.rose, bg = p.highlight },
@@ -221,7 +230,7 @@ theme.treesitter = {
 	TSBoolean = { fg = p.rose },
 	-- TSCharacter = {},
 	TSComment = {
-		fg = p.subtle,
+		fg = groups.comment,
 		style = maybe_italic,
 	},
 	-- TSConditional = {},
diff --git a/readme.md b/readme.md
index 5da8160..2a596a3 100644
--- a/readme.md
+++ b/readme.md
@@ -46,9 +46,9 @@ local colors = require("galaxyline.themes.colors")["rose-pine"]
 ```
 
 > **IMPORTANT**:
-> 
+>
 > 1. This requires [NTBBloodbath's galaxyline fork](https://github.com/NTBBloodbath/galaxyline.nvim) in order to work.
-> 
+>
 > 2. You can see the list of available colors [here](https://github.com/NTBBloodbath/galaxyline.nvim/blob/main/docs/themes.md#colors-standards).
 
 ## Plugin Support
@@ -97,6 +97,15 @@ vim.g.rose_pine_disable_background = false
 -- Use bold vertical split line
 vim.g.rose_pine_bold_vertical_split_line = true
 
+-- Override theme groups
+vim.g.rose_pine_colors = {
+	comment = '#ffffff',
+  hint = '#9745be',
+  info = '#78ccc5',
+  warn = '#f5c359',
+  error = '#c75c6a',
+}
+
 -- Set colorscheme after options
 vim.cmd('colorscheme rose-pine')
 ```