;;; agent-shell-styles.el --- Alternative status/kind label styles. -*- lexical-binding: t; -*-

;; Copyright (C) 2024 Alvaro Ramirez

;; Author: Alvaro Ramirez https://xenodium.com
;; URL: https://github.com/xenodium/agent-shell

;; This package is free software; you can redistribute it and/or modify
;; it under the terms of the GNU General Public License as published by
;; the Free Software Foundation; either version 3, or (at your option)
;; any later version.

;; This package is distributed in the hope that it will be useful,
;; but WITHOUT ANY WARRANTY; without even the implied warranty of
;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
;; GNU General Public License for more details.

;; You should have received a copy of the GNU General Public License
;; along with GNU Emacs.  If not, see <https://www.gnu.org/licenses/>.

;;; Commentary:
;;
;; Alternative functions for `agent-shell-status-kind-label-function'.
;;
;; Report issues at https://github.com/xenodium/agent-shell/issues
;;
;; ✨ Please support this work https://github.com/sponsors/xenodium ✨

;;; Code:

(require 'map)
(require 'seq)
(require 'agent-shell-faces)

(declare-function agent-shell--add-text-properties "agent-shell-ui")

(defun agent-shell--short-kind-label (kind)
  "Return a short label for tool call KIND string."
  (pcase kind
    ("search" "find")
    ("execute" "command")
    (_ kind)))

(defun agent-shell--status-config (status)
  "Return alist with :label, :icon, and :face for STATUS string.

  (agent-shell--status-config \"completed\")
  ;; => ((:label . \"done\") (:icon . \"✓\") (:face . agent-shell-success))"
  (pcase status
    ("pending" '((:label . "wait") (:icon . "◔") (:face . agent-shell-pending)))
    ("in_progress" '((:label . "busy") (:icon . "◔") (:face . agent-shell-warning)))
    ("completed" '((:label . "done") (:icon . "✓") (:face . agent-shell-success)))
    ("failed" '((:label . "error") (:icon . "✗") (:face . agent-shell-error)))
    (_ '((:label . "unknown") (:icon . "?") (:face . agent-shell-warning)))))

(defun agent-shell--inverse-label-status-kind-label (status kind)
  "Render STATUS as an inverse-video word label and KIND as boxed text.

Uses the word label from `agent-shell--status-config' (e.g. \"done\")
with `(:inverse-video t)' so the text sits on a face-colored
rectangle.  Sibling to `agent-shell--inverse-icon-status-kind-label',
which uses an icon glyph instead of a word.

