blorkx

package com.sforzando.jscratch2;

public class BlockReader {
    public void read (JSONObject input, ScratchProject project, ScratchTarget target) {
        String[] ids = new String[input.length()];
        Object[] jblocks = new Object[input.length()];
        Iterator<String> keys = input.keySet().iterator();
        {
            int keysI = 0;
            while (keys.hasNext()) {
                ids[keysI] = keys.next();
                jblocks[keysI] = input.get(ids[keysI]);
                keysI++;
            }
        }
        ScratchBlock[] blocks = new ScratchBlock[jblocks.length];
        for (int i = 0; i < ids.length; i++)
            blocks[i] = parseBlock(jblocks[i], project, target);
        target.addBlocks(blocks);
    }

    public static ScratchBlock parseBlock (Object o, ScratchProject project, ScratchTarget target) {
            if (o instanceof JSONArray) {
                JSONArray array = (JSONArray) o;
                int id =  array.getInt(0);
                switch (id) {
                    case 4:
                        double value = array.getDouble(1);
                        return new NumberBlock(value, false);
                    case 5:
                        double value = array.getDouble(1);
                        return new NumberBlock(value, true);
                    case 6:
                        int value = array.getInt(1);
                        return new NumberBlock(value, true);
                    case 7:
                        int value = array.getInt(1);
                        return new NumberBlock(value, false);
                    case 8:
                        int value = array.getInt(1);
                        return new AngleBlock(value);
                    case 9:
                        String value = array.getString(1);
                        Color color;
                        if (value.length() == 7) color = new Color(Integer.parseInt(value.substring(1), 16));
                        else if (value.length() == 9) {
                            int alpha = Integer.parseInt(value.substring(7), 16);
                            int r = Integer.parseInt(value.substring(1, 2), 16);
                            int g = Integer.parseInt(value.substring(3, 4), 16);
                            int b = Integer.parseInt(value.substring(5, 6), 16);
                            color = new Color(r, g, b, alpha);
                        } else break;
                        return new ColorBlock(color);
                    case 10:
                        String value = array.getString(1);
                        return new StringBlock(value);
                    case 11:
                        String name = array.getString(1);
                        String id = array.getString(2);
                        return new BroadcastBlock(name, id);
                    case 12:
                        String name = array.getString(1);
                        String id = array.getString(2);
                        return new VariableBlock(name, id);
                    case 13:
                        String name = array.getString(1);
                        String id = array.getString(2);
                        int x = array.getInt(3);
                        int y = array.getInt(4);
                        ScratchBlock block = new VariableBlock(name, id);
                        block.setPosition(x, y);
                        return block;
                }
            }
            JSONObject json = (JSONObject) o;
            ScratchBlockType<? extends ScratchBlock> type = ScratchBlockTypes.get(json.getString("opcode"));
            Supplier<String> next = json.getString("next") == null || type.isFinal() || checkStopFinality(json) ? () -> null : () -> blocks[indexOf(json.getString("next"), ids)];
            Supplier<String> parent = json.getString("parent") == null || json.getBoolean("topLevel") ? () -> null : () -> blocks[indexOf(json.getString("parent"), ids)];
            ScratchBlockInput[] inputs = new ScratchBlockInput[json.getJSONObject("inputs").length()];
            Iterator<String> inputKeys = json.getJSONObject("inputs").keySet();
            int j = 0;
            while (inputKeys.hasNext()) {
                String next = inputKeys.next();
                inputs[j] = type.parseInput(next, json.getJSONObject("inputs").get(next));
                j++;
            }
            ScratchBlockInput[] fields = new ScratchBlockInput[json.getJSONObject("fields").length()];
            Iterator<String> fieldKeys = json.getJSONObject("fields").keySet();
            int j = 0;
            while (fieldKeys.hasNext()) {
                String next = fieldKeys.next();
                fields[j] = type.parseField(next, json.getJSONObject("inputs").get(next));
                j++;
            }
        boolean shadow = json.getBoolean("shadow");
        boolean topLevel = json.getBoolean("topLevel");
        if (type.hasMutations())
            return type.create(next, parent, inputs, fields, shadow, topLevel, type.createMutations(json.getJSONObject("mutation")));
        else
            return type.create(next, parent, inputs, fields, shadow, topLevel);
    }

