use regex::Regex;
use std::sync::LazyLock;
static COMMENT_SINGLE: LazyLock<Regex> = LazyLock::new(|| Regex::new(r"//.*$").unwrap());
static COMMENT_MULTI: LazyLock<Regex> = LazyLock::new(|| Regex::new(r"(?s)/\*.*?\*/").unwrap());
static GO_DECL: LazyLock<Regex> = LazyLock::new(|| {
Regex::new(r"(?m)^(?:func|type|var|const)\s+(?:\([^)]*\)\s+)?([A-Z]\w*)").unwrap()
});
static GO_METHOD: LazyLock<Regex> =
LazyLock::new(|| Regex::new(r"(?m)^func\s+\([^)]+\)\s+([A-Z]\w*)").unwrap());
pub fn extract_exports(content: &str) -> Vec<String> {
let stripped = COMMENT_SINGLE.replace_all(content, "");
let stripped = COMMENT_MULTI.replace_all(&stripped, "");
let mut symbols = Vec::new();
for caps in GO_DECL.captures_iter(&stripped) {
if let Some(name) = caps.get(1) {
symbols.push(name.as_str().to_string());
}
}
for caps in GO_METHOD.captures_iter(&stripped) {
if let Some(name) = caps.get(1) {
let n = name.as_str().to_string();
if !symbols.contains(&n) {
symbols.push(n);
}
}
}
symbols
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_go_exports() {
let src = r#"
package auth
func CreateAuth(config Config) Auth {}
func privateFunc() {}
type AuthService struct {}
type authInternal struct {}
const DefaultTTL = 3600
var GlobalInstance *Auth
"#;
let symbols = extract_exports(src);
assert!(symbols.contains(&"CreateAuth".to_string()));
assert!(symbols.contains(&"AuthService".to_string()));
assert!(symbols.contains(&"DefaultTTL".to_string()));
assert!(symbols.contains(&"GlobalInstance".to_string()));
assert!(!symbols.contains(&"privateFunc".to_string()));
assert!(!symbols.contains(&"authInternal".to_string()));
}
#[test]
fn test_go_methods() {
let src = r#"
package auth
func (a *Auth) Validate(token string) bool {}
func (a *Auth) internal() {}
"#;
let symbols = extract_exports(src);
assert!(symbols.contains(&"Validate".to_string()));
assert!(!symbols.contains(&"internal".to_string()));
}
#[test]
fn test_go_comments_stripped() {
let src = r#"
package main
// func FakeExport() {}
/* func AlsoFake() {} */
func RealExport() {}
/*
func MultiLineFake() {}
type FakeType struct {}
*/
type RealType struct {}
"#;
let symbols = extract_exports(src);
assert!(symbols.contains(&"RealExport".to_string()));
assert!(symbols.contains(&"RealType".to_string()));
assert!(!symbols.contains(&"FakeExport".to_string()));
assert!(!symbols.contains(&"AlsoFake".to_string()));
assert!(!symbols.contains(&"MultiLineFake".to_string()));
assert!(!symbols.contains(&"FakeType".to_string()));
}
#[test]
fn test_go_interface_declarations() {
let src = r#"
package service
type Reader interface {
Read(p []byte) (n int, err error)
}
type Writer interface {
Write(p []byte) (n int, err error)
}
type internalHelper interface {}
"#;
let symbols = extract_exports(src);
assert!(symbols.contains(&"Reader".to_string()));
assert!(symbols.contains(&"Writer".to_string()));
assert!(!symbols.contains(&"internalHelper".to_string()));
}
#[test]
fn test_go_const_var_groups() {
let src = r#"
package config
const MaxRetries = 3
const minTimeout = 100
var DefaultClient *Client
var debugMode = false
type Config struct {}
"#;
let symbols = extract_exports(src);
assert!(symbols.contains(&"MaxRetries".to_string()));
assert!(symbols.contains(&"DefaultClient".to_string()));
assert!(symbols.contains(&"Config".to_string()));
assert!(!symbols.contains(&"minTimeout".to_string()));
assert!(!symbols.contains(&"debugMode".to_string()));
}
#[test]
fn test_go_value_receiver() {
let src = r#"
package auth
func (a Auth) String() string {}
func (a Auth) serialize() string {}
"#;
let symbols = extract_exports(src);
assert!(symbols.contains(&"String".to_string()));
assert!(!symbols.contains(&"serialize".to_string()));
}
#[test]
fn test_go_empty_file() {
let src = "package main\n";
let symbols = extract_exports(src);
assert!(symbols.is_empty());
}
}