Merge pull request #5 from ArkieCoder/kittysupport

add support for Kitty

Author: ArkieCoder

install.sh

@@ -4,7 +4,7 @@
 # Detects dependencies and installs the Ruby Primer app.
 
 REQUIRED_RUBY_VERSION="3.2"
-REQUIRED_UTILITIES=("pandoc" "xelatex")
+REQUIRED_UTILITIES=("pandoc" "xelatex" "convert")
 FONT_NAME="IBM Plex Serif Light"
 
 # Colors for output
@@ -112,19 +112,19 @@ if [ ${#missing_utils[@]} -ne 0 ]; then
     echo -e "${YELLOW}Missing utilities: ${missing_utils[*]}${NC}"
     case "$PACKAGE_MANAGER" in
         brew)
-            echo "Install with: brew install pandoc mactex"
+            echo "Install with: brew install pandoc mactex imagemagick"
             echo "Note: Ensure IBM Plex Serif Light font is available (included in mactex or install separately)."
             ;;
         apt)
-            echo "Install with: sudo apt update && sudo apt install pandoc texlive-latex-base texlive-fonts-recommended texlive-xelatex"
+            echo "Install with: sudo apt update && sudo apt install pandoc texlive-latex-base texlive-fonts-recommended texlive-xelatex imagemagick"
             echo "Note: Ensure IBM Plex Serif Light font is installed (may require additional font packages)."
             ;;
         yum)
-            echo "Install with: sudo yum install pandoc texlive texlive-fonts"
+            echo "Install with: sudo yum install pandoc texlive texlive-fonts ImageMagick"
             echo "Note: Ensure IBM Plex Serif Light font is installed."
             ;;
         *)
-            echo "Please install pandoc and xelatex manually."
+            echo "Please install pandoc, xelatex, and imagemagick manually."
             echo "Note: Ensure IBM Plex Serif Light font is available for xelatex."
             ;;
     esac

lib/stringformat.rb

@@ -3,6 +3,15 @@ def format(color: 'white')
     require 'tempfile'
     require 'base64'
 
+    # Terminal detection
+    def kitty?
+      ENV['TERM'] == 'xterm-kitty' || ENV['TERM_PROGRAM'] == 'Kitty'
+    end
+
+    def iterm2?
+      ENV['TERM_PROGRAM'] == 'iTerm.app'
+    end
+
     # Get terminal columns, similar to tput cols
     cols = `tput cols`.to_i
     em = (cols * 0.60).to_f
@@ -38,15 +47,42 @@ def format(color: 'white')
     system("pandoc --pdf-engine=xelatex -V mainfont=\"IBM Plex Serif Light\" -f commonmark_x #{tmp_file.path} -o #{pdf_file} >/dev/null 2>&1")
 
     if File.exist?(pdf_file)
-      pdf_data = File.read(pdf_file)
-      base64_data = Base64.strict_encode64(pdf_data)
-      # Clean up
-      tmp_file.unlink
-      File.unlink(pdf_file)
-      # Return iTerm2 inline image sequence
-      "\e]1337;File=inline=1;preserveAspectRatio=1:#{base64_data}\a"
+      if kitty?
+        # Convert PDF to PNG for Kitty
+        png_file = tmp_file.path.sub('.md', '.png')
+        system("convert #{pdf_file} #{png_file} >/dev/null 2>&1")
+        if File.exist?(png_file)
+          png_data = File.read(png_file)
+          base64_data = Base64.strict_encode64(png_data)
+          # Clean up
+          tmp_file.unlink
+          File.unlink(pdf_file)
+          File.unlink(png_file)
+          # Return Kitty inline image sequence
+          "\e_Gf=100,a=T,t=d;#{base64_data}\e\\"
+        else
+          # Fallback if PNG conversion fails
+          tmp_file.unlink
+          File.unlink(pdf_file)
+          self
+        end
+      elsif iterm2?
+        pdf_data = File.read(pdf_file)
+        base64_data = Base64.strict_encode64(pdf_data)
+        # Clean up
+        tmp_file.unlink
+        File.unlink(pdf_file)
+        # Return iTerm2 inline image sequence
+        "\e]1337;File=inline=1;preserveAspectRatio=1:#{base64_data}\a"
+      else
+        # Fallback for other terminals
+        tmp_file.unlink
+        File.unlink(pdf_file)
+        self
+      end
     else
       # Fallback if PDF generation fails
+      tmp_file.unlink
       self
     end
   end