feat(orchestrator): design_type — detect_design_type + presets

crates/op-orchestrator/src/design_type.rs

@@ -0,0 +1,216 @@
+//! 设计类型分类 —— port of
+//! `apps/web/src/services/ai/design-type-presets.ts`。
+//!
+//! `detect_design_type` 首个命中胜出:Component → Mobile →
+//! Desktop → 默认 LandingPage。每个类型带一个固定 preset。
+
+/// 四种设计类型(TS `DesignType` union)。
+#[derive(Debug, Clone, Copy, PartialEq, Eq)]
+pub enum DesignType {
+    MobileScreen,
+    DesktopScreen,
+    LandingPage,
+    Component,
+}
+
+/// 一个设计类型的尺寸 preset(TS `DesignTypePreset`)。
+/// `height` = 区块总高(0 = 按区块数自适应);`root_height` =
+/// 根 frame 显式高(0 = 自适应)。preset 上无 layout/gap/fill。
+#[derive(Debug, Clone, Copy)]
+pub struct DesignTypePreset {
+    pub type_: DesignType,
+    pub width: f64,
+    pub height: f64,
+    pub root_height: f64,
+    pub default_sections: &'static [&'static str],
+}
+
+const COMPONENT: DesignTypePreset = DesignTypePreset {
+    type_: DesignType::Component,
+    width: 400.0,
+    height: 0.0,
+    root_height: 0.0,
+    default_sections: &["Component"],
+};
+const MOBILE: DesignTypePreset = DesignTypePreset {
+    type_: DesignType::MobileScreen,
+    width: 375.0,
+    height: 812.0,
+    root_height: 812.0,
+    default_sections: &["Top Summary", "Main Content"],
+};
+const DESKTOP: DesignTypePreset = DesignTypePreset {
+    type_: DesignType::DesktopScreen,
+    width: 1200.0,
+    height: 800.0,
+    root_height: 800.0,
+    default_sections: &["Header", "Main Content", "Actions"],
+};
+const LANDING: DesignTypePreset = DesignTypePreset {
+    type_: DesignType::LandingPage,
+    width: 1200.0,
+    height: 0.0,
+    root_height: 0.0,
+    default_sections: &["Header", "Main Content", "Supporting Content", "Footer"],
+};
+
+/// 单组件触发词(Latin)—— 对齐 TS `COMPONENT_TRIGGER_LATIN_RE`。
+const COMPONENT_TRIGGER_LATIN: &[&str] = &[
+    "card", "badge", "chip", "tag", "tile", "pill", "label", "row", "item", "button", "toggle",
+    "switch", "selector", "modal", "dialog", "tooltip", "popover", "sheet", "widget", "panel",
+    "avatar", "stepper", "stat", "metric", "chart",
+];
+/// 单组件触发词(CJK)—— `COMPONENT_TRIGGER_CJK_RE`。
+const COMPONENT_TRIGGER_CJK: &[&str] = &[
+    "卡片",
+    "徽章",
+    "标签",
+    "按钮",
+    "开关",
+    "对话框",
+    "提示",
+    "气泡",
+    "图表",
+];
+/// 取消单组件资格的词 —— `COMPONENT_DISQUALIFIER_RE`。
+const COMPONENT_DISQUALIFIER: &[&str] = &[
+    "screen",
+    "page",
+    "app",
+    "home",
+    "onboarding",
+    "flow",
+    "mobile",
+    "phone",
+    "ios",
+    "android",
+    "dashboard",
+    "admin",
+    "workspace",
+    "console",
+    "网页",
+    "页面",
+    "屏幕",
+    "手机",
+    "移动端",
+    "管理",
+    "后台",
+    "控制台",
+    "工作台",
+    "工作区",
+];
+/// 移动端触发词 —— 命中 → MobileScreen。
+const MOBILE_WORDS: &[&str] = &["mobile", "手机", "phone", "移动端", "ios", "android"];
+/// 数据型工作区 / dashboard 触发词 —— 命中 → DesktopScreen。
+const DASHBOARD_WORDS: &[&str] = &[
+    "dashboard",
+    "admin",
+    "workspace",
+    "console",
+    "管理",
+    "后台",
+    "控制台",
+    "工作台",
+    "工作区",
+];
+
+/// `needle` 在 `haystack` 中出现 —— ASCII needle 按 JS `\b` 词边界
+/// 匹配(避免 "app" 误命中 "happen");含非 ASCII 的 needle(CJK)
+/// 按子串(TS 的 CJK 段本就无 `\b`)。`pub(crate)` —— Plan B 的
+/// `infer_tags_from_prompt` 复用同一 helper。
+pub(crate) fn contains_word(haystack: &str, needle: &str) -> bool {
+    if needle.is_empty() {
+        return false;
+    }
+    if !needle.is_ascii() {
+        return haystack.contains(needle);
+    }
+    let bytes = haystack.as_bytes();
+    let is_word = |b: u8| b.is_ascii_alphanumeric() || b == b'_';
+    let mut from = 0;
+    while let Some(rel) = haystack[from..].find(needle) {
+        let start = from + rel;
+        let end = start + needle.len();
+        let before_ok = start == 0 || !is_word(bytes[start - 1]);
+        let after_ok = end == bytes.len() || !is_word(bytes[end]);
+        if before_ok && after_ok {
+            return true;
+        }
+        from = start + 1;
+    }
+    false
+}
+
+/// `haystack`(已小写)是否含 `needles` 任一(按 `contains_word`)。
+fn contains_any(haystack: &str, needles: &[&str]) -> bool {
+    needles.iter().any(|n| contains_word(haystack, n))
+}
+
+/// 按 prompt 的意图分类 → preset。首个命中胜出。
+pub fn detect_design_type(prompt: &str) -> DesignTypePreset {
+    let lower = prompt.to_lowercase();
+    // ① 单组件:触发词命中 且 disqualifier 不命中。
+    let trigger = contains_any(&lower, COMPONENT_TRIGGER_LATIN)
+        || contains_any(&lower, COMPONENT_TRIGGER_CJK);
+    if trigger && !contains_any(&lower, COMPONENT_DISQUALIFIER) {
+        return COMPONENT;
+    }
+    // ② 移动端。
+    if contains_any(&lower, MOBILE_WORDS) {
+        return MOBILE;
+    }
+    // ③ 数据型工作区 / dashboard。
+    if contains_any(&lower, DASHBOARD_WORDS) {
+        return DESKTOP;
+    }
+    // ④ 默认:多区块落地页。
+    LANDING
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn component_detected_and_not_disqualified() {
+        // "profile card" → 触发词 card,无 disqualifier → Component
+        assert_eq!(
+            detect_design_type("a profile card").type_,
+            DesignType::Component
+        );
+        assert_eq!(
+            detect_design_type("一个统计徽章").type_,
+            DesignType::Component
+        );
+    }
+
+    #[test]
+    fn component_disqualified_by_screen_word() {
+        // "card" 触发,但 "screen" 是 disqualifier → 落到后续分类
+        let p = detect_design_type("a card on the home screen");
+        assert_eq!(p.type_, DesignType::LandingPage);
+    }
+
+    #[test]
+    fn mobile_detected() {
+        let p = detect_design_type("a mobile login screen");
+        assert_eq!(p.type_, DesignType::MobileScreen);
+        assert_eq!(p.width, 375.0);
+        assert_eq!(p.height, 812.0);
+    }
+
+    #[test]
+    fn dashboard_detected() {
+        let p = detect_design_type("an analytics dashboard");
+        assert_eq!(p.type_, DesignType::DesktopScreen);
+        assert_eq!(p.width, 1200.0);
+    }
+
+    #[test]
+    fn default_is_landing_page() {
+        let p = detect_design_type("a coffee brand site");
+        assert_eq!(p.type_, DesignType::LandingPage);
+        assert_eq!(p.width, 1200.0);
+        assert_eq!(p.root_height, 0.0);
+    }
+}

crates/op-orchestrator/src/lib.rs

@@ -7,6 +7,7 @@
 //! Plan B 提供"零件"(类型 + intent/plan/parse/normalize/variables);
 //! Plan C 在 `run` 模块接出四阶段主轴。
 
+pub mod design_type;
 pub mod intent;
 pub mod model_profile;
 pub mod parse;
@@ -24,6 +25,7 @@ pub mod subagent;
 #[cfg(test)]
 mod test_support;
 
+pub use design_type::{detect_design_type, DesignType, DesignTypePreset};
 pub use intent::classify_intent;
 pub use model_profile::{resolve_model_profile, ModelProfile, ModelTier};
 pub use run::Orchestrator;