diff --git a/src/nsl/instruction/CPUInstruction.java b/src/nsl/instruction/CPUInstruction.java new file mode 100644 index 0000000..8e8f1f0 --- /dev/null +++ b/src/nsl/instruction/CPUInstruction.java @@ -0,0 +1,69 @@ +/* + * CPUInstruction.java + */ + +package nsl.instruction; + +import java.io.IOException; +import java.util.ArrayList; +import java.util.Arrays; +import java.util.EnumSet; +import java.util.HashSet; +import java.util.Set; +import nsl.*; +import nsl.expression.*; + +/** + * @author Jan + */ +public class CPUInstruction extends AssembleExpression { + public static final String name = "CPU"; + + private static final Set VALID_CPUS = new HashSet<>(Arrays.asList("x86", "amd64")); + + private final Expression cpu; + + /** + * Class constructor. + * + * @param returns the number of values to return + */ + public CPUInstruction(int returns) { + if (!ScriptParser.inGlobalContext()) + throw new NslContextException(EnumSet.of(NslContext.Global), name); + if (returns > 0) throw new NslReturnValueException(name); + + ArrayList paramsList = Expression.matchList(); + if (paramsList.size() != 1) throw new NslArgumentException(name, 1); + + this.cpu = paramsList.get(0); + if (!ExpressionType.isString(this.cpu)) + throw new NslArgumentException(name, 1, ExpressionType.String); + + String cpuValue = this.cpu.getStringValue(); + if (!VALID_CPUS.contains(cpuValue)) { + throw new NslException( + String.format( + "%s: Invalid CPU architecture \"%s\". Valid values are: %s", + name, cpuValue, String.join(", ", VALID_CPUS)), + true); + } + } + + /** Assembles the source code. */ + @Override + public void assemble() throws IOException { + AssembleExpression.assembleIfRequired(this.cpu); + ScriptParser.writeLine(name + " " + this.cpu); + } + + /** + * Assembles the source code. + * + * @param var the variable to assign the value to + */ + @Override + public void assemble(Register var) throws IOException { + throw new UnsupportedOperationException("Not supported."); + } +} diff --git a/src/nsl/statement/Statement.java b/src/nsl/statement/Statement.java index f5a0a9f..2b16063 100644 --- a/src/nsl/statement/Statement.java +++ b/src/nsl/statement/Statement.java @@ -93,6 +93,7 @@ public static AssembleExpression matchInstruction(int returns) { return new ComponentTextInstruction(returns); if (ScriptParser.tokenizer.match(CopyFilesInstruction.name)) return new CopyFilesInstruction(returns); + if (ScriptParser.tokenizer.match(CPUInstruction.name)) return new CPUInstruction(returns); if (ScriptParser.tokenizer.match(CRCCheckInstruction.name)) return new CRCCheckInstruction(returns); if (ScriptParser.tokenizer.match(CreateDirectoryInstruction.name))