noctalia-shell/Services/MatugenService.qml

pragma Singleton

import QtQuick
import Quickshell
import Quickshell.Io
import qs.Commons
import qs.Services
import "../Helpers/ColorsConvert.js" as ColorsConvert

Singleton {
  id: root

  readonly property string colorsApplyScript: Quickshell.shellDir + '/Bin/colors-apply.sh'

  property string dynamicConfigPath: Settings.cacheDir + "matugen.dynamic.toml"

  // External state management
  Connections {
    target: WallpaperService
    function onWallpaperChanged(screenName, path) {
      // Only detect changes on main screen
      if (screenName === Screen.name && Settings.data.colorSchemes.useWallpaperColors) {
        generateFromWallpaper()
      }
    }
  }

  Connections {
    target: Settings.data.colorSchemes
    function onDarkModeChanged() {
      Logger.log("Matugen", "Detected dark mode change")
      if (Settings.data.colorSchemes.useWallpaperColors) {
        MatugenService.generateFromWallpaper()
      }
    }
  }

  // --------------------------------
  function init() {
    // does nothing but ensure the singleton is created
    // do not remove
    Logger.log("Matugen", "Service started")
  }

  // --------------------------------
  // Generate colors using current wallpaper and settings
  function generateFromWallpaper() {
    Logger.log("Matugen", "Generating from wallpaper on screen:", Screen.name)
    
    var wp = WallpaperService.getWallpaper(Screen.name).replace(/'/g, "'\\''")
    if (wp === "") {
      Logger.error("Matugen", "No wallpaper was found")
      return
    }

    var content = MatugenTemplates.buildConfigToml()
    var mode = Settings.data.colorSchemes.darkMode ? "dark" : "light"
    var pathEsc = dynamicConfigPath.replace(/'/g, "'\\''")

    // Build the main script
    var script = buildMatugenScript(content, pathEsc, wp, mode)

    generateProcess.command = ["bash", "-lc", script]
    generateProcess.running = true
  }

  // --------------------------------
  function buildMatugenScript(content, pathEsc, wallpaper, mode) {
    var script = "cat > '" + pathEsc + "' << 'EOF'\n" + content + "EOF\n"
    
    // Main matugen command
    script += "matugen image '" + wallpaper + "' --config '" + pathEsc + "' --mode " + mode + " --type " + Settings.data.colorSchemes.matugenSchemeType

    // Add user template execution if enabled
    script += addUserTemplateExecution(wallpaper, mode)

    return script + "\n"
  }

  // --------------------------------
  function addUserTemplateExecution(input, mode) {
    if (!Settings.data.templates.enableUserTemplates) {
      return ""
    }

    var userConfigPath = getUserConfigPath()
    var script = "\n# Execute user config if it exists\n"
    script += "if [ -f '" + userConfigPath + "' ]; then\n"
    script += "  matugen image '" + input + "' --config '" + userConfigPath + "' --mode " + mode + " --type " + Settings.data.colorSchemes.matugenSchemeType + "\n"
    script += "fi"
    
    return script
  }

  // --------------------------------
  function getUserConfigPath() {
    return (Quickshell.env("HOME") + "/.config/matugen/config.toml").replace(/'/g, "'\\''")
  }

  // --------------------------------
  function selectVibrantColor(schemeData, mode) {
    var colors = []
    colors.push(schemeData[mode]["mPrimary"]);
    colors.push(schemeData[mode]["mSecondary"]);
    colors.push(schemeData[mode]["mTertiary"]);


    var bestScore = 0
    var bestScoreIndex = -1
    for (var i=0; i<colors.length; i++) {
      var hsl = ColorsConvert.hexToHSL(colors[i])

      var score = hsl['s'];
      if (score > bestScore) {
        bestScore = score
        bestScoreIndex = i
      }
    }

    return colors[bestScoreIndex]
  }

  // --------------------------------
  // Generate templates from predefined color scheme
  function generateFromPredefinedScheme(schemeData) {
    Logger.log("Matugen", "Generating templates from predefined color scheme")

    var content = MatugenTemplates.buildConfigToml()
    var mode = Settings.data.colorSchemes.darkMode ? "dark" : "light"
    var pathEsc = dynamicConfigPath.replace(/'/g, "'\\''")
    const color = selectVibrantColor(schemeData, mode)

    // Build the script
    var script = buildPredefinedSchemeScript(content, pathEsc, color, mode)

    generateProcess.command = ["bash", "-lc", script]
    generateProcess.running = true

    // Handle terminal theme copying for predefined schemes
    handleTerminalThemes()
  }

  // --------------------------------
  function buildPredefinedSchemeScript(content, pathEsc, color, mode) {
    var script = "cat > '" + pathEsc + "' << 'EOF'\n" + content + "EOF\n\n"
    script += "matugen color hex '" + color + "' --config '" + pathEsc + "' --mode " + mode + "\n"
    
    // Add user template execution if enabled
    script += addUserTemplateExecutionForColor(color, mode)
    
    return script
  }

  // --------------------------------
  function addUserTemplateExecutionForColor(color, mode) {
    if (!Settings.data.templates.enableUserTemplates) {
      return ""
    }

    var userConfigPath = getUserConfigPath()
    var script = "\n# Execute user config if it exists\n"
    script += "if [ -f '" + userConfigPath + "' ]; then\n"
    script += "  matugen color hex '" + color + "' --config '" + userConfigPath + "' --mode " + mode + "\n"
    script += "fi"
    
    return script
  }

  // --------------------------------
  function handleTerminalThemes() {
    var terminals = {
      foot: "~/.config/foot/themes/noctalia",
      ghostty: "~/.config/ghostty/themes/noctalia",
      kitty: "~/.config/kitty/themes/noctalia.conf",
    }

    var copyCmd = Object.entries(terminals)
      .filter(([terminal, colorsPath]) => Settings.data.templates[terminal])
      .map(([terminal, colorsPath]) => {
        var colorsPathParent = colorsPath.replace(/[^\/]*$/, "")
        var terminalColorsTemplate = getTerminalColorsTemplate(terminal)
        return [
          `mkdir -p ${colorsPathParent}`,
          `cp -f ${terminalColorsTemplate} ${colorsPath}`,
          `${colorsApplyScript} ${terminal}`,
        ]
      })
      .reduce((arr1, arr2) => arr1.concat(arr2), [])
      .join("; ")

    if (copyCmd !== "") {
      copyProcess.command = ["bash", "-lc", copyCmd]
      copyProcess.running = true
    }
  }

  // --------------------------------
  function getTerminalColorsTemplate(terminal) {
    var colorScheme = Settings.data.colorSchemes.predefinedScheme
    const darkLight = Settings.data.colorSchemes.darkMode ? 'dark' : 'light'

    // Convert display names back to folder names
    var schemeMap = {
      "Noctalia (default)": "Noctalia-default",
      "Noctalia (legacy)": "Noctalia-legacy",
      "Tokyo Night": "Tokyo-Night"
    }
    
    colorScheme = schemeMap[colorScheme] || colorScheme
    var extension = terminal === 'kitty' ? ".conf" : ""

    return `${Quickshell.shellDir}/Assets/ColorScheme/${colorScheme}/terminal/${terminal}/${colorScheme}-${darkLight}${extension}`
  }

  // --------------------------------
  Process {
    id: generateProcess
    workingDirectory: Quickshell.shellDir
    running: false
    stderr: StdioCollector {
      onStreamFinished: {
        if (this.text !== "") {
          Logger.warn("MatugenService", "GenerateProcess stderr:", this.text)
        }
      }
    }
  }

  // --------------------------------
  Process {
    id: copyProcess
    running: false
    stderr: StdioCollector {
      onStreamFinished: {
        if (this.text !== "") {
          Logger.warn("MatugenService", "CopyProcess stderr:", this.text)
        }
      }
    }
  }
}