    public static boolean checkStopFinality (JSONObject json) {
        if (!json.getString("opcode").equals("control_stop")) return false;
        return json.getJSONObject("mutation").getBoolean("hasNext");
    }
}

package com.sforzando.jscratch2.scratch.blocks;

import com.sforzando.jscratch2.scratch.blocks.ScratchBlock.Mutation;

public class ScratchBlockType<T extends ScratchBlock> {
    private final boolean isFinal;
    private final boolean hasMutations;

    private final String opcode;

    private final BlockBuilder<T> builder;

    public ScratchBlockType (String opcode, BlockBuilder<T> builder) {
        this.hasMutations = opcode.equals("control_stop") || opcode.equals("procedures_call") || opcode.equals("procedures_prototype");
        this.isFinal = opcode.equals("control_forever") || opcode.equals("control_delete_this_clone");
        this.opcode = opcode;
        this.builder = builder;
    }

    public boolean equals (Object other) {
        if (other.hashCode() == this.hashCode()) return true;
        if (other instanceof ScratchBlockType && other.opcode == this.opcode) return true;
        return false;
    }

    public boolean hasMutations () {
        return this.hasMutations;
    }

    public boolean isFinal () {
        return this.isFinal;
    }

    public Map<String, Object> createMutations (JSONObject obj) {
        if (this.opcode.equals("control_stop")) return Map.<String, Object>of("isFinal", Boolean.valueOf(!obj.getBoolean("hasNext")));
        if (this.opcode.equals("procedures_call") || this.opcode.equals("procedures_prototype")) {
            HashMap<String, Object> mutations = new HashMap<String, Object>();
            String signature = obj.getString("proccode");
            mutations.put("name", signature.replaceAll("%b", "{}").replaceAll("%s", "{}"));
            char[] chars = signature.toCharArray();
            ArrayList<String> args = new ArrayList<String>();
            for (int i = 0; i < chars.length - 1; i++)
                if (chars[i] == '%') {
                    if (chars[i + 1] == 'b')
                        args.add("<>");
                    if (chars[i + 1] == 's')
                        args.add("()");
                }
            String[] matchedArgs = new String[args.size()];
            for (int i = 0; i < args.size(); i++)
                matchedArgs[i] = args.get(i).charAt(0) + new JSONArray(obj.getString("argumentids")).getString(i) + args.get(i).charAt(1);
            mutations.put("arguments", matchedArgs);
            mutations.put("refreshScreen", Boolean.valueOf(!obj.getBoolean("warp")));
            if (this.opcode.equals("procedures_prototype")) {
                String[] matchedNames = new String[args.size()];
                for (int i = 0; i < args.size(); i++)
                    matchedNames[i] = args.get(i).charAt(0) + new JSONArray(obj.getString("argumentnames")).getString(i) + args.get(i).charAt(1);
                mutations.put("argumentNames", matchedNames);
            }
            return mutations;
        }
        return null;
    }

    public ScratchBlockInput parseInputs (String type, Object obj) {
        if ((this.opcode.equals("argument_reporter_string") || this.opcode.equals("argument_reporter_boolean")) && type.equals("VALUE"))
            return new SimpleStringBlockInput(((JSONArray) obj).getString(0));
        if ((this.opcode.equals("argument_reporter_string") || this.opcode.equals("argument_reporter_boolean")) && type.equals("VALUE"))
            return new SimpleStringBlockInput(((JSONArray) obj).getString(0));

    }

    public ScratchBlockInput parseFields (String type, JSONObject json) {
    }
}