DFScript/src/main/java/io/github/techstreet/dfscript/script/event/ScriptFunction.java at 1935949bcd6a399d08223739318880e472234bf4 · XTerPL/DFScript · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
package io.github.techstreet.dfscript.script.event;

import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import com.google.gson.JsonSerializationContext;
import com.google.gson.JsonSerializer;
import io.github.techstreet.dfscript.screen.widget.CItem;
import io.github.techstreet.dfscript.screen.widget.CScrollPanel;
import io.github.techstreet.dfscript.screen.widget.CText;
import io.github.techstreet.dfscript.script.Script;
import io.github.techstreet.dfscript.script.action.ScriptActionArgument;
import io.github.techstreet.dfscript.script.action.ScriptActionArgumentList;
import net.minecraft.item.Item;
import net.minecraft.item.ItemStack;
import net.minecraft.item.Items;
import net.minecraft.nbt.NbtList;
import net.minecraft.nbt.NbtString;
import net.minecraft.registry.Registries;
import net.minecraft.text.Style;
import net.minecraft.text.Text;
import net.minecraft.util.Formatting;
import net.minecraft.util.Identifier;

import java.lang.reflect.Type;

public class ScriptFunction extends ScriptHeader {

    public static final ItemStack functionIcon;

    static {
        functionIcon = new ItemStack(Items.LAPIS_BLOCK).setCustomName(Text.literal("Function").setStyle(Style.EMPTY.withColor(Formatting.WHITE).withItalic(false)));
    }
    private String name;

    private String description;
    private Item icon;

    private ScriptActionArgumentList argList;

    public ScriptFunction(String name, Item icon, ScriptActionArgumentList argList, String description) {
        this.name = name;
        this.icon = icon;
        this.argList = argList;
        this.description = description;
    }

    public String getName() {
        return name;
    }

    public ItemStack getIcon() {
        ItemStack icon = new ItemStack(this.icon);

        icon.setCustomName(Text.literal(getName()).setStyle(Style.EMPTY.withColor(Formatting.WHITE).withItalic(false)));

        NbtList lore = new NbtList();

        if(description != "") {
            for (String descriptionLine: description.split("\n")) {
                lore.add(NbtString.of(Text.Serialization.toJsonString(Text.literal(descriptionLine)
                        .fillStyle(Style.EMPTY
                                .withColor(Formatting.GRAY)
                                .withItalic(false)))));
            }

            if(argList.size() > 0)
                lore.add(NbtString.of(Text.Serialization.toJsonString(Text.literal(""))));
        }

        for (ScriptActionArgument arg : argList) {
            for (Text txt : arg.text()) {
                lore.add(NbtString.of(Text.Serialization.toJsonString(txt)));
            }
        }

        icon.getSubNbt("display")
                .put("Lore", lore);

        return icon;
    }

    public Item getRawIcon() {
        return icon;
    }

    public void setName(String text) {
        this.name = text;
    }

    public void setIcon(Item newIcon) {
        icon = newIcon;
    }

    public void replaceArgument(String oldArg, String newArg) {
        forEach((snippet) -> {
            snippet.replaceFunctionArgument(oldArg, newArg);
        });
    }

    public void removeArgument(String arg) {
        forEach((snippet) -> {
            snippet.removeFunctionArgument(arg);
        });
    }

    public void setDescription(String description) {
        this.description = description;
    }

    public String getDescription() {
        return description;
    }

    public static class Serializer implements JsonSerializer<ScriptFunction> {

        @Override
        public JsonElement serialize(ScriptFunction src, Type typeOfSrc, JsonSerializationContext context) {
            JsonObject obj = new JsonObject();
            obj.addProperty("type", "function");
            obj.addProperty("name", src.getName());
            obj.addProperty("description", src.getDescription());
            obj.addProperty("icon", Registries.ITEM.getId(src.getRawIcon()).toString());
            obj.add("args", context.serialize(src.argList()));
            obj.add("snippet", context.serialize(src.container().getSnippet(0)));
            return obj;
        }
    }

    public int create(CScrollPanel panel, int y, int index, Script script) {
        panel.add(new CItem(5, y, functionIcon));
        panel.add(new CText(15, y + 2, Text.literal(getName())));

        return super.create(panel, y, index, script);
    }

    public ScriptActionArgumentList argList() {
        return argList;
    }
}