在Hygen模板生成器中动态获取文件目录结构

Hygen作为一款高效的代码生成器工具，提供了强大的模板功能。在实际开发中，我们经常需要根据现有文件结构动态生成代码，而不是简单地使用静态模板。本文将介绍如何在Hygen中嵌入Node.js代码来获取目录结构，实现更智能的模板生成。

动态参数注入原理

Hygen允许通过prompt.js文件向模板注入动态参数。这个特性不仅限于简单的用户输入，还可以执行任意Node.js代码来准备模板所需的数据。核心机制是通过params函数处理传入参数，并将计算结果附加到args对象上。

实现文件目录扫描

以下是一个实用的实现示例，展示了如何扫描指定目录并过滤出符合条件的文件夹：

const fs = require('fs');
const BASE_ASSETS_FOLDER = './raw';

module.exports = {
  params: ({args}) => {
    // 读取基础目录下的所有条目
    // 过滤掉包含点的条目（通常是文件而非文件夹）
    args.assets = fs.readdirSync(BASE_ASSETS_FOLDER)
                  .filter(folder => !folder.includes('.'));
    return args;
  }
}

这段代码会扫描./raw目录，获取所有不包含点号(.)的条目（通常就是子目录），然后将结果数组赋值给args.assets变量。

模板中使用动态数据

在Hygen模板中，可以直接使用注入的变量。例如，可以这样遍历所有找到的资产目录：

// 模板文件示例
---
to: src/assets/index.js
---
<% for (const asset of assets) { %>
import <%= asset %> from './<%= asset %>';
<% } %>

实际应用场景

这种技术特别适用于以下场景：

1. 自动化生成资源索引文件
2. 根据现有模块结构生成路由配置
3. 为项目中的每个组件生成测试文件
4. 创建多语言资源映射

注意事项

1. 文件操作是同步的，对于大型目录结构考虑性能影响
2. 确保Hygen有足够的权限访问目标目录
3. 错误处理很重要，特别是当目录可能不存在时
4. 路径处理最好使用path模块而非硬编码

通过这种方式，Hygen从简单的模板工具升级为能够感知项目现状的智能生成器，大幅提升了开发效率和代码一致性。