kittenipc/kitcom/internal/ts/tsparser.go

package ts

import (
	"fmt"
	"io"
	"os"
	"strings"

	"efprojects.com/kitten-ipc/kitcom/internal/api"
	"efprojects.com/kitten-ipc/kitcom/internal/common"
	"efprojects.com/kitten-ipc/kitcom/internal/tsgo/ast"
	"efprojects.com/kitten-ipc/kitcom/internal/tsgo/core"
	"efprojects.com/kitten-ipc/kitcom/internal/tsgo/parser"
	"efprojects.com/kitten-ipc/kitcom/internal/tsgo/tspath"
	"efprojects.com/kitten-ipc/types"
)

type TypescriptApiParser struct {
	*common.Parser
}

func (p *TypescriptApiParser) Parse() (*api.Api, error) {
	return p.MapFiles(p.parseFile)
}

func (p *TypescriptApiParser) parseFile(sourceFilePath string) ([]api.Endpoint, error) {
	var endpoints []api.Endpoint

	f, err := os.Open(sourceFilePath)
	if err != nil {
		return nil, fmt.Errorf("failed to open file: %w", err)
	}
	defer f.Close()

	fileContents, err := io.ReadAll(f)
	if err != nil {
		return nil, fmt.Errorf("failed to read file: %w", err)
	}

	sourceFile := parser.ParseSourceFile(ast.SourceFileParseOptions{
		FileName:                       sourceFilePath,
		Path:                           tspath.Path(sourceFilePath),
		CompilerOptions:                core.SourceFileAffectingCompilerOptions{},
		ExternalModuleIndicatorOptions: ast.ExternalModuleIndicatorOptions{},
		JSDocParsingMode:               ast.JSDocParsingModeParseAll,
	}, string(fileContents), core.ScriptKindTS)
	_ = sourceFile

	sourceFile.ForEachChild(func(node *ast.Node) bool {
		if node.Kind != ast.KindClassDeclaration {
			return false
		}
		cls := node.AsClassDeclaration()

		if !p.isApiNode(cls) {
			return false
		}

		var endpoint api.Endpoint
		endpoint.Name = cls.Name().Text()

		for _, member := range cls.MemberList().Nodes {
			if member.Kind != ast.KindMethodDeclaration {
				continue
			}

			method := member.AsMethodDeclaration()
			if method.ModifierFlags()&ast.ModifierFlagsPrivate > 0 || method.ModifierFlags()&ast.ModifierFlagsProtected > 0 {
				continue
			}

			var apiMethod api.Method
			apiMethod.Name = method.Name().Text()
			for _, parNode := range method.ParameterList().Nodes {
				par := parNode.AsParameterDeclaration()
				var apiPar api.Val
				apiPar.Name = par.Name().Text()
				t, typeErr := p.fieldToVal(par.Type)
				if typeErr != nil {
					err = fmt.Errorf("failed to parse parameter %s: %w", apiPar.Name, typeErr)
					return false
				}
				apiPar.Type = t
				apiMethod.Params = append(apiMethod.Params, apiPar)
			}

			if method.Type != nil {
				var apiRet api.Val
				t, typeErr := p.fieldToVal(method.Type)
				if typeErr != nil {
					err = fmt.Errorf("failed to parse return type: %w", typeErr)
					return false
				}
				apiRet.Type = t
				apiMethod.Ret = []api.Val{apiRet}
			}
			endpoint.Methods = append(endpoint.Methods, apiMethod)
		}

		endpoints = append(endpoints, endpoint)

		return false
	})

	if err != nil {
		return nil, err
	}

	return endpoints, nil
}

func (p *TypescriptApiParser) fieldToVal(typ *ast.TypeNode) (types.ValType, error) {
	switch typ.Kind {
	case ast.KindNumberKeyword:
		return types.TInt, nil
	case ast.KindStringKeyword:
		return types.TString, nil
	case ast.KindBooleanKeyword:
		return types.TBool, nil
	case ast.KindTypeReference:
		refNode := typ.AsTypeReferenceNode()
		ident := refNode.TypeName.AsIdentifier()
		switch ident.Text {
		case "Buffer":
			return types.TBlob, nil
		default:
			return 0, fmt.Errorf("reference type %s is not supported yet", ident.Text)
		}
	default:
		return 0, fmt.Errorf("type %s is not supported yet", typ.Kind)
	}
}

const TagName = "kittenipc"
const TagComment = "api"

func (p *TypescriptApiParser) isApiNode(cls *ast.ClassDeclaration) bool {
	jsDocNodes := cls.JSDoc(nil)
	if len(jsDocNodes) == 0 {
		return false
	}
	for _, jsDocNode := range jsDocNodes {
		jsDoc := jsDocNode.AsJSDoc()
		if jsDoc.Tags == nil {
			continue
		}
		for _, tag := range jsDoc.Tags.Nodes {
			if tag.TagName().Text() == TagName {
				for _, com := range tag.Comments() {
					if strings.TrimSpace(com.Text()) == TagComment {
						return true
					}
				}
			}
		}
	}
	return false
}