/* * SPDX-FileCopyrightText: syuilo and misskey-project * SPDX-License-Identifier: AGPL-3.0-only */ import { parse as vueSfcParse } from 'vue/compiler-sfc'; import type { Plugin } from 'vite'; import fs from 'node:fs'; import { glob } from 'glob'; import JSON5 from 'json5'; import { randomUUID } from 'crypto'; import MagicString from 'magic-string'; import path from 'node:path' import { hash, toBase62 } from '../vite.config'; export interface AnalysisResult { filePath: string; usage: ComponentUsageInfo[]; } export interface ComponentUsageInfo { staticProps: Record; bindProps: Record; } function outputAnalysisResultAsTS(outputPath: string, analysisResults: AnalysisResult[]): void { // (outputAnalysisResultAsTS 関数の実装は前回と同様) const varName = 'searchIndexes'; //  変数名 const jsonString = JSON5.stringify(analysisResults, { space: "\t", quote: "'" }); //  JSON.stringify で JSON 文字列を生成 //  bindProps の値を文字列置換で修正する関数 function modifyBindPropsInString(jsonString: string): string { const modifiedString = jsonString.replace( /bindProps:\s*\{([^}]*)\}/g, //  bindProps: { ... } にマッチ (g フラグで複数箇所を置換) (match, bindPropsBlock) => { //  bindPropsBlock ( { ... } 内) の各プロパティをさらに置換 const modifiedBlock = bindPropsBlock.replace( /(.*):\s*\'(.*)\'/g, //  propName: 'propValue' にマッチ (propMatch, propName, propValue) => { return `${propName}: ${propValue}`; // propValue のクォートを除去 } ).replaceAll("\\'", "'"); return `bindProps: {${modifiedBlock}}`; //  置換後の block で bindProps: { ... } を再構成 } ); return modifiedString; } const tsOutput = ` /* * SPDX-FileCopyrightText: syuilo and misskey-project * SPDX-License-Identifier: AGPL-3.0-only */ // This file was automatically generated by create-search-index. // Do not edit this file. import { i18n } from '@/i18n.js'; export const ${varName} = ${modifyBindPropsInString(jsonString)} as const; export type AnalysisResults = typeof ${varName}; export type ComponentUsageInfo = AnalysisResults[number]['usage'][number]; `; try { fs.writeFileSync(outputPath, tsOutput, 'utf-8'); } catch (error) { console.error('[create-search-index]: error: ', error); } } function extractUsageInfoFromTemplateAst( templateAst: any, code: string, ): ComponentUsageInfo[] { const usageInfoList: ComponentUsageInfo[] = []; if (!templateAst) { return usageInfoList; } function traverse(node: any) { if (node.type === 1 && node.tag === 'SearchMarker') { // 元々の props を staticProps に全て展開する const staticProps: Record = {}; if (node.props && Array.isArray(node.props)) { node.props.forEach((prop: any) => { if (prop.type === 6 && prop.name) { staticProps[prop.name] = prop.value?.content || ''; } }); } // markerId は __markerId または既存の props から取得 const markerId = node.__markerId || staticProps['markerId']; if (markerId) { staticProps['markerId'] = markerId; } const bindProps: Record = {}; // 元々の props から bindProps を抽出 if (node.props && Array.isArray(node.props)) { node.props.forEach((prop: any) => { if (prop.type === 7 && prop.name === 'bind' && prop.arg.content) { bindProps[prop.arg.content] = prop.exp?.content || ''; } }); } // __children がある場合、bindProps に children を追加 if (node.__children) { bindProps['children'] = node.__children; } else if (node.props && Array.isArray(node.props)) { const childrenProp = node.props.find( (prop: any) => prop.type === 7 && prop.name === 'bind' && prop.arg?.content === 'children' ); if (childrenProp && childrenProp.exp) { try { bindProps['children'] = JSON5.parse( code.slice(childrenProp.exp.loc.start.offset, childrenProp.exp.loc.end.offset).replace(/'/g, '"') ); } catch (e) { console.error('Error parsing :children attribute', e); } } } usageInfoList.push({ staticProps, bindProps, }); } if (node.children && Array.isArray(node.children)) { node.children.forEach((child: any) => traverse(child)); } } traverse(templateAst); return usageInfoList; } export async function analyzeVueProps(options: { targetFilePaths: string[], exportFilePath: string, transformedCodeCache: Record }): Promise { const analysisResults: AnalysisResult[] = []; //  対象ファイルパスを glob で展開 const filePaths = options.targetFilePaths.reduce((acc, filePathPattern) => { const matchedFiles = glob.sync(filePathPattern); return [...acc, ...matchedFiles]; }, []); for (const filePath of filePaths) { const code = options.transformedCodeCache[path.resolve(filePath)]; // options 経由でキャッシュ参照 if (!code) { // キャッシュミスの場合 console.error(`[create-search-index] Error: No cached code found for: ${filePath}.`); // エラーログ continue; } const { descriptor, errors } = vueSfcParse(code, { filename: filePath, }); if (errors.length) { console.error(`[create-search-index] Compile Error: ${filePath}`, errors); continue; // エラーが発生したファイルはスキップ } const usageInfo = extractUsageInfoFromTemplateAst(descriptor.template?.ast, code); if (!usageInfo) continue; if (usageInfo.length > 0) { analysisResults.push({ filePath: filePath, usage: usageInfo, }); } } outputAnalysisResultAsTS(options.exportFilePath, analysisResults); // outputAnalysisResultAsTS を呼び出す } interface MarkerRelation { parentId?: string; markerId: string; node: any; } async function processVueFile( code: string, id: string, options: { targetFilePaths: string[], exportFilePath: string }, transformedCodeCache: Record ) { // すでにキャッシュに存在する場合は、そのまま返す if (transformedCodeCache[id] && transformedCodeCache[id].includes('markerId=')) { console.log(`[create-search-index] Using cached version for ${id}`); return { code: transformedCodeCache[id], map: null }; } const s = new MagicString(code); // magic-string のインスタンスを作成 const parsed = vueSfcParse(code, { filename: id }); if (!parsed.descriptor.template) { return; } const ast = parsed.descriptor.template.ast; // テンプレート AST を取得 const markerRelations: MarkerRelation[] = []; // MarkerRelation 配列を初期化 if (ast) { function traverse(node: any, currentParent?: any) { if (node.type === 1 && node.tag === 'SearchMarker') { // 行番号はコード先頭からの改行数で取得 const lineNumber = code.slice(0, node.loc.start.offset).split('\n').length; // ファイルパスと行番号からハッシュ値を生成 const generatedMarkerId = toBase62(hash(`${id}:${lineNumber}`)); const props = node.props || []; const hasMarkerIdProp = props.some((prop: any) => prop.type === 6 && prop.name === 'markerId'); const nodeMarkerId = hasMarkerIdProp ? props.find((prop: any) => prop.type === 6 && prop.name === 'markerId')?.value?.content as string : generatedMarkerId; node.__markerId = nodeMarkerId; // 子マーカーの場合、親ノードに __children を設定しておく if (currentParent && currentParent.type === 1 && currentParent.tag === 'SearchMarker') { currentParent.__children = currentParent.__children || []; currentParent.__children.push(nodeMarkerId); } const parentMarkerId = currentParent && currentParent.__markerId; markerRelations.push({ parentId: parentMarkerId, markerId: nodeMarkerId, node: node, }); if (!hasMarkerIdProp) { const nodeStart = node.loc.start.offset; let endOfStartTag; if (node.children && node.children.length > 0) { // 子要素がある場合、最初の子要素の開始位置を基準にする endOfStartTag = code.lastIndexOf('>', node.children[0].loc.start.offset); } else if (node.loc.end.offset > nodeStart) { // 子要素がない場合、自身の終了位置から逆算 const nodeSource = code.substring(nodeStart, node.loc.end.offset); // 自己終了タグか通常の終了タグかを判断 if (nodeSource.includes('/>')) { endOfStartTag = code.indexOf('/>', nodeStart) - 1; } else { endOfStartTag = code.indexOf('>', nodeStart); } } if (endOfStartTag !== undefined && endOfStartTag !== -1) { // markerId が既に存在しないことを確認 const tagText = code.substring(nodeStart, endOfStartTag + 1); const markerIdRegex = /\s+markerId\s*=\s*["'][^"']*["']/; if (!markerIdRegex.test(tagText)) { s.appendRight(endOfStartTag, ` markerId="${generatedMarkerId}"`); console.log(`[create-search-index] Adding markerId="${generatedMarkerId}" to ${id}:${lineNumber}`); } else { console.log(`[create-search-index] markerId already exists in ${id}:${lineNumber}`); } } } } const newParent = node.type === 1 && node.tag === 'SearchMarker' ? node : currentParent; if (node.children && Array.isArray(node.children)) { node.children.forEach(child => traverse(child, newParent)); } } traverse(ast); // AST を traverse (1段階目: ID 生成と親子関係記録) // 2段階目: :children 属性の追加 // 最初に親マーカーごとに子マーカーIDを集約する処理を追加 const parentChildrenMap = new Map(); // 1. まず親ごとのすべての子マーカーIDを収集 markerRelations.forEach(relation => { if (relation.parentId) { if (!parentChildrenMap.has(relation.parentId)) { parentChildrenMap.set(relation.parentId, []); } parentChildrenMap.get(relation.parentId)?.push(relation.markerId); } }); // 2. 親ごとにまとめて :children 属性を処理 for (const [parentId, childIds] of parentChildrenMap.entries()) { const parentRelation = markerRelations.find(r => r.markerId === parentId); if (!parentRelation || !parentRelation.node) continue; const parentNode = parentRelation.node; const childrenProp = parentNode.props?.find((prop: any) => prop.type === 7 && prop.name === 'bind' && prop.arg?.content === 'children'); // 親ノードの開始位置を特定 const parentNodeStart = parentNode.loc.start.offset; const endOfParentStartTag = parentNode.children && parentNode.children.length > 0 ? code.lastIndexOf('>', parentNode.children[0].loc.start.offset) : code.indexOf('>', parentNodeStart); if (endOfParentStartTag === -1) continue; // 親タグのテキストを取得 const parentTagText = code.substring(parentNodeStart, endOfParentStartTag + 1); if (childrenProp) { // AST で :children 属性が検出された場合、それを更新 try { const childrenStart = code.indexOf('[', childrenProp.exp.loc.start.offset); const childrenEnd = code.indexOf(']', childrenProp.exp.loc.start.offset); if (childrenStart !== -1 && childrenEnd !== -1) { const childrenArrayStr = code.slice(childrenStart, childrenEnd + 1); let childrenArray = JSON5.parse(childrenArrayStr.replace(/'/g, '"')); // 新しいIDを追加(重複は除外) const newIds = childIds.filter(id => !childrenArray.includes(id)); if (newIds.length > 0) { childrenArray = [...childrenArray, ...newIds]; const updatedChildrenArrayStr = JSON.stringify(childrenArray).replace(/"/g, "'"); s.overwrite(childrenStart, childrenEnd + 1, updatedChildrenArrayStr); console.log(`[create-search-index] Added ${newIds.length} child markerIds to existing :children in ${id}`); } } } catch (e) { console.error('[create-search-index] Error updating :children attribute:', e); } } else { // AST では検出されなかった場合、タグテキストを調べる const childrenRegex = /:children\s*=\s*["']\[(.*?)\]["']/; const childrenMatch = parentTagText.match(childrenRegex); if (childrenMatch) { // テキストから :children 属性値を解析して更新 try { const childrenContent = childrenMatch[1]; const childrenArrayStr = `[${childrenContent}]`; const childrenArray = JSON5.parse(childrenArrayStr.replace(/'/g, '"')); // 新しいIDを追加(重複は除外) const newIds = childIds.filter(id => !childrenArray.includes(id)); if (newIds.length > 0) { childrenArray.push(...newIds); // :children="[...]" の位置を特定して上書き const attrStart = parentTagText.indexOf(':children='); if (attrStart > -1) { const attrValueStart = parentTagText.indexOf('[', attrStart); const attrValueEnd = parentTagText.indexOf(']', attrValueStart) + 1; if (attrValueStart > -1 && attrValueEnd > -1) { const absoluteStart = parentNodeStart + attrValueStart; const absoluteEnd = parentNodeStart + attrValueEnd; const updatedArrayStr = JSON.stringify(childrenArray).replace(/"/g, "'"); s.overwrite(absoluteStart, absoluteEnd, updatedArrayStr); console.log(`[create-search-index] Updated existing :children in tag text for ${id}`); } } } } catch (e) { console.error('[create-search-index] Error updating :children in tag text:', e); } } else { // :children 属性がまだない場合、新規作成 s.appendRight(endOfParentStartTag, ` :children="${JSON.stringify(childIds).replace(/"/g, "'")}"`); console.log(`[create-search-index] Created new :children attribute with ${childIds.length} markerIds in ${id}`); } } } } const transformedCode = s.toString(); // 変換後のコードを取得 transformedCodeCache[id] = transformedCode; // 変換後のコードをキャッシュに保存 return { code: transformedCode, // 変更後のコードを返す map: s.generateMap({ source: id, includeContent: true }), // ソースマップも生成 (sourceMap: true が必要) }; } // Rollup プラグインとして export export default function pluginCreateSearchIndex(options: { targetFilePaths: string[], exportFilePath: string }): Plugin { let transformedCodeCache: Record = {}; // キャッシュオブジェクトをプラグインスコープで定義 const isDevServer = process.env.NODE_ENV === 'development'; // 開発サーバーかどうか return { name: 'createSearchIndex', enforce: 'pre', async buildStart() { if (!isDevServer) { return; } const filePaths = options.targetFilePaths.reduce((acc, filePathPattern) => { const matchedFiles = glob.sync(filePathPattern); return [...acc, ...matchedFiles]; }, []); for (const filePath of filePaths) { const id = path.resolve(filePath); // 絶対パスに変換 const code = fs.readFileSync(filePath, 'utf-8'); // ファイル内容を読み込む await processVueFile(code, id, options, transformedCodeCache); // processVueFile 関数を呼び出す } await analyzeVueProps({ ...options, transformedCodeCache }); // 開発サーバー起動時にも analyzeVueProps を実行 }, async transform(code, id) { if (!id.endsWith('.vue')) { return; } // targetFilePaths にマッチするファイルのみ処理を行う // glob パターンでマッチング let isMatch = false; // isMatch の初期値を false に設定 for (const pattern of options.targetFilePaths) { // パターンごとにマッチング確認 const globbedFiles = glob.sync(pattern); for (const globbedFile of globbedFiles) { const normalizedGlobbedFile = path.resolve(globbedFile); // glob 結果を絶対パスに const normalizedId = path.resolve(id); // id を絶対パスに if (normalizedGlobbedFile === normalizedId) { // 絶対パス同士で比較 isMatch = true; break; // マッチしたらループを抜ける } } if (isMatch) break; // いずれかのパターンでマッチしたら、outer loop も抜ける } if (!isMatch) { return; } const transformed = await processVueFile(code, id, options, transformedCodeCache); if (isDevServer) { await analyzeVueProps({ ...options, transformedCodeCache }); // analyzeVueProps を呼び出す } return transformed; }, async writeBundle() { await analyzeVueProps({ ...options, transformedCodeCache }); // ビルド時にも analyzeVueProps を実行 }, }; }