com.caoccao.javet.swc4j.ast.miscs.Swc4jAstOptCall Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of swc4j-android Show documentation
swc4j (SWC for Java) is an ultra-fast JavaScript and TypeScript compilation and bundling tool on JVM.
The 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.miscs;

import com.caoccao.javet.swc4j.ast.Swc4jAst;
import com.caoccao.javet.swc4j.ast.enums.Swc4jAstType;
import com.caoccao.javet.swc4j.ast.expr.Swc4jAstExprOrSpread;
import com.caoccao.javet.swc4j.ast.interfaces.ISwc4jAst;
import com.caoccao.javet.swc4j.ast.interfaces.ISwc4jAstExpr;
import com.caoccao.javet.swc4j.ast.interfaces.ISwc4jAstOptChainBase;
import com.caoccao.javet.swc4j.ast.ts.Swc4jAstTsTypeParamInstantiation;
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 Swc4jAstOptCall
        extends Swc4jAst
        implements ISwc4jAstOptChainBase {
    protected final List args;
    @Jni2RustField(box = true)
    protected ISwc4jAstExpr callee;
    @Jni2RustField(syntaxContext = true)
    protected int ctxt;
    @Jni2RustField(componentBox = true)
    protected Optional typeArgs;

    @Jni2RustMethod
    public Swc4jAstOptCall(
            @Jni2RustParam(syntaxContext = true) int ctxt,
            ISwc4jAstExpr callee,
            List args,
            @Jni2RustParam(optional = true) Swc4jAstTsTypeParamInstantiation typeArgs,
            Swc4jSpan span) {
        super(span);
        setCallee(callee);
        setCtxt(ctxt);
        setTypeArgs(typeArgs);
        this.args = AssertionUtils.notNull(args, "Args");
        this.args.forEach(node -> node.setParent(this));
    }

    public static Swc4jAstOptCall create(ISwc4jAstExpr callee) {
        return create(callee, SimpleList.of());
    }

    public static Swc4jAstOptCall create(ISwc4jAstExpr callee, List args) {
        return create(callee, args, null);
    }

    public static Swc4jAstOptCall create(
            ISwc4jAstExpr callee,
            List args,
            Swc4jAstTsTypeParamInstantiation typeArgs) {
        return create(0, callee, args, typeArgs);
    }

    public static Swc4jAstOptCall create(
            int ctxt,
            ISwc4jAstExpr callee,
            List args,
            Swc4jAstTsTypeParamInstantiation typeArgs) {
        return new Swc4jAstOptCall(ctxt, callee, args, typeArgs, Swc4jSpan.DUMMY);
    }

    @Jni2RustMethod
    public List getArgs() {
        return args;
    }

    @Jni2RustMethod
    public ISwc4jAstExpr getCallee() {
        return callee;
    }

    @Override
    public List getChildNodes() {
        List childNodes = SimpleList.copyOf(args);
        childNodes.add(callee);
        typeArgs.ifPresent(childNodes::add);
        return childNodes;
    }

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

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

    @Jni2RustMethod
    public Optional getTypeArgs() {
        return typeArgs;
    }

    public boolean isSpreadPresent() {
        return args.stream().anyMatch(arg -> arg.getSpread().isPresent());
    }

    @Override
    public boolean replaceNode(ISwc4jAst oldNode, ISwc4jAst newNode) {
        if (!args.isEmpty() && newNode instanceof Swc4jAstExprOrSpread) {
            final int size = args.size();
            for (int i = 0; i < size; i++) {
                if (args.get(i) == oldNode) {
                    args.set(i, (Swc4jAstExprOrSpread) newNode);
                    newNode.setParent(this);
                    return true;
                }
            }
        }
        if (callee == oldNode && newNode instanceof ISwc4jAstExpr) {
            setCallee((ISwc4jAstExpr) newNode);
            return true;
        }
        if (typeArgs.isPresent() && typeArgs.get() == oldNode && (newNode == null || newNode instanceof Swc4jAstTsTypeParamInstantiation)) {
            setTypeArgs((Swc4jAstTsTypeParamInstantiation) newNode);
            return true;
        }
        return false;
    }

    public Swc4jAstOptCall setCallee(ISwc4jAstExpr callee) {
        this.callee = AssertionUtils.notNull(callee, "Callee");
        this.callee.setParent(this);
        return this;
    }

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

    public Swc4jAstOptCall setTypeArgs(Swc4jAstTsTypeParamInstantiation typeArgs) {
        this.typeArgs = Optional.ofNullable(typeArgs);
        this.typeArgs.ifPresent(node -> node.setParent(this));
        return this;
    }

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