STATUS is a string like \"completed\" or nil.
KIND is a string like \"read\" or nil.
Returns a propertized string or nil."
  (let* ((status-config (agent-shell--status-config status))
         (label-format (if (display-graphic-p) " %s " "[%s]"))
         (status-text (when status
                        (let ((label (map-elt status-config :label))
                              (face (map-elt status-config :face)))
                          (agent-shell--add-text-properties
                           (propertize (format label-format label)
                                       'font-lock-face 'default)
                           'font-lock-face (list face '(:inverse-video t))))))
         (kind-text (when kind
                      (let ((box-color (face-foreground
                                        (map-elt status-config :face) nil t)))
                        (agent-shell--add-text-properties
                         (propertize (format label-format
                                             (agent-shell--short-kind-label kind))
                                     'font-lock-face 'default)
                         'font-lock-face `((:box (:color ,box-color))))))))
    (concat status-text kind-text)))

(defun agent-shell--background-tint-status-kind-label (status kind)
  "Render STATUS and KIND as tinted background labels.

Derives background by blending the face foreground (30%) with the
default background (70%), so it adapts to any theme.

  (agent-shell--background-tint-status-kind-label \"completed\" \"read\")
  ;; => #(\" done \" ...) #(\" read \" ...)

STATUS is a string like \"completed\" or nil.
KIND is a string like \"read\" or nil.
Returns a propertized string or nil."
  (let* ((status-config (agent-shell--status-config status))
         (fg (face-foreground (map-elt status-config :face) nil t))
         (bg-base (face-background 'default nil t))
         (bg (when (and fg bg-base)
               (apply #'format "#%02x%02x%02x"
                      (seq-mapn (lambda (f b)
                                  (/ (+ (* f 3) (* b 7)) 10))
                                (color-values fg)
                                (color-values bg-base)))))
         (label-format (if (display-graphic-p) " %s " "[%s]"))
         (status-text (when status
                        (propertize (format label-format
                                            (map-elt status-config :label))
                                    'font-lock-face
                                    `(:background ,bg :foreground ,fg :weight bold))))
         (kind-text (when kind
                      (propertize (format label-format
                                          (agent-shell--short-kind-label kind))
                                  'font-lock-face
                                  `(:background ,bg :foreground ,fg :slant italic)))))
    (concat status-text kind-text)))

(defun agent-shell--unicode-icons-status-kind-label (status kind)
  "Render STATUS as a unicode icon and KIND as typed text.

  (agent-shell--unicode-icons-status-kind-label \"completed\" \"read\")
  ;; => \"✓ read\"

  (agent-shell--unicode-icons-status-kind-label \"completed\" nil)
  ;; => \"✓\"

STATUS is a string like \"completed\" or nil.
KIND is a string like \"read\" or nil.
Returns a propertized string or nil."
  (let ((status-config (agent-shell--status-config status))
        (status-text nil)
        (kind-text nil))
    (when status
      (setq status-text (propertize (map-elt status-config :icon)
                                    'font-lock-face
                                    (map-elt status-config :face))))
    (when kind
      (setq kind-text (propertize (agent-shell--short-kind-label kind)
                                  'font-lock-face 'font-lock-type-face)))
    (if (and status-text kind-text)
        (concat status-text " " kind-text)
      (or status-text kind-text))))

(defun agent-shell--icon-and-kind-status-kind-label (status kind)
  "Render STATUS as a colored icon followed by KIND as a heading.

KIND is capitalized and shares `agent-shell-section-heading' with the
\"Thinking\" label and the activity group header, so every entry in a
group reads at the same weight, with the title beside it plain.  Nothing
is padded: fragments sit a blank line apart, so a fixed kind column buys
no scanning and only opens a gap.  Entries carrying no KIND, like plan
steps, render the icon alone.

  (agent-shell--icon-and-kind-status-kind-label \"completed\" \"execute\")
  ;; => \"✓ Command\"

  (agent-shell--icon-and-kind-status-kind-label \"pending\" nil)
  ;; => \"◔\"

STATUS is a string like \"completed\" or nil.
KIND is a string like \"read\" or nil.
Returns a propertized string or nil."
  (let* ((status-config (agent-shell--status-config status))
         (status-text (when status
                        (propertize (map-elt status-config :icon)
                                    'font-lock-face (map-elt status-config :face))))
         (kind-text (when kind
                      (propertize (capitalize
                                   (string-replace
                                    "_" " "
                                    (agent-shell--short-kind-label kind)))
                                  'font-lock-face 'agent-shell-section-heading))))
    (if (and status-text kind-text)
        (concat status-text " " kind-text)
      (or status-text kind-text))))

(defun agent-shell--plain-colored-status-kind-label (status kind)
  "Render STATUS and KIND as plain colored text with no decoration.

  (agent-shell--plain-colored-status-kind-label \"completed\" \"read\")
  ;; => #(\" done \" ...) #(\" read \" ...)

STATUS is a string like \"completed\" or nil.
KIND is a string like \"read\" or nil.
Returns a propertized string or nil."
  (let* ((status-config (agent-shell--status-config status))
         (face (map-elt status-config :face))
         (label-format (if (display-graphic-p) " %s " "[%s]"))
         (status-text (when status
                        (propertize (format label-format
                                            (map-elt status-config :label))
                                    'font-lock-face face)))
         (kind-text (when kind
                      (propertize (format label-format
                                          (agent-shell--short-kind-label kind))
                                  'font-lock-face face))))
    (concat status-text kind-text)))

(defun agent-shell--inverse-icon-status-kind-label (status kind)
  "Render STATUS as an inverse-video icon glyph and KIND as boxed text.

Uses the unicode glyph from `agent-shell--status-config' with
`(:inverse-video t)' so the glyph sits on a face-colored rectangle.

  (agent-shell--inverse-icon-status-kind-label \"completed\" \"read\")
  ;; => #(\" ✓ \" ...) #(\" read \" ...)

STATUS is a string like \"completed\" or nil.
KIND is a string like \"read\" or nil.
Returns a propertized string or nil."
  (let* ((status-config (agent-shell--status-config status))
         (face (map-elt status-config :face))
         (label-format (if (display-graphic-p) " %s " "[%s]"))
         (status-text
          (when status
            (agent-shell--add-text-properties
             (propertize (format label-format
                                 (map-elt status-config :icon))
                         'font-lock-face 'default)
             'font-lock-face (list face '(:inverse-video t)))))
         (kind-text
          (when kind
            (agent-shell--add-text-properties
             (propertize (format label-format
                                 (agent-shell--short-kind-label kind))
                         'font-lock-face 'default)
             'font-lock-face
             `((:box (:color ,(face-foreground face nil t))))))))
    (concat status-text kind-text)))

(provide 'agent-shell-styles)

;;; agent-shell-styles.el ends here
