@voerkai18n/utils/src/extract/ast/ast.ts

utils for voerkai18n

/**
 * 
 * 解析源码中的t("xxxx")的内容
 *  
 */

import { parse, Lang, SgNode, Range } from '@ast-grep/napi' 
import { MessageNode, ExtractorOptions  } from './types'
import { getFileNamespace } from '../../getFileNamespace';
import { parseTranslateMessages } from './utils/parseMessage'; 
import { trimChars } from '../../trimChars';


function getMessageRange(range:Range,options:ExtractorOptions){
    const { sectionFlag } = options
    return {
        start: `${sectionFlag}${range.start.line}:${range.start.column}`,
        end: `${sectionFlag}${range.end.line}:${range.end.column}`
    }
}


export function extractSectionMessages(node:SgNode,options:ExtractorOptions){
    const { namespaces,file } = options
    const namespace = file ? getFileNamespace(file, namespaces) : 'default'
    const msgNode = node.getMatch("MESSAGE")!   
    const results:any = [] 
    if(msgNode){  
        const message = trimChars(msgNode.text())
        results.push({
            message,
            rang     : getMessageRange(msgNode.range(),options),
            vars     : node.getMatch("VARS")?.text(),
            options  : node.getMatch("OPTIONS")?.text(),
            namespace,
            file
        })
    }else{                
        const text = node.text()
        const nodeCtx = {
            rang     : getMessageRange(node.range(),options),
            vars     : node.getMatch("VARS")?.text(),
            options  : node.getMatch("OPTIONS")?.text(),
            namespace,
            file
        }
        if(text){
            const messages = parseTranslateMessages(text)
            messages.forEach((message:any)=>{
                results.push({
                    message,
                    ...nodeCtx
                })
            })
        }
    }
    return results
}
 




/**
 * 使用@ast-grep/napi解析code中的t("xxxx")的内容 
 * 
 * @param code 
 * @returns
 * [
 *      {text:"xxxx",rang,vars:[],options:{}},
 * ]
 */
export default function extractor(code:string,options:ExtractorOptions):MessageNode[]{
    const { language, ast: astConfig } = Object.assign({
        language: Lang.TypeScript,
    },options)
    if(!astConfig) return []
    const results:any = [] 
    const sectionAst = parse(language, code)        
    const nodes = sectionAst.root().findAll(astConfig)
    nodes.forEach(node=>{
        results.push(...extractSectionMessages(node,options))
    })  
    return results
}