From 1da4fd1eafaab685e33f21a638f80eb02bebe395 Mon Sep 17 00:00:00 2001
From: Simon Michael <simon@joyful.com>
Date: Thu, 26 Jan 2023 00:30:18 -1000
Subject: [PATCH] imp: lib: terminal colour detection

added:
terminalIsLight
terminalFgColor
terminalBgColor
---
 hledger-lib/Hledger/Utils/IO.hs | 37 ++++++++++++++++++++++++++++++---
 hledger-lib/package.yaml        |  1 +
 2 files changed, 35 insertions(+), 3 deletions(-)

diff --git a/hledger-lib/Hledger/Utils/IO.hs b/hledger-lib/Hledger/Utils/IO.hs
index 908ec3e03..329137fa4 100644
--- a/hledger-lib/Hledger/Utils/IO.hs
+++ b/hledger-lib/Hledger/Utils/IO.hs
@@ -34,7 +34,10 @@ module Hledger.Utils.IO (
   color,
   bgColor,
   colorB,
-  bgColorB,  
+  bgColorB,
+  terminalIsLight,
+  terminalFgColor,
+  terminalBgColor,
 
   -- * Errors
   error',
@@ -57,6 +60,8 @@ module Hledger.Utils.IO (
 where
 
 import           Control.Monad (when)
+import           Data.Colour.RGBSpace (RGB(RGB))
+import           Data.Colour.RGBSpace.HSL (lightness)
 import           Data.FileEmbed (makeRelativeToProject, embedStringFile)
 import           Data.List hiding (uncons)
 import           Data.Maybe (isJust)
@@ -67,9 +72,10 @@ import qualified Data.Text.Lazy.Builder as TB
 import           Data.Time.Clock (getCurrentTime)
 import           Data.Time.LocalTime
   (LocalTime, ZonedTime, getCurrentTimeZone, utcToLocalTime, utcToZonedTime)
+import           Data.Word (Word16)
 import           Language.Haskell.TH.Syntax (Q, Exp)
 import           System.Console.ANSI
-  (Color,ColorIntensity,ConsoleLayer(..), SGR(..), hSupportsANSIColor, setSGRCode)
+  (Color,ColorIntensity,ConsoleLayer(..), SGR(..), hSupportsANSIColor, setSGRCode, getLayerColor)
 import           System.Directory (getHomeDirectory)
 import           System.Environment (getArgs, lookupEnv)
 import           System.FilePath (isRelative, (</>))
@@ -77,7 +83,7 @@ import           System.IO
   (Handle, IOMode (..), hGetEncoding, hSetEncoding, hSetNewlineMode,
    openFile, stdin, stdout, stderr, universalNewlineMode, utf8_bom)
 import           System.IO.Unsafe (unsafePerformIO)
-import System.Pager
+import           System.Pager
 import           Text.Pretty.Simple
   (CheckColorTty(CheckColorTty), OutputOptions(..), 
   defaultOutputOptionsDarkBg, defaultOutputOptionsNoColor, pShowOpt, pPrintOpt)
@@ -228,6 +234,31 @@ bgColorB :: ColorIntensity -> Color -> WideBuilder -> WideBuilder
 bgColorB int col (WideBuilder s w) =
     WideBuilder (TB.fromString (setSGRCode [SetColor Background int col]) <> s <> TB.fromString (setSGRCode [])) w
 
+-- | Detect whether the terminal currently has a light background colour,
+-- if possible, using unsafePerformIO.
+terminalIsLight :: Maybe Bool
+terminalIsLight = (>lightthreshold).lightness <$> terminalColor Background
+  where lightthreshold = 0.7
+
+-- | Detect the terminal's current background colour, if possible, using unsafePerformIO.
+terminalBgColor :: Maybe (RGB Float)
+terminalBgColor = terminalColor Background
+
+-- | Detect the terminal's current foreground colour, if possible, using unsafePerformIO.
+terminalFgColor :: Maybe (RGB Float)
+terminalFgColor = terminalColor Foreground
+
+{-# NOINLINE terminalColor #-}
+terminalColor :: ConsoleLayer -> Maybe (RGB Float)
+terminalColor layer = unsafePerformIO $ do
+  inemacs <- not.null <$> lookupEnv "INSIDE_EMACS"
+  if inemacs  -- skip this in emacs shell buffers, the terminal escape sequence is visible for some reason
+  then return Nothing
+  else fmap fractionalRGB <$> getLayerColor layer
+  where
+    fractionalRGB :: (Fractional a) => RGB Word16 -> RGB a
+    fractionalRGB (RGB r g b) = RGB (fromIntegral r / 65535) (fromIntegral g / 65535) (fromIntegral b / 65535)  -- chatgpt
+
 -- Errors
 
 -- | Simpler alias for errorWithoutStackTrace
diff --git a/hledger-lib/package.yaml b/hledger-lib/package.yaml
index 20f240200..f3e7acb1d 100644
--- a/hledger-lib/package.yaml
+++ b/hledger-lib/package.yaml
@@ -45,6 +45,7 @@ dependencies:
 - containers >=0.5.9
 - cassava
 - cassava-megaparsec
+- colour >=2.3.6
 - data-default >=0.5
 - deepseq
 - Decimal >=0.5.1