From ac30b071d2f152305b7ae393c328ae316dff2791 Mon Sep 17 00:00:00 2001
From: The Mist <mist.zzh@gmail.com>
Date: Mon, 15 Apr 2024 22:33:57 +0800
Subject: [PATCH] [Fix] auto detect content which should use source syntax
 (#86)

---
 generator/src/highlight.rs | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/generator/src/highlight.rs b/generator/src/highlight.rs
index 88c3e00..3d2096f 100644
--- a/generator/src/highlight.rs
+++ b/generator/src/highlight.rs
@@ -1,3 +1,5 @@
+use std::collections::HashMap;
+
 use cosmic_text::{Attrs, Family, Style, Weight};
 use syntect::{
     easy::HighlightLines,
@@ -8,11 +10,14 @@ use syntect::{
 
 use crate::components::interface::render_error::RenderError;
 
+type SourceMap = HashMap<&'static str, &'static str>;
+
 pub struct Highlight {
     content: String,
     code_file_path: String,
     extension: Option<String>,
     font_family: String,
+    highlighting_language_source_map: SourceMap,
 }
 
 pub type HighlightResult<'a> = Vec<(&'a str, Attrs<'a>)>;
@@ -29,6 +34,7 @@ impl Highlight {
             code_file_path,
             extension,
             font_family,
+            highlighting_language_source_map: HashMap::from([("PHP", "<?php")]),
         }
     }
 
@@ -47,6 +53,15 @@ impl Highlight {
                 ))?,
         };
 
+        if let Some(identifier) = self.highlighting_language_source_map.get(&syntax.name[..]) {
+            if !self.content.contains(identifier) {
+                return Ok(syntax_set
+                    .find_syntax_by_name(&format!("{} Source", &syntax.name))
+                    .unwrap_or(syntax)
+                    .to_owned());
+            }
+        }
+
         Ok(syntax.to_owned())
     }