From 2f324c432695d2c5080a1091576a0a1ae6aac996 Mon Sep 17 00:00:00 2001
From: Samuel Meuli <me@samuelmeuli.com>
Date: Sat, 21 Mar 2020 21:09:17 +0100
Subject: [PATCH] Add support for more UTIs and file extensions

---
 QLPlugin/Info.plist                           | 19 ++++++++++-
 .../FileTypeRenderers/CodeRenderer.swift      | 32 ++++++++++++++++++-
 2 files changed, 49 insertions(+), 2 deletions(-)

diff --git a/QLPlugin/Info.plist b/QLPlugin/Info.plist
index 840d282..838948a 100644
--- a/QLPlugin/Info.plist
+++ b/QLPlugin/Info.plist
@@ -28,6 +28,18 @@
 		<dict>
 			<key>QLSupportedContentTypes</key>
 			<array>
+				<!--
+					UTIs that macOS does not allow us to handle:
+
+					- .pem (public.x509-certificate)
+					- .plist (com.apple.property-list)
+					- .ts (could be TypeScript, is identified as public.mpeg-2-transport-stream)
+					- .xml (public.xml)
+				-->
+
+				<!-- Base UTI -->
+				<string>public.data</string>
+
 				<!-- CSV -->
 				<string>dyn.ah62d4rv4ge81k2pc</string> <!-- .tab -->
 				<string>public.comma-separated-values-text</string> <!-- .csv -->
@@ -45,8 +57,9 @@
 				<string>com.adobe.jsx</string>
 				<string>com.apple.applescript.script</string>
 				<string>com.apple.applescript.text</string>
-				<string>com.apple.property-list</string>
+				<string>com.apple.property-list</string> <!-- .plist -->
 				<string>com.apple.rez-source</string>
+				<string>com.apple.xcode.entitlements-property-list</string> <!-- .entitlements -->
 				<string>com.barebones.bbedit.scss-source</string>
 				<string>com.barebones.bbedit.verilog-hdl-source</string>
 				<string>com.barebones.bbedit.vhdl-source</string>
@@ -70,6 +83,7 @@
 				<string>com.sun.javafx</string>
 				<string>com.sun.java-class</string>
 				<string>com.sun.manifest</string>
+				<string>com.vallettaventures.texpadm.tex</string> <!-- Texpad LaTeX (.cls, .sty, .tex) -->
 				<string>com.unknown.lhs</string>
 				<string>org.applescript.source</string>
 				<string>org.arduino.ino-source</string>
@@ -120,6 +134,7 @@
 				<string>org.ocaml.ocaml-source</string>
 				<string>org.omg.ecore</string>
 				<string>org.rdf.source</string>
+				<string>com.runningwithcrayons.alfred.appearance</string> <!-- Alfred theme (.alfredappearance) -->
 				<string>org.rust-lang.source</string>
 				<string>org.sbarex.conf</string>
 				<string>org.sbarex.dart</string>
@@ -155,6 +170,7 @@
 				<string>tk.tcl.tcl</string>
 				<string>tk.tcl.tcl-source</string>
 				<string>dyn.ah62d4rv4ge8007a</string>
+				<string>dyn.ah62d4rv4ge80255drq</string> <!-- Lockfile (.lock, e.g. for Yarn and Poetry) -->
 				<string>dyn.ah62d4rv4ge80c75p</string>
 				<string>dyn.ah62d4rv4ge80g55sq2</string>
 				<string>dyn.ah62d4rv4ge80g62</string>
@@ -162,6 +178,7 @@
 				<string>dyn.ah62d4rv4ge80s6xbqv0gn</string>
 				<string>dyn.ah62d4rv4ge80w5pq</string>
 				<string>dyn.ah62d4rv4ge80y652</string>
+				<string>dyn.ah62d4rv4ge81e3pxr70hq3pe</string> <!-- Lockfile (.resolved, for Swift Package Manager) -->
 				<string>dyn.ah62d4rv4ge81g22</string>
 				<string>dyn.ah62d4rv4ge81g25brvuu</string>
 				<string>dyn.ah62d4rv4ge81g25xsq</string>
diff --git a/QLPlugin/Renderers/FileTypeRenderers/CodeRenderer.swift b/QLPlugin/Renderers/FileTypeRenderers/CodeRenderer.swift
index 9196825..418c2d0 100644
--- a/QLPlugin/Renderers/FileTypeRenderers/CodeRenderer.swift
+++ b/QLPlugin/Renderers/FileTypeRenderers/CodeRenderer.swift
@@ -1,9 +1,39 @@
 import Foundation
 import os.log
 
+let dotfileLexers = [
+	".dockerignore": "bash",
+	".editorconfig": "ini",
+	".gitattributes": "bash",
+	".gitignore": "bash",
+	".npmignore": "bash",
+]
+
+let fileExtensionLexers = [
+	"alfredappearance": "json", // Alfred theme
+	"cls": "tex", // LaTeX classes file
+	"entitlements": "xml",
+	"iml": "xml",
+	"plist": "xml",
+	"resolved": "json", // Swift Package Manager lockfile (Package.resolved)
+	"sty": "tex", // LaTeX styles file
+]
+
 class CodeRenderer: Renderer {
 	private let chromaBinUrl = Bundle.main.url(forAuxiliaryExecutable: "chroma-v0.7.0")
 
+	/// Returns the name of the Chroma lexer to use for the file. This is determined based on the
+	/// file name/extension
+	private func getLexer() -> String {
+		if fileExtension.isEmpty {
+			// Dotfile
+			return dotfileLexers[fileUrl.lastPathComponent, default: "autodetect"]
+		} else {
+			// File with extension
+			return fileExtensionLexers[fileExtension, default: "autodetect"]
+		}
+	}
+
 	override func getStylesheets() -> [Stylesheet] {
 		var stylesheets = super.getStylesheets()
 		if let cssUrl = chromaCssUrl {
@@ -22,7 +52,7 @@ class CodeRenderer: Renderer {
 
 		let (status, stdout, stderr) = Shell.run(
 			url: binaryUrlResolved,
-			arguments: [fileUrl.path, "--html", "--html-only"]
+			arguments: [fileUrl.path, "--html", "--html-only", "--lexer", getLexer()]
 		)
 
 		guard status == 0 else {