flu-cli 0.0.4 → 2.0.0

package/lib/generators/state_manager_generator.js

@@ -0,0 +1,402 @@
+// 统一状态管理生成器：按模板与状态管理器选择执行依赖注入、入口增强、路由增强与变体生成
+// 原则：最小侵入、单一代码路径、可演进（便于后续扩展 riverpod 专用变体等）
+import fsx from 'fs-extra';
+import { join, dirname } from 'path';
+import { fileURLToPath } from 'url';
+import Handlebars from 'handlebars';
+
+import { logger } from '../utils/logger.js';
+import { detectProjectTemplate } from '../utils/project_detector.js';
+
+const __dirname = dirname(fileURLToPath(import.meta.url));
+
+/**
+ * 状态管理器生成器
+ * 功能：
+ * - 注入依赖（get/provider/flutter_riverpod）
+ * - 所有模板生成 BaseViewModel 变体（ChangeNotifier/GetX）
+ * - 所有模板按需增强入口（GetMaterialApp/ProviderScope）与路由（GetPage 列表）
+ */
+export class StateManagerGenerator {
+  /**
+   * 构造函数
+   * @param {string} projectPath 项目根目录
+   * @param {'default'|'provider'|'getx'|'riverpod'} stateManager 选中的状态管理器
+   * @param {{module?: string}} options 额外选项
+   */
+  constructor(projectPath, stateManager = 'default', options = {}) {
+    this.projectPath = projectPath;
+    this.stateManager = stateManager;
+    this.module = options.module || null;
+  }
+
+  /**
+   * 主流程：依赖 → BaseViewModel 生成 → 入口 → App/路由增强 → 完成
+   */
+  async generate () {
+    logger.info(`🔧 配置状态管理器: ${this.stateManager}`);
+    try {
+      await this.addDependencies();
+
+      // 所有模板都生成 BaseViewModel (统一策略)
+      await this.generateBaseViewModel();
+
+      await this.configureMain();
+      await this.configureAppWidget();
+      await this.configureRoutes();
+
+      logger.success('✅ 状态管理器配置完成');
+    } catch (error) {
+      logger.error(`❌ 状态管理器配置失败: ${error.message}`);
+      throw error;
+    }
+  }
+
+  /**
+   * 注入依赖到 pubspec.yaml：
+   * - 处理模板占位 {{{state_manager_dependency}}} / {{{flu_core_dependency}}}
+   * - 兼容非模板 pubspec，通过正则插入
+   */
+  async addDependencies () {
+    const pubspecPath = join(this.projectPath, 'pubspec.yaml');
+
+    if (!(await fsx.pathExists(pubspecPath))) {
+      throw new Error(`pubspec.yaml not found at ${pubspecPath}`);
+    }
+
+    let content = await fsx.readFile(pubspecPath, 'utf8');
+    const dependency = this.getDependency();
+
+    // 移除 flu_core_dependency 占位符
+    if (content.includes('{{{flu_core_dependency}}}')) {
+      content = content.replace(/\s*\{{{flu_core_dependency}}}\s*\n?/, '');
+    }
+
+    if (content.includes('{{{state_manager_dependency}}}')) {
+      content = content.replace('{{{state_manager_dependency}}}', dependency);
+    } else if (dependency) {
+      content = content.replace(/(dependencies:\s*\n(?:[\s\S]*?))(\n\w)/, `$1\n    ${dependency}$2`);
+    }
+
+    await fsx.writeFile(pubspecPath, content);
+    if (dependency) logger.info(`  ✓ 添加依赖: ${dependency.split(':')[0]}`);
+  }
+
+  /**
+   * 返回选中状态管理器的依赖字符串
+   */
+  getDependency () {
+    const map = {
+      default: '',
+      provider: 'provider: ^6.1.1',
+      getx: 'get: ^4.6.6',
+      riverpod: 'flutter_riverpod: ^2.4.9'
+    };
+    return map[this.stateManager] || '';
+  }
+
+  /**
+   * 为所有模板生成 BaseViewModel 变体并重导出统一入口
+   */
+  async generateBaseViewModel () {
+    const { detectProjectTemplate } = await import('../utils/project_detector.js');
+    const tpl = detectProjectTemplate(this.projectPath);
+
+    // 根据模板类型确定目标路径
+    let baseDir;
+    if (tpl === 'lite') {
+      baseDir = join(this.projectPath, 'lib', 'base');
+    } else if (tpl === 'modular') {
+      baseDir = join(this.projectPath, 'lib', 'core', 'base');
+    } else if (tpl === 'clean') {
+      baseDir = join(this.projectPath, 'lib', 'core', 'base');
+    } else {
+      return;  // 未知模板，跳过
+    }
+
+    await fsx.ensureDir(baseDir);
+    const target = join(baseDir, 'base_viewmodel.dart');
+
+    const content = this.buildBaseViewModelContent();
+    await fsx.writeFile(target, content);
+
+    const relPath = target.replace(this.projectPath + '/', '');
+    logger.info(`  ✓ 生成 BaseViewModel: ${relPath}`);
+  }
+
+  /**
+   * 构建 BaseViewModel 变体源码：
+   * - getx：GetxController + 自管理监听列表（兼容 BasePage 的 addListener/removeListener）
+   * - default/provider：ChangeNotifier 变体
+   */
+  buildBaseViewModelContent () {
+
+
+    const commonInfo = `
+  ViewState _state = ViewState.idle;
+  String? _errorMessage;
+  bool _isRefreshing = false;
+ 
+  // ==================== Getters ====================
+  // 当前视图状态
+  ViewState get state => _state;
+  // 错误信息
+  String? get errorMessage => _errorMessage;
+  // 是否正在加载
+  bool get isLoading => _state == ViewState.loading;
+  // 是否加载失败
+  bool get isError => _state == ViewState.error;
+  // 是否加载成功
+  bool get isSuccess => _state == ViewState.success;
+  // 是否空闲状态
+  bool get isIdle => _state == ViewState.idle;
+  // 是否正在刷新
+  bool get isRefreshing => _isRefreshing;
+
+
+  // ==================== 状态设置 ====================
+
+  /// 设置视图状态
+  ///
+  /// [state] 新的状态
+  /// [error] 错误信息（仅在 error 状态时有效）
+  void setState(ViewState state, {String? error}) {
+    _state = state;
+    _errorMessage = error;
+    notifyListeners();
+  }
+  
+  /// 设置刷新状态
+  void setRefreshing(bool refreshing) {
+    _isRefreshing = refreshing;
+    notifyListeners();
+  }
+ 
+
+`;
+    if (this.stateManager === 'getx') {
+      return `import 'package:get/get.dart';
+enum ViewState { idle, loading, success, error }
+class BaseViewModel extends GetxController {
+  ${commonInfo}
+
+  ///  刷新数据（通知所有监听器）
+  void notifyListeners() {
+    notifyDataChange();
+  }
+  /// 通知数据变化（仅用于局部刷新）
+  void notifyDataChange() {
+    update();
+  }
+
+  // ==================== 生命周期 ====================
+  /// 初始化钩子（函数级注释）
+  @override
+  Future<void> onInit() async {  
+    super.onInit();
+    // 子类可以重写此方法进行初始化
+  }
+
+  /// 刷新数据（钩子）
+  Future<void> refreshData() async {  
+    // 子类可以重写此方法实现刷新逻辑
+  }
+}
+`;
+    }
+
+    // provider 与 default 采用 ChangeNotifier 变体
+    return `import 'package:flutter/foundation.dart';
+enum ViewState { idle, loading, success, error }
+
+class BaseViewModel extends ChangeNotifier {
+  ${commonInfo}
+
+    /// 通知数据变化（仅用于局部刷新）
+  void notifyDataChange() {
+    notifyListeners();
+  }
+
+  // ==================== 生命周期 ====================
+  /// 初始化钩子（函数级注释）
+  Future<void> onInit() async {
+     // 子类可以重写此方法进行初始化
+  }
+
+  /// 刷新数据（钩子）
+  Future<void> refreshData() async {  
+    // 子类可以重写此方法实现刷新逻辑
+  }
+}
+`;
+  }
+
+  /**
+   * 配置入口 main.dart：
+   * - 模板文件执行 Handlebars 渲染（只处理 riverpod 条件）
+   * - 非模板文件无需额外处理
+   */
+  async configureMain () {
+    const mainPath = join(this.projectPath, 'lib', 'main.dart');
+    if (!(await fsx.pathExists(mainPath))) {
+      throw new Error(`main.dart not found at ${mainPath}`);
+    }
+
+    const raw = await fsx.readFile(mainPath, 'utf8');
+
+    // 检查是否为模板文件
+    const isTemplate = raw.includes('{{#if_riverpod}}');
+    if (isTemplate) {
+      this.registerHandlebarsHelpers();
+      const compiled = Handlebars.compile(raw);
+      const rendered = compiled({
+        if_riverpod: this.stateManager === 'riverpod'
+      });
+      await fsx.writeFile(mainPath, rendered);
+    }
+  }
+
+  /**
+   * 增强 App 入口组件（GetX）：
+   * - lite/clean：lib/app.dart → MaterialApp 替换为 GetMaterialApp，注入 get 导入与 getPages
+   * - modular：lib/main.dart → MaterialApp 替换为 GetMaterialApp，注入 get 导入与 getPages
+   */
+  async configureAppWidget () {
+    try {
+      const tpl = detectProjectTemplate(this.projectPath);
+      const isGetX = this.stateManager === 'getx';
+      if (!isGetX) return;
+
+      if (tpl === 'lite' || tpl === 'clean') {
+        const appPath = join(this.projectPath, 'lib', 'app.dart');
+        if (!(await fsx.pathExists(appPath))) return;
+        let raw = await fsx.readFile(appPath, 'utf8');
+        if (!raw.includes('GetMaterialApp(')) {
+          raw = raw.replace(/MaterialApp\s*\(/, 'GetMaterialApp(');
+        }
+        if (!/package:get\/get\.dart/.test(raw)) {
+          raw = `import 'package:get/get.dart';\n` + raw;
+        }
+        const pagesAccessor = tpl === 'lite' ? 'AppRoutes.pages' : 'AppRoutes.pages';
+        if (!new RegExp(`getPages:\\s*${pagesAccessor}`).test(raw)) {
+          raw = raw.replace(/initialRoute:\s*AppRoutes\.home,/, `initialRoute: AppRoutes.home,\n      getPages: ${pagesAccessor},`);
+        }
+        raw = raw.replace(/\s*routes:\s*AppRoutes\.routes,\s*\n/, '\n');
+        await fsx.writeFile(appPath, raw);
+      } else if (tpl === 'modular') {
+        const mainPath = join(this.projectPath, 'lib', 'main.dart');
+        if (!(await fsx.pathExists(mainPath))) return;
+        let raw = await fsx.readFile(mainPath, 'utf8');
+        if (!raw.includes('GetMaterialApp(')) {
+          raw = raw.replace(/MaterialApp\s*\(/, 'GetMaterialApp(');
+        }
+        if (!/package:get\/get\.dart/.test(raw)) {
+          raw = raw.replace(/^(import\s+['"].+?['"];\s*\n)+/m, (m) => `${m}import 'package:get/get.dart';\n`);
+          if (!raw.includes("import 'package:get/get.dart';")) {
+            raw = `import 'package:get/get.dart';\n` + raw;
+          }
+        }
+        if (!/getPages:\s*AppRouter\.pages/.test(raw)) {
+          raw = raw.replace(/initialRoute:\s*AppRouter\.home,/, "initialRoute: AppRouter.home,\n      getPages: AppRouter.pages,");
+        }
+        raw = raw.replace(/\s*routes:\s*AppRouter\.routes,\s*\n/, '\n');
+        await fsx.writeFile(mainPath, raw);
+      }
+    } catch (_) { }
+  }
+
+  /**
+   * 增强路由文件(GetX):为不同模板注入 List<GetPage> pages 与 get 导入
+   * 智能转换原有路由定义,保留所有路由
+   */
+  async configureRoutes () {
+    try {
+      const tpl = detectProjectTemplate(this.projectPath);
+      if (this.stateManager !== 'getx') return;
+      let routesPath;
+      if (tpl === 'lite') {
+        routesPath = join(this.projectPath, 'lib', 'config', 'routes.dart');
+      } else if (tpl === 'modular') {
+        routesPath = join(this.projectPath, 'lib', 'core', 'router', 'app_router.dart');
+      } else if (tpl === 'clean') {
+        routesPath = join(this.projectPath, 'lib', 'config', 'routes', 'app_routes.dart');
+      } else {
+        return;
+      }
+      if (!(await fsx.pathExists(routesPath))) return;
+      let raw = await fsx.readFile(routesPath, 'utf8');
+
+      // 替换 flutter import 为 get import
+      if (!/package:get\/get\.dart/.test(raw)) {
+        raw = raw.replace("import 'package:flutter/material.dart';\n", "import 'package:get/get.dart';\n");
+      }
+
+      // 如果已经是 GetPage 格式,跳过
+      if (/static\s+List<\s*GetPage/.test(raw)) {
+        return;
+      }
+
+      // 提取路由常量定义
+      const routeConstantsMatch = raw.match(/static\s+const\s+String\s+\w+\s*=\s*['"][^'"]+['"];/g);
+      const routeConstants = routeConstantsMatch ? routeConstantsMatch.join('\n  ') : "static const String home = '/';";
+
+      // 提取原有的 routes map 并转换为 GetPage
+      let getPages = [];
+      const routesMapMatch = raw.match(/routes:\s*\(context\)\s*=>\s*const\s+(\w+)\(\)/g) ||
+        raw.match(/(\w+):\s*\(context\)\s*=>\s*const\s+(\w+)\(\)/g);
+
+      if (routesMapMatch) {
+        // 从 routes getter 中提取
+        const routesGetterMatch = raw.match(/static\s+Map<String,\s*WidgetBuilder>\s+get\s+routes\s*\{[^}]*return\s*\{([^}]+)\};/s);
+        if (routesGetterMatch) {
+          const routesContent = routesGetterMatch[1];
+          // 解析每个路由
+          const routePattern = /(\w+):\s*\(context\)\s*=>\s*const\s+(\w+)\(\)/g;
+          let match;
+          while ((match = routePattern.exec(routesContent)) !== null) {
+            const [, routeName, widgetName] = match;
+            getPages.push(`      GetPage(name: ${routeName}, page: () => const ${widgetName}())`);
+          }
+        }
+      }
+
+      // 如果没有提取到任何路由,使用默认的 home
+      if (getPages.length === 0) {
+        getPages.push(`      GetPage(name: home, page: () => const HomePage())`);
+      }
+
+      const className = raw.match(/class\s+(\w+)/)?.[1] || 'AppRoutes';
+
+      // 构建新的 class
+      const newClass = `class ${className} {
+  ${className}._();
+
+  ${routeConstants}
+
+  static List<GetPage<dynamic>> get pages {
+    return [
+${getPages.join(',\n')},
+    ];
+  }
+}
+`;
+
+      // 替换整个 class
+      raw = raw.replace(/class\s+\w+[\s\S]*?^}/m, newClass);
+
+      await fsx.writeFile(routesPath, raw);
+    } catch (error) {
+      logger.warn(`  ⚠️ 路由配置可能需要手动调整: ${error.message}`);
+    }
+  }
+
+  /**
+   * 注册 Handlebars 条件助手：控制模板片段的渲染/隐藏
+   */
+  registerHandlebarsHelpers () {
+    Handlebars.registerHelper('if_riverpod', function (options) {
+      return options.data.root.if_riverpod ? options.fn(this) : options.inverse(this);
+    });
+  }
+
+}

package/lib/generators/viewmodel_generator.js

@@ -0,0 +1,115 @@
+/**
+ * ViewModel 生成器
+ */
+
+import { existsSync, mkdirSync, writeFileSync } from 'fs';
+import { join } from 'path';
+import { logger } from '../utils/logger.js';
+import { toPascalCase, toSnakeCase } from '../utils/string_helper.js';
+import { updateIndexFile } from '../utils/index_updater.js';
+
+/**
+ * 生成 ViewModel 文件
+ */
+export function generateViewModel (name, options = {}) {
+  try {
+    const {
+      feature = null,
+      outputDir = process.cwd()
+    } = options;
+
+    // 转换命名
+    const namePascal = toPascalCase(name);
+    const nameSnake = toSnakeCase(name);
+
+    // 确定输出路径
+    let vmDir;
+    if (feature) {
+      // Modular/Clean 架构
+      vmDir = join(outputDir, 'lib', 'features', feature, 'viewmodels');
+    } else {
+      // Lite 架构
+      vmDir = join(outputDir, 'lib', 'viewmodels');
+    }
+
+    // 创建目录
+    if (!existsSync(vmDir)) {
+      mkdirSync(vmDir, { recursive: true });
+    }
+
+    // 生成文件内容
+    const content = generateViewModelContent(namePascal);
+
+    // 写入文件
+    const filePath = join(vmDir, `${nameSnake}_viewmodel.dart`);
+    if (existsSync(filePath)) {
+      logger.error(`文件已存在: ${filePath}`);
+      return false;
+    }
+
+    writeFileSync(filePath, content, 'utf8');
+    logger.success(`ViewModel 创建成功: ${filePath}`);
+
+    // 更新 index.dart
+    updateIndexFile(vmDir, `${nameSnake}_viewmodel.dart`);
+
+    return true;
+
+  } catch (error) {
+    logger.error(`生成 ViewModel 失败: ${error.message}`);
+    return false;
+  }
+}
+
+/**
+ * 生成 ViewModel 内容
+ */
+function generateViewModelContent (namePascal) {
+  return `import 'package:flutter/foundation.dart';
+
+class ${namePascal}ViewModel extends ChangeNotifier {
+  // 状态
+  bool _isLoading = false;
+  bool get isLoading => _isLoading;
+
+  String? _error;
+  String? get error => _error;
+
+  // 初始化
+  void init() {
+    // 初始化逻辑
+    loadData();
+  }
+
+  // 加载数据
+  Future<void> loadData() async {
+    _isLoading = true;
+    _error = null;
+    notifyListeners();
+
+    try {
+      // TODO: 实现数据加载逻辑
+      await Future.delayed(const Duration(seconds: 1));
+      
+      _isLoading = false;
+      notifyListeners();
+    } catch (e) {
+      _error = e.toString();
+      _isLoading = false;
+      notifyListeners();
+    }
+  }
+
+  // 刷新
+  Future<void> refresh() async {
+    await loadData();
+  }
+
+  @override
+  void dispose() {
+    // 清理资源
+    super.dispose();
+  }
+}
+`;
+}

package/lib/generators/widget_generator.js

@@ -0,0 +1,104 @@
+/**
+ * Widget 生成器
+ */
+
+import { existsSync, mkdirSync, writeFileSync } from 'fs';
+import { join } from 'path';
+import { logger } from '../utils/logger.js';
+import { toPascalCase, toSnakeCase } from '../utils/string_helper.js';
+import { updateIndexFile } from '../utils/index_updater.js';
+import { detectProjectTemplate, getWidgetPath } from '../utils/project_detector.js';
+import { getSnippetContent } from '../utils/snippet_loader.js';
+
+/**
+ * 生成 Widget 文件
+ */
+export function generateWidget (name, options = {}) {
+  try {
+    const {
+      feature = null,
+      stateful = false,
+      outputDir = process.cwd()
+    } = options;
+
+    // 检测项目模板类型
+    const template = detectProjectTemplate(outputDir);
+    logger.info(`检测到项目类型: ${template || '未知'}`);
+
+    // 转换命名
+    const namePascal = toPascalCase(name);
+    const nameSnake = toSnakeCase(name);
+
+    // 根据模板类型确定输出路径
+    const widgetsDir = getWidgetPath(outputDir, feature);
+
+    // 创建目录
+    if (!existsSync(widgetsDir)) {
+      mkdirSync(widgetsDir, { recursive: true });
+    }
+
+    // 生成文件内容（优先片段）
+    const key = stateful ? 'flu.stWidget' : 'flu.lessWidget';
+    const contentFromSnippet = getSnippetContent(outputDir, key, { Name: namePascal });
+    const content = contentFromSnippet || generateWidgetContent(namePascal, stateful);
+
+    // 写入文件
+    const filePath = join(widgetsDir, `${nameSnake}_widget.dart`);
+    if (existsSync(filePath)) {
+      logger.error(`文件已存在: ${filePath}`);
+      return false;
+    }
+
+    writeFileSync(filePath, content, 'utf8');
+    logger.success(`Widget 创建成功: ${filePath}`);
+
+    // 更新 index.dart
+    updateIndexFile(widgetsDir, `${nameSnake}_widget.dart`);
+
+    return true;
+
+  } catch (error) {
+    logger.error(`生成 Widget 失败: ${error.message}`);
+    return false;
+  }
+}
+
+/**
+ * 生成 Widget 内容
+ */
+function generateWidgetContent (namePascal, stateful) {
+  if (stateful) {
+    return `import 'package:flutter/material.dart';
+
+class ${namePascal}Widget extends StatefulWidget {
+  const ${namePascal}Widget({super.key});
+
+  @override
+  State<${namePascal}Widget> createState() => _${namePascal}WidgetState();
+}
+
+class _${namePascal}WidgetState extends State<${namePascal}Widget> {
+  @override
+  Widget build(BuildContext context) {
+    return Container(
+      child: const Text('${namePascal}Widget'),
+    );
+  }
+}
+`;
+  } else {
+    return `import 'package:flutter/material.dart';
+
+class ${namePascal}Widget extends StatelessWidget {
+  const ${namePascal}Widget({super.key});
+
+  @override
+  Widget build(BuildContext context) {
+    return Container(
+      child: const Text('${namePascal}Widget'),
+    );
+  }
+}
+`;
+  }
+}