tkmind/tkmind_go
1
0
Fork 0
Code Issues Pull Requests Actions 4 Packages Projects Releases Wiki Activity

fix: use lazy for regex to avoid recompile (#6865)

Browse Source 
Signed-off-by: lsytj0413 <511121939@qq.com>
This commit is contained in:
Songlin Yang
2026-02-03 15:00:58 +08:00
committed by GitHub
parent e32e117b7e
commit 3d0bb3c670
1 changed files with 15 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/goose-mcp/src/developer/rmcp_developer.rs
+15 -12
View File
@@ -4,6 +4,7 @@ use etcetera::AppStrategy;
use ignore::gitignore::{Gitignore, GitignoreBuilder}; use ignore::gitignore::{Gitignore, GitignoreBuilder};
use include_dir::{include_dir, Dir}; use include_dir::{include_dir, Dir};
use indoc::{formatdoc, indoc}; use indoc::{formatdoc, indoc};
use once_cell::sync::Lazy;
use rmcp::{ use rmcp::{
handler::server::{router::tool::ToolRouter, wrapper::Parameters}, handler::server::{router::tool::ToolRouter, wrapper::Parameters},
model::{ model::{
@@ -121,6 +122,13 @@ pub struct PromptArgumentTemplate {
// Embeds the prompts directory to the build // Embeds the prompts directory to the build
static PROMPTS_DIR: Dir = include_dir!("$CARGO_MANIFEST_DIR/src/developer/prompts"); static PROMPTS_DIR: Dir = include_dir!("$CARGO_MANIFEST_DIR/src/developer/prompts");
static MACOS_SCREENSHOT_FILENAME_RE: Lazy<regex::Regex> = Lazy::new(|| {
regex::Regex::new(
r"^Screenshot \d{4}-\d{2}-\d{2} at \d{1,2}\.\d{2}\.\d{2} (AM|PM|am|pm)(?: \(\d+\))?\.png$",
)
.expect("macOS screenshot filename regex should be valid")
});
const DEFAULT_GOOSEIGNORE_CONTENT: &str = concat!( const DEFAULT_GOOSEIGNORE_CONTENT: &str = concat!(
"# This file is created automatically if no .gooseignore exists.\n", "# This file is created automatically if no .gooseignore exists.\n",
"# Customize or uncomment the patterns below instead of deleting the file.\n", "# Customize or uncomment the patterns below instead of deleting the file.\n",
@@ -1387,27 +1395,22 @@ impl DeveloperServer {
if let Some(filename) = path.file_name().and_then(|f| f.to_str()) { if let Some(filename) = path.file_name().and_then(|f| f.to_str()) {
// Check if this matches Mac screenshot pattern: // Check if this matches Mac screenshot pattern:
// "Screenshot YYYY-MM-DD at H.MM.SS AM/PM.png" // "Screenshot YYYY-MM-DD at H.MM.SS AM/PM.png"
if let Some(captures) = regex::Regex::new(r"^Screenshot \d{4}-\d{2}-\d{2} at \d{1,2}\.\d{2}\.\d{2} (AM|PM|am|pm)(?: \(\d+\))?\.png$") if let Some(captures) = MACOS_SCREENSHOT_FILENAME_RE.captures(filename) {
.ok()
.and_then(|re| re.captures(filename))
{
// Get the AM/PM part // Get the AM/PM part
let meridian = captures.get(1).unwrap().as_str(); let meridian = captures.get(1).unwrap().as_str();
// Find the last space before AM/PM and replace it with U+202F // Find the last space before AM/PM and replace it with U+202F
let space_pos = filename.rfind(meridian) let space_pos = filename
.rfind(meridian)
.and_then(|pos| filename.get(..pos).map(|s| s.trim_end().len())) .and_then(|pos| filename.get(..pos).map(|s| s.trim_end().len()))
.unwrap_or(0); .unwrap_or(0);
if space_pos > 0 { if space_pos > 0 {
let parent = path.parent().unwrap_or(Path::new("")); let parent = path.parent().unwrap_or(Path::new(""));
if let (Some(before), Some(after)) = (filename.get(..space_pos), filename.get(space_pos+1..)) { if let (Some(before), Some(after)) =
let new_filename = format!( (filename.get(..space_pos), filename.get(space_pos + 1..))
"{}{}{}", {
before, let new_filename = format!("{}{}{}", before, '\u{202F}', after);
'\u{202F}',
after
);
let new_path = parent.join(new_filename); let new_path = parent.join(new_filename);
return new_path; return new_path;