326 lines
8.9 KiB
JavaScript
326 lines
8.9 KiB
JavaScript
#!/usr/bin/env node
|
|
const fs = require('fs');
|
|
const path = require('path');
|
|
|
|
function findWorkspaceRoot(startDir) {
|
|
let current = startDir;
|
|
while (true) {
|
|
if (fs.existsSync(path.join(current, 'workspace.yml'))) {
|
|
return current;
|
|
}
|
|
const parent = path.dirname(current);
|
|
if (parent === current) {
|
|
throw new Error('workspace.yml not found from the provided start directory');
|
|
}
|
|
current = parent;
|
|
}
|
|
}
|
|
|
|
function readText(filePath) {
|
|
return fs.readFileSync(filePath, 'utf8');
|
|
}
|
|
|
|
function stripQuotes(value) {
|
|
const trimmed = String(value).trim();
|
|
if ((trimmed.startsWith('"') && trimmed.endsWith('"')) || (trimmed.startsWith("'") && trimmed.endsWith("'"))) {
|
|
return trimmed.slice(1, -1);
|
|
}
|
|
return trimmed;
|
|
}
|
|
|
|
function parseWorkspace(workspacePath) {
|
|
const text = readText(workspacePath);
|
|
const lines = text.split(/\r?\n/);
|
|
const collections = [];
|
|
let inCollections = false;
|
|
let currentCollection = null;
|
|
|
|
for (const line of lines) {
|
|
const trimmed = line.trim();
|
|
if (!inCollections && trimmed === 'collections:') {
|
|
inCollections = true;
|
|
continue;
|
|
}
|
|
|
|
if (!inCollections) {
|
|
continue;
|
|
}
|
|
|
|
if (!line.startsWith(' ') && !line.startsWith('\t') && trimmed) {
|
|
break;
|
|
}
|
|
|
|
const nameMatch = line.match(/^\s*-\s+name:\s*(.+)$/);
|
|
if (nameMatch) {
|
|
currentCollection = { name: stripQuotes(nameMatch[1]) };
|
|
collections.push(currentCollection);
|
|
continue;
|
|
}
|
|
|
|
const pathMatch = line.match(/^\s*path:\s*(.+)$/);
|
|
if (pathMatch && currentCollection) {
|
|
currentCollection.path = stripQuotes(pathMatch[1]);
|
|
}
|
|
}
|
|
|
|
return { collections };
|
|
}
|
|
|
|
function sanitizeVarName(value) {
|
|
return String(value)
|
|
.trim()
|
|
.replace(/[{}]/g, '')
|
|
.replace(/[^A-Za-z0-9_]/g, '_')
|
|
.replace(/^([0-9])/, '_$1') || 'value';
|
|
}
|
|
|
|
function collectPlaceholders(value) {
|
|
if (typeof value !== 'string') {
|
|
return [];
|
|
}
|
|
const placeholders = [];
|
|
const regex = /\{\{([^{}]+)\}\}/g;
|
|
let match;
|
|
while ((match = regex.exec(value)) !== null) {
|
|
placeholders.push(match[1]);
|
|
}
|
|
return placeholders;
|
|
}
|
|
|
|
function findBlock(lines, keyName) {
|
|
for (let i = 0; i < lines.length; i += 1) {
|
|
const trimmed = lines[i].trim();
|
|
if (trimmed !== keyName && !trimmed.startsWith(`${keyName}:`)) {
|
|
continue;
|
|
}
|
|
|
|
const lineIndent = lines[i].match(/^\s*/)[0].length;
|
|
const block = [];
|
|
for (let j = i + 1; j < lines.length; j += 1) {
|
|
const currentLine = lines[j];
|
|
const currentTrimmed = currentLine.trim();
|
|
if (!currentTrimmed) {
|
|
block.push(currentLine);
|
|
continue;
|
|
}
|
|
const currentIndent = currentLine.match(/^\s*/)[0].length;
|
|
if (currentIndent <= lineIndent && !currentLine.startsWith(' ')) {
|
|
break;
|
|
}
|
|
if (currentIndent <= lineIndent && currentTrimmed.startsWith('#')) {
|
|
block.push(currentLine);
|
|
continue;
|
|
}
|
|
if (currentIndent <= lineIndent) {
|
|
break;
|
|
}
|
|
block.push(currentLine);
|
|
}
|
|
return block;
|
|
}
|
|
return [];
|
|
}
|
|
|
|
function parseRequestInfo(text) {
|
|
const lines = text.split(/\r?\n/);
|
|
const infoLines = findBlock(lines, 'info');
|
|
const nameMatch = infoLines.join('\n').match(/^\s*name:\s*(.+)$/m);
|
|
return nameMatch ? stripQuotes(nameMatch[1]) : '';
|
|
}
|
|
|
|
function parseHttpBlock(text) {
|
|
const lines = text.split(/\r?\n/);
|
|
const blockLines = findBlock(lines, 'http');
|
|
const blockText = blockLines.join('\n');
|
|
const methodMatch = blockText.match(/^\s*method:\s*(.+)$/m);
|
|
const urlMatch = blockText.match(/^\s*url:\s*(.+)$/m);
|
|
|
|
const params = [];
|
|
const paramLines = blockLines.join('\n').split(/\r?\n/);
|
|
let inParamsBlock = false;
|
|
let paramsIndent = 0;
|
|
let currentParam = null;
|
|
|
|
for (const line of paramLines) {
|
|
const trimmed = line.trim();
|
|
if (!inParamsBlock) {
|
|
if (trimmed === 'params:') {
|
|
inParamsBlock = true;
|
|
paramsIndent = line.match(/^\s*/)[0].length;
|
|
}
|
|
continue;
|
|
}
|
|
|
|
if (!trimmed) {
|
|
continue;
|
|
}
|
|
|
|
const indent = line.match(/^\s*/)[0].length;
|
|
if (indent <= paramsIndent) {
|
|
break;
|
|
}
|
|
|
|
const listItemMatch = line.match(/^\s*-\s+name:\s*(.+)$/);
|
|
if (listItemMatch) {
|
|
currentParam = { name: stripQuotes(listItemMatch[1]) };
|
|
params.push(currentParam);
|
|
continue;
|
|
}
|
|
|
|
if (!currentParam) {
|
|
continue;
|
|
}
|
|
|
|
const propertyMatch = line.match(/^\s*([A-Za-z0-9_]+):\s*(.+)$/);
|
|
if (propertyMatch) {
|
|
const [, key, value] = propertyMatch;
|
|
currentParam[key] = stripQuotes(value);
|
|
}
|
|
}
|
|
|
|
return {
|
|
method: methodMatch ? stripQuotes(methodMatch[1]) : 'GET',
|
|
url: urlMatch ? stripQuotes(urlMatch[1]) : '',
|
|
params,
|
|
};
|
|
}
|
|
|
|
function buildRequestContent(request, requestName) {
|
|
const lines = [];
|
|
const placeholders = new Set();
|
|
|
|
const addPlaceholders = (value) => {
|
|
for (const placeholder of collectPlaceholders(value)) {
|
|
placeholders.add(placeholder);
|
|
}
|
|
};
|
|
|
|
let url = request.url || '';
|
|
if (url) {
|
|
addPlaceholders(url);
|
|
url = url.replace(/:([A-Za-z0-9_]+)/g, (_, name) => `{{${name}}}`);
|
|
}
|
|
|
|
const queryParams = [];
|
|
const headers = [];
|
|
for (const param of request.params || []) {
|
|
const name = param.name || '';
|
|
const value = param.value || '';
|
|
const type = (param.type || 'query').toLowerCase();
|
|
const disabled = String(param.disabled).toLowerCase() === 'true';
|
|
|
|
if (disabled) {
|
|
continue;
|
|
}
|
|
|
|
addPlaceholders(value);
|
|
if (type === 'header') {
|
|
headers.push({ name, value });
|
|
} else {
|
|
queryParams.push({ name, value });
|
|
}
|
|
}
|
|
|
|
if (placeholders.size > 0) {
|
|
lines.push(`# Variables for ${requestName}`);
|
|
for (const placeholder of [...placeholders].sort()) {
|
|
lines.push(`@${sanitizeVarName(placeholder)} = YOUR_VALUE_HERE`);
|
|
}
|
|
lines.push('');
|
|
}
|
|
|
|
const method = (request.method || 'GET').toUpperCase();
|
|
let requestUrl = url;
|
|
for (const param of queryParams) {
|
|
if (!param.name) {
|
|
continue;
|
|
}
|
|
const separator = requestUrl.includes('?') ? '&' : '?';
|
|
requestUrl = `${requestUrl}${separator}${param.name}=${param.value}`;
|
|
}
|
|
|
|
lines.push(`${method} ${requestUrl}`);
|
|
for (const header of headers) {
|
|
lines.push(`${header.name}: ${header.value}`);
|
|
}
|
|
return lines.join('\n');
|
|
}
|
|
|
|
function ensureDir(dirPath) {
|
|
fs.mkdirSync(dirPath, { recursive: true });
|
|
}
|
|
|
|
function walkYamlFiles(rootDir) {
|
|
const results = [];
|
|
const entries = fs.readdirSync(rootDir, { withFileTypes: true });
|
|
for (const entry of entries) {
|
|
if (entry.name.startsWith('.') || entry.name === 'node_modules') {
|
|
continue;
|
|
}
|
|
const fullPath = path.join(rootDir, entry.name);
|
|
if (entry.isDirectory()) {
|
|
results.push(...walkYamlFiles(fullPath));
|
|
} else if (entry.isFile() && /\.ya?ml$/i.test(entry.name)) {
|
|
results.push(fullPath);
|
|
}
|
|
}
|
|
return results;
|
|
}
|
|
|
|
function main() {
|
|
const workspaceRoot = findWorkspaceRoot(__dirname);
|
|
const workspaceFile = path.join(workspaceRoot, 'workspace.yml');
|
|
const workspace = parseWorkspace(workspaceFile);
|
|
const collections = Array.isArray(workspace.collections) ? workspace.collections : [];
|
|
|
|
if (collections.length === 0) {
|
|
throw new Error('No collections found in workspace.yml');
|
|
}
|
|
|
|
for (const collection of collections) {
|
|
if (!collection || !collection.name || !collection.path) {
|
|
continue;
|
|
}
|
|
|
|
const sourceDir = path.join(workspaceRoot, collection.path);
|
|
if (!fs.existsSync(sourceDir)) {
|
|
console.warn(`Skipping missing collection path: ${collection.path}`);
|
|
continue;
|
|
}
|
|
|
|
const outputRoot = path.join(workspaceRoot, 'autodocs', 'http', collection.name);
|
|
ensureDir(outputRoot);
|
|
|
|
const yamlFiles = walkYamlFiles(sourceDir);
|
|
let processed = 0;
|
|
|
|
for (const yamlFile of yamlFiles) {
|
|
const content = readText(yamlFile);
|
|
const requestName = parseRequestInfo(content) || path.basename(yamlFile, path.extname(yamlFile));
|
|
const httpBlock = parseHttpBlock(content);
|
|
if (!httpBlock || !httpBlock.url) {
|
|
continue;
|
|
}
|
|
|
|
const relativePath = path.relative(sourceDir, yamlFile);
|
|
const parsedPath = path.parse(relativePath);
|
|
const targetDir = path.join(outputRoot, parsedPath.dir);
|
|
ensureDir(targetDir);
|
|
|
|
const outputFile = path.join(targetDir, `${parsedPath.name}.http`);
|
|
const requestContent = buildRequestContent(httpBlock, requestName);
|
|
fs.writeFileSync(outputFile, `${requestContent}\n`, 'utf8');
|
|
processed += 1;
|
|
}
|
|
|
|
console.log(`Generated ${processed} .http file(s) for ${collection.name}`);
|
|
}
|
|
}
|
|
|
|
try {
|
|
main();
|
|
} catch (error) {
|
|
console.error(error.message);
|
|
process.exit(1);
|
|
}
|