All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.caoccao.javet.swc4j.ast.clazz.Swc4jAstFunction Maven / Gradle / Ivy

Go to download

swc4j (SWC for Java) is an ultra-fast JavaScript and TypeScript compilation and bundling tool on JVM.

There is a newer version: 1.3.0
Show newest version
/*
 * Copyright (c) 2024. caoccao.com Sam Cao
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.caoccao.javet.swc4j.ast.clazz;

import com.caoccao.javet.swc4j.ast.Swc4jAst;
import com.caoccao.javet.swc4j.ast.enums.Swc4jAstType;
import com.caoccao.javet.swc4j.ast.interfaces.ISwc4jAst;
import com.caoccao.javet.swc4j.ast.stmt.Swc4jAstBlockStmt;
import com.caoccao.javet.swc4j.ast.ts.Swc4jAstTsTypeAnn;
import com.caoccao.javet.swc4j.ast.ts.Swc4jAstTsTypeParamDecl;
import com.caoccao.javet.swc4j.ast.visitors.ISwc4jAstVisitor;
import com.caoccao.javet.swc4j.ast.visitors.Swc4jAstVisitorResponse;
import com.caoccao.javet.swc4j.jni2rust.*;
import com.caoccao.javet.swc4j.span.Swc4jSpan;
import com.caoccao.javet.swc4j.utils.AssertionUtils;
import com.caoccao.javet.swc4j.utils.SimpleList;

import java.util.List;
import java.util.Optional;

@Jni2RustClass(filePath = Jni2RustFilePath.AstUtils)
public class Swc4jAstFunction
        extends Swc4jAst {
    protected final List decorators;
    protected final List params;
    @Jni2RustField(name = "is_async")
    protected boolean _async;
    protected Optional body;
    @Jni2RustField(syntaxContext = true)
    protected int ctxt;
    @Jni2RustField(name = "is_generator")
    protected boolean generator;
    @Jni2RustField(componentBox = true)
    protected Optional returnType;
    @Jni2RustField(componentBox = true)
    protected Optional typeParams;

    @Jni2RustMethod
    public Swc4jAstFunction(
            @Jni2RustParam(syntaxContext = true) int ctxt,
            List params,
            List decorators,
            @Jni2RustParam(optional = true) Swc4jAstBlockStmt body,
            @Jni2RustParam(name = "is_generator") boolean generator,
            @Jni2RustParam(name = "is_async") boolean _async,
            @Jni2RustParam(optional = true) Swc4jAstTsTypeParamDecl typeParams,
            @Jni2RustParam(optional = true) Swc4jAstTsTypeAnn returnType,
            Swc4jSpan span) {
        super(span);
        setAsync(_async);
        setBody(body);
        setCtxt(ctxt);
        setGenerator(generator);
        setReturnType(returnType);
        setTypeParams(typeParams);
        this.decorators = AssertionUtils.notNull(decorators, "Decorators");
        this.decorators.forEach(node -> node.setParent(this));
        this.params = AssertionUtils.notNull(params, "Params");
        this.params.forEach(node -> node.setParent(this));
    }

    public static Swc4jAstFunction create() {
        return create(SimpleList.of());
    }

    public static Swc4jAstFunction create(List params) {
        return create(params, SimpleList.of());
    }

    public static Swc4jAstFunction create(List params, List decorators) {
        return create(params, decorators, null);
    }

    public static Swc4jAstFunction create(
            List params,
            List decorators,
            Swc4jAstBlockStmt body) {
        return create(params, decorators, body, false);
    }

    public static Swc4jAstFunction create(
            List params,
            List decorators,
            Swc4jAstBlockStmt body,
            boolean generator) {
        return create(params, decorators, body, generator, false, null, null);
    }

    public static Swc4jAstFunction create(
            List params,
            List decorators,
            Swc4jAstBlockStmt body,
            boolean generator,
            boolean _async,
            Swc4jAstTsTypeParamDecl typeParams) {
        return create(params, decorators, body, generator, _async, typeParams, null);
    }

    public static Swc4jAstFunction create(
            List params,
            List decorators,
            Swc4jAstBlockStmt body,
            boolean generator,
            boolean _async,
            Swc4jAstTsTypeAnn returnType) {
        return create(params, decorators, body, generator, _async, null, returnType);
    }

    public static Swc4jAstFunction create(
            List params,
            List decorators,
            Swc4jAstBlockStmt body,
            boolean generator,
            boolean _async,
            Swc4jAstTsTypeParamDecl typeParams,
            Swc4jAstTsTypeAnn returnType) {
        return create(0, params, decorators, body, generator, _async, typeParams, returnType);
    }

    public static Swc4jAstFunction create(
            int ctxt,
            List params,
            List decorators,
            Swc4jAstBlockStmt body,
            boolean generator,
            boolean _async,
            Swc4jAstTsTypeParamDecl typeParams,
            Swc4jAstTsTypeAnn returnType) {
        return new Swc4jAstFunction(
                ctxt, params, decorators, body, generator,
                _async, typeParams, returnType, Swc4jSpan.DUMMY);
    }

    @Jni2RustMethod
    public Optional getBody() {
        return body;
    }

    @Override
    public List getChildNodes() {
        List childNodes = SimpleList.copyOf(decorators);
        childNodes.addAll(params);
        body.ifPresent(childNodes::add);
        typeParams.ifPresent(childNodes::add);
        returnType.ifPresent(childNodes::add);
        return childNodes;
    }

    @Jni2RustMethod
    public int getCtxt() {
        return ctxt;
    }

    @Jni2RustMethod
    public List getDecorators() {
        return decorators;
    }

    @Jni2RustMethod
    public List getParams() {
        return params;
    }

    @Jni2RustMethod
    public Optional getReturnType() {
        return returnType;
    }

    @Override
    public Swc4jAstType getType() {
        return Swc4jAstType.Function;
    }

    @Jni2RustMethod
    public Optional getTypeParams() {
        return typeParams;
    }

    @Jni2RustMethod
    public boolean isAsync() {
        return _async;
    }

    @Jni2RustMethod
    public boolean isGenerator() {
        return generator;
    }

    @Override
    public boolean replaceNode(ISwc4jAst oldNode, ISwc4jAst newNode) {
        if (body.isPresent() && body.get() == oldNode && (newNode == null || newNode instanceof Swc4jAstBlockStmt)) {
            setBody((Swc4jAstBlockStmt) newNode);
            return true;
        }
        if (!decorators.isEmpty() && newNode instanceof Swc4jAstDecorator) {
            final int size = decorators.size();
            for (int i = 0; i < size; i++) {
                if (decorators.get(i) == oldNode) {
                    decorators.set(i, (Swc4jAstDecorator) newNode);
                    newNode.setParent(this);
                    return true;
                }
            }
        }
        if (!params.isEmpty() && newNode instanceof Swc4jAstParam) {
            final int size = params.size();
            for (int i = 0; i < size; i++) {
                if (params.get(i) == oldNode) {
                    params.set(i, (Swc4jAstParam) newNode);
                    newNode.setParent(this);
                    return true;
                }
            }
        }
        if (returnType.isPresent() && returnType.get() == oldNode && (newNode == null || newNode instanceof Swc4jAstTsTypeAnn)) {
            setReturnType((Swc4jAstTsTypeAnn) newNode);
            return true;
        }
        if (typeParams.isPresent() && typeParams.get() == oldNode && (newNode == null || newNode instanceof Swc4jAstTsTypeParamDecl)) {
            setTypeParams((Swc4jAstTsTypeParamDecl) newNode);
            return true;
        }
        return false;
    }

    public Swc4jAstFunction setAsync(boolean _async) {
        this._async = _async;
        return this;
    }

    public Swc4jAstFunction setBody(Swc4jAstBlockStmt body) {
        this.body = Optional.ofNullable(body);
        this.body.ifPresent(node -> node.setParent(this));
        return this;
    }

    public Swc4jAstFunction setCtxt(int ctxt) {
        this.ctxt = ctxt;
        return this;
    }

    public Swc4jAstFunction setGenerator(boolean generator) {
        this.generator = generator;
        return this;
    }

    public Swc4jAstFunction setReturnType(Swc4jAstTsTypeAnn returnType) {
        this.returnType = Optional.ofNullable(returnType);
        this.returnType.ifPresent(node -> node.setParent(this));
        return this;
    }

    public Swc4jAstFunction setTypeParams(Swc4jAstTsTypeParamDecl typeParams) {
        this.typeParams = Optional.ofNullable(typeParams);
        this.typeParams.ifPresent(node -> node.setParent(this));
        return this;
    }

    @Override
    public Swc4jAstVisitorResponse visit(ISwc4jAstVisitor visitor) {
        switch (visitor.visitFunction(this)) {
            case Error:
                return Swc4jAstVisitorResponse.Error;
            case OkAndBreak:
                return Swc4jAstVisitorResponse.OkAndContinue;
            default:
                return super.visit(visitor);
        }
    }
}




© 2015 - 2025 Weber Informatics LLC | Privacy Policy