rustra-macros 0.9.0

/// `#[command]` 속성의 파싱 결과입니다.
///
/// `#[command]`, `#[command(name = "customName")]`,
/// `#[command(capability = "compute:secure")]`,
/// `#[command(error("math.divide_by_zero"))]`,
/// `#[command(device(camera, "clipboard-read"))]` 형태를 지원합니다.
struct CommandAttr {
    /// 명시적으로 지정한 명령 이름. 없으면 함수 이름에서 자동 추론합니다.
    name: Option<String>,
    /// 이 명령이 요구하는 capability. `require_capability` 문자열 결합을 대체한다.
    capability: Option<String>,
    /// 이 명령이 구현되는 플랫폼 목록 (`platform(windows, macos)`).
    /// None 이면 전 플랫폼 명령.
    platforms: Option<Vec<String>>,
    /// 이 명령이 반환할 수 있는 도메인 에러 코드 목록
    /// (`error("math.divide_by_zero")`). 설명/retryable 메타데이터는 빌더
    /// `command_errors` 체인으로 — 속성은 코드 문자열 목록만 받는다.
    errors: Option<Vec<String>>,
    /// 이 명령이 전제하는 디바이스 역량 토큰 목록
    /// (`device(camera, "clipboard-read")`). 카탈로그 검증은 등록 시점
    /// (`command_devices`)에 loud-fail 한다 — 매크로 크레이트는 카탈로그를
    /// 모른다(중복 유지 대신 단일 소스).
    devices: Option<Vec<String>>,
}

/// `#[command]` 속성의 입력을 파싱합니다.
///
/// 빈 입력(`#[command]`)이면 둘 다 `None`. `name = "foo"` / `capability = "cap"` 키를
/// 쉼표로 구분해 받는다. 알 수 없는 키는 지원 목록을 안내하는 에러가 된다.
impl Parse for CommandAttr {
    fn parse(input: ParseStream) -> syn::Result<Self> {
        let mut attr = CommandAttr {
            name: None,
            capability: None,
            platforms: None,
            errors: None,
            devices: None,
        };
        if input.is_empty() {
            return Ok(attr);
        }

        loop {
            let key: Ident = input.parse()?;
            if key == "name" {
                let _: Token![=] = input.parse()?;
                let name: LitStr = input.parse()?;
                attr.name = Some(name.value());
            } else if key == "capability" {
                let _: Token![=] = input.parse()?;
                let cap: LitStr = input.parse()?;
                attr.capability = Some(cap.value());
            } else if key == "platform" {
                // platform(windows, macos) — 괄호 안 플랫폼 식별자 목록.
                let content;
                let _: syn::token::Paren = syn::parenthesized!(content in input);
                let mut platforms = Vec::new();
                loop {
                    let ident: Ident = content.parse()?;
                    platforms.push(ident.to_string());
                    if content.parse::<Token![,]>().is_err() {
                        break;
                    }
                }
                if platforms.is_empty() {
                    return Err(syn::Error::new(
                        key.span(),
                        "platform(...) requires at least one platform",
                    ));
                }
                attr.platforms = Some(platforms);
            } else if key == "error" {
                // error("code.a", "code.b") — 괄호 안 도메인 에러 코드 문자열 목록.
                let content;
                let _: syn::token::Paren = syn::parenthesized!(content in input);
                if content.is_empty() {
                    return Err(syn::Error::new(
                        key.span(),
                        "error(...) requires at least one error code",
                    ));
                }
                let mut errors = Vec::new();
                loop {
                    let code: LitStr = content.parse()?;
                    errors.push(code.value());
                    if content.parse::<Token![,]>().is_err() {
                        break;
                    }
                }
                attr.errors = Some(errors);
            } else if key == "device" {
                // device(camera, "clipboard-read") — 괄호 안 역량 토큰 목록.
                // 식별자와 문자열 리터럴을 모두 받는다(platform 파싱의 변주) —
                // kebab-case 토큰(clipboard-read)은 식별자로 쓸 수 없어 문자열
                // 리터럴 경로가 필요하다.
                let content;
                let _: syn::token::Paren = syn::parenthesized!(content in input);
                if content.is_empty() {
                    return Err(syn::Error::new(
                        key.span(),
                        "device(...) requires at least one capability token",
                    ));
                }
                let mut devices = Vec::new();
                loop {
                    if content.peek(LitStr) {
                        let token: LitStr = content.parse()?;
                        devices.push(token.value());
                    } else {
                        let ident: Ident = content.parse()?;
                        devices.push(ident.to_string());
                    }
                    if content.parse::<Token![,]>().is_err() {
                        break;
                    }
                }
                attr.devices = Some(devices);
            } else {
                return Err(syn::Error::new(
                    key.span(),
                    "unsupported `#[command]` key; supported keys: `name`, `capability`, `platform`, `error`, `device`",
                ));
            }
            if input.parse::<Token![,]>().is_err() {
                break;
            }
        }

        Ok(attr)
    }
}

/// Type이 `State<T>` 형태인지 검사하고 내부 `T`를 반환합니다.
fn extract_state_inner(ty: &Type) -> Option<Type> {
    let Type::Path(type_path) = ty else {
        return None;
    };
    let segment = type_path.path.segments.last()?;
    if segment.ident == "State"
        && let PathArguments::AngleBracketed(args) = &segment.arguments
        && let Some(GenericArgument::Type(inner_ty)) = args.args.first()
    {
        return Some(inner_ty.clone());
    }
    None
}

/// `Result<O>` 타입에서 내부 `O` 타입을 추출합니다.
///
/// `Result<O>`가 아니면 `None`을 반환합니다.
fn extract_result_inner(ty: &Type) -> Option<TokenStream2> {
    let Type::Path(type_path) = ty else {
        return None;
    };
    let segment = type_path.path.segments.last()?;
    if segment.ident != "Result" {
        return None;
    }
    let PathArguments::AngleBracketed(args) = &segment.arguments else {
        return None;
    };
    let GenericArgument::Type(inner_ty) = args.args.first()? else {
        return None;
    };
    Some(quote! { #inner_ty })
}

fn command_doc_comment(func: &ItemFn) -> String {
    func.attrs
        .iter()
        .filter_map(|attr| {
            if attr.path().is_ident("doc")
                && let syn::Meta::NameValue(nv) = &attr.meta
                && let syn::Expr::Lit(syn::ExprLit {
                    lit: syn::Lit::Str(s),
                    ..
                }) = &nv.value
            {
                return Some(s.value().trim().to_string());
            }
            None
        })
        .collect::<Vec<_>>()
        .join("\n")
}