package io.gitlab.jfronny.muscript.ast.dynamic.unpack;

import io.gitlab.jfronny.muscript.compiler.ExprWriter;
import io.gitlab.jfronny.muscript.data.Scope;
import io.gitlab.jfronny.muscript.data.dynamic.DynamicTypeConversionException;
import io.gitlab.jfronny.muscript.ast.DynamicExpr;
import io.gitlab.jfronny.muscript.ast.StringExpr;
import io.gitlab.jfronny.muscript.annotations.CanThrow;
import io.gitlab.jfronny.muscript.annotations.UncheckedDynamic;
import io.gitlab.jfronny.muscript.ast.dynamic.DynamicCoerce;

import java.io.IOException;

@CanThrow
@UncheckedDynamic
public class StringUnpack extends StringExpr {
    public final DynamicExpr inner;

    public StringUnpack(int chStart, int chEnd, DynamicExpr inner) {
        super(inner.order, chStart, chEnd);
        this.inner = inner;
    }

    @Override
    public String get(Scope dataRoot) {
        try {
            return inner.get(dataRoot).asString().getValue();
        } catch (DynamicTypeConversionException e) {
            throw e.locational(chStart, chEnd);
        }
    }

    @Override
    public StringExpr optimize() {
        DynamicExpr inner = this.inner.optimize();
        if (inner instanceof DynamicCoerce coerce) return coerce.inner.asStringExpr();
        return new StringUnpack(chStart, chEnd, inner);
    }

    @Override
    public void decompile(ExprWriter writer) throws IOException {
        inner.decompile(writer);
    }

    @Override
    public boolean equals(Object obj) {
        return obj instanceof StringUnpack unpack && inner.equals(unpack.inner);
    }
}