Files
antigravity-skills-reference/scripts/normalize-frontmatter.js

152 lines
4.2 KiB
JavaScript

const fs = require('fs');
const path = require('path');
const yaml = require('yaml');
const { listSkillIds, parseFrontmatter } = require('../lib/skill-utils');
const ROOT = path.resolve(__dirname, '..');
const SKILLS_DIR = path.join(ROOT, 'skills');
const ALLOWED_FIELDS = new Set([
'name',
'description',
'risk',
'source',
'license',
'compatibility',
'metadata',
'allowed-tools',
]);
function isPlainObject(value) {
return value && typeof value === 'object' && !Array.isArray(value);
}
function coerceToString(value) {
if (value === null || value === undefined) return '';
if (typeof value === 'string') return value.trim();
if (typeof value === 'number' || typeof value === 'boolean') return String(value);
if (Array.isArray(value)) {
const simple = value.every(item => ['string', 'number', 'boolean'].includes(typeof item));
return simple ? value.map(item => String(item).trim()).filter(Boolean).join(', ') : JSON.stringify(value);
}
if (isPlainObject(value)) {
return JSON.stringify(value);
}
return String(value).trim();
}
function appendMetadata(metadata, key, value) {
const nextValue = coerceToString(value);
if (!nextValue) return;
if (!metadata[key]) {
metadata[key] = nextValue;
return;
}
if (metadata[key].includes(nextValue)) return;
metadata[key] = `${metadata[key]}, ${nextValue}`;
}
function collectAllowedTools(value, toolSet) {
if (!value) return;
if (typeof value === 'string') {
value
.split(/[\s,]+/)
.map(token => token.trim())
.filter(Boolean)
.forEach(token => toolSet.add(token));
return;
}
if (Array.isArray(value)) {
value
.map(token => String(token).trim())
.filter(Boolean)
.forEach(token => toolSet.add(token));
}
}
function normalizeSkill(skillId) {
const skillPath = path.join(SKILLS_DIR, skillId, 'SKILL.md');
const content = fs.readFileSync(skillPath, 'utf8');
const { data, body, hasFrontmatter } = parseFrontmatter(content);
if (!hasFrontmatter) return false;
let modified = false;
const updated = { ...data };
const metadata = isPlainObject(updated.metadata) ? { ...updated.metadata } : {};
if (updated.metadata !== undefined && !isPlainObject(updated.metadata)) {
appendMetadata(metadata, 'legacy_metadata', updated.metadata);
modified = true;
}
const allowedTools = new Set();
collectAllowedTools(updated['allowed-tools'], allowedTools);
collectAllowedTools(updated.tools, allowedTools);
collectAllowedTools(updated.tool_access, allowedTools);
if (updated.tools !== undefined) {
delete updated.tools;
modified = true;
}
if (updated.tool_access !== undefined) {
delete updated.tool_access;
modified = true;
}
for (const key of Object.keys(updated)) {
if (ALLOWED_FIELDS.has(key)) continue;
if (key === 'tags') {
appendMetadata(metadata, 'tags', updated[key]);
} else {
appendMetadata(metadata, key, updated[key]);
}
delete updated[key];
modified = true;
}
if (allowedTools.size) {
updated['allowed-tools'] = Array.from(allowedTools).join(' ');
modified = true;
} else if (updated['allowed-tools'] !== undefined) {
delete updated['allowed-tools'];
modified = true;
}
if (Object.keys(metadata).length) {
updated.metadata = metadata;
modified = true;
} else if (updated.metadata !== undefined) {
delete updated.metadata;
modified = true;
}
if (!modified) return false;
const ordered = {};
for (const key of ['name', 'description', 'license', 'compatibility', 'allowed-tools', 'metadata']) {
if (updated[key] !== undefined) {
ordered[key] = updated[key];
}
}
const fm = yaml.stringify(ordered).trimEnd();
const bodyPrefix = body.length && (body.startsWith('\n') || body.startsWith('\r\n')) ? '' : '\n';
const next = `---\n${fm}\n---${bodyPrefix}${body}`;
fs.writeFileSync(skillPath, next);
return true;
}
function run() {
const skillIds = listSkillIds(SKILLS_DIR);
let updatedCount = 0;
for (const skillId of skillIds) {
if (normalizeSkill(skillId)) updatedCount += 1;
}
console.log(`Normalized frontmatter for ${updatedCount} skills.`);
}
if (require.main === module) {
run();
}
module.exports = { run };