SnuviParser.java 33 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841
  1. package me.hammerle.code;
  2. import java.io.File;
  3. import java.io.IOException;
  4. import java.lang.reflect.InvocationTargetException;
  5. import java.nio.charset.StandardCharsets;
  6. import java.nio.file.Files;
  7. import java.nio.file.Paths;
  8. import me.hammerle.exceptions.HoldCodeException;
  9. import java.time.ZonedDateTime;
  10. import java.util.ArrayList;
  11. import java.util.Arrays;
  12. import java.util.Calendar;
  13. import java.util.Collection;
  14. import java.util.Collections;
  15. import java.util.GregorianCalendar;
  16. import java.util.HashMap;
  17. import java.util.HashSet;
  18. import java.util.List;
  19. import java.util.Map;
  20. import java.util.Set;
  21. import java.util.function.BiFunction;
  22. import java.util.function.BiConsumer;
  23. import java.util.function.Consumer;
  24. import java.util.function.Predicate;
  25. import java.util.stream.Collectors;
  26. import me.hammerle.exceptions.FileIOException;
  27. import me.hammerle.exceptions.PreScriptException;
  28. import me.hammerle.math.Fraction;
  29. public class SnuviParser
  30. {
  31. protected final ISnuviLogger logger;
  32. private final ISnuviScheduler scheduler;
  33. private final HashMap<String, BiFunction<Object[], Script, Object>> functions;
  34. public boolean printStack;
  35. private int idCounter;
  36. private final HashMap<Integer, Script> scripts;
  37. private final ArrayList<Integer> termQueue;
  38. public SnuviParser(ISnuviLogger logger, ISnuviScheduler scheduler)
  39. {
  40. this.logger = logger;
  41. this.scheduler = scheduler;
  42. functions = new HashMap<>();
  43. registerStandardLibraries();
  44. printStack = false;
  45. scripts = new HashMap<>();
  46. termQueue = new ArrayList<>();
  47. idCounter = 0;
  48. }
  49. // -----------------------------------------------------------------------------------
  50. // function registry
  51. // -----------------------------------------------------------------------------------
  52. public boolean isRegisteredFunction(String s)
  53. {
  54. return functions.containsKey(s);
  55. }
  56. public void registerFunction(String s, BiFunction<Object[], Script, Object> f)
  57. {
  58. functions.put(s, f);
  59. }
  60. public void registerConsumer(String s, BiConsumer<Object[], Script> f)
  61. {
  62. functions.put(s, (BiFunction<Object[], Script, Object>) (args, sc) ->
  63. {
  64. f.accept(args, sc);
  65. return Void.TYPE;
  66. });
  67. }
  68. public void registerAlias(String alias, String original)
  69. {
  70. BiFunction<Object[], Script, Object> f = functions.get(original);
  71. if(f == null)
  72. {
  73. throw new IllegalArgumentException("registering alias for non existent function");
  74. }
  75. registerFunction(alias, functions.get(original));
  76. }
  77. @SuppressWarnings("unchecked")
  78. private void registerStandardLibraries()
  79. {
  80. registerFunction("nothing", (args, sc) ->
  81. 0);
  82. registerFunction("", (args, sc) ->
  83. args[0]);
  84. registerConsumer("error", (args, sc) ->
  85. printStack = !printStack);
  86. // -------------------------------------------------------------------------------
  87. // event
  88. // -------------------------------------------------------------------------------
  89. registerConsumer("event.load", (args, sc) ->
  90. sc.loadEvent(args[0].toString()));
  91. registerConsumer("event.unload", (args, sc) ->
  92. sc.unloadEvent(args[0].toString()));
  93. registerFunction("event.isloaded", (args, sc) ->
  94. sc.isLoadedEvent(args[0].toString()));
  95. // -------------------------------------------------------------------------------
  96. // bit
  97. // -------------------------------------------------------------------------------
  98. registerFunction("bit.rightshiftn", (args, qd) ->
  99. ((Fraction) args[0]).rightShiftNumerator(getInt(args[1])));
  100. registerFunction("bit.rightshiftd", (args, qd) ->
  101. ((Fraction) args[0]).rightShiftDenominator(getInt(args[1])));
  102. registerFunction("bit.leftshiftn", (args, qd) ->
  103. ((Fraction) args[0]).leftShiftNumerator(getInt(args[1])));
  104. registerFunction("bit.leftshiftd", (args, qd) ->
  105. ((Fraction) args[0]).leftShiftDenominator(getInt(args[1])));
  106. registerFunction("bit.and", (args, qd) ->
  107. ((Fraction) args[0]).and((Fraction) args[1]));
  108. registerFunction("bit.or", (args, qd) ->
  109. ((Fraction) args[0]).or((Fraction) args[1]));
  110. registerFunction("bit.xor", (args, qd) ->
  111. ((Fraction) args[0]).xor((Fraction) args[1]));
  112. registerFunction("bit.invert", (args, qd) ->
  113. ((Fraction) args[0]).invert());
  114. registerFunction("bit.setn", (args, qd) ->
  115. ((Fraction) args[0]).setNumeratorBit(getInt(args[1])));
  116. registerFunction("bit.setd", (args, qd) ->
  117. ((Fraction) args[0]).setDenominatorBit(getInt(args[1])));
  118. registerFunction("bit.unsetn", (args, qd) ->
  119. ((Fraction) args[0]).unsetNumeratorBit(getInt(args[1])));
  120. registerFunction("bit.unsetd", (args, qd) ->
  121. ((Fraction) args[0]).unsetDenominatorBit(getInt(args[1])));
  122. registerFunction("bit.getn", (args, qd) ->
  123. ((Fraction) args[0]).getNumeratorBit(getInt(args[1])));
  124. registerFunction("bit.getd", (args, qd) ->
  125. ((Fraction) args[0]).getDenominatorBit(getInt(args[1])));
  126. // -------------------------------------------------------------------------------
  127. // math
  128. // -------------------------------------------------------------------------------
  129. registerFunction("math.mod", (args, sc) ->
  130. new Fraction(getInt(args[0]) % getInt(args[1])));
  131. registerFunction("math.abs", (args, sc) ->
  132. ((Fraction) args[0]).abs());
  133. registerFunction("math.pow", (args, sc) ->
  134. ((Fraction) args[0]).power((Fraction) args[1]));
  135. registerFunction("math.root", (args, sc) ->
  136. ((Fraction) args[0]).power(((Fraction) args[1]).invert()));
  137. registerFunction("math.sin", (args, sc) ->
  138. ((Fraction) args[0]).sin());
  139. registerFunction("math.cos", (args, sc) ->
  140. ((Fraction) args[0]).cos());
  141. registerFunction("math.tan", (args, sc) ->
  142. ((Fraction) args[0]).tan());
  143. registerFunction("math.asin", (args, sc) ->
  144. ((Fraction) args[0]).asin());
  145. registerFunction("math.acos", (args, sc) ->
  146. ((Fraction) args[0]).acos());
  147. registerFunction("math.atan", (args, sc) ->
  148. ((Fraction) args[0]).atan());
  149. registerFunction("math.e", (args, sc) ->
  150. Fraction.E);
  151. registerFunction("math.pi", (args, sc) ->
  152. Fraction.PI);
  153. registerFunction("math.ln", (args, sc) ->
  154. ((Fraction) args[0]).log());
  155. registerFunction("math.log", (args, sc) ->
  156. ((Fraction) args[0]).log());
  157. registerFunction("math.random", (args, sc) ->
  158. new Fraction(ScriptUtils.randomInt(getInt(args[0]), getInt(args[1]))));
  159. registerFunction("math.round", (args, sc) ->
  160. ((Fraction) args[0]).round());
  161. registerFunction("math.rounddown", (args, sc) ->
  162. ((Fraction) args[0]).floor());
  163. registerFunction("math.roundup", (args, sc) ->
  164. ((Fraction) args[0]).ceil());
  165. registerFunction("math.roundcomma", (args, sc) ->
  166. ((Fraction) args[0]).round(getInt(args[1])));
  167. // -------------------------------------------------------------------------------
  168. // lists
  169. // -------------------------------------------------------------------------------
  170. registerConsumer("list.new", (args, sc) ->
  171. sc.setVar(args[0].toString(), new ArrayList<>()));
  172. registerFunction("list.exists", (args, sc) ->
  173. args[0] instanceof List);
  174. registerConsumer("list.add", (args, sc) ->
  175. ((List) args[0]).add(args[1]));
  176. registerConsumer("list.remove", (args, sc) ->
  177. ((List) args[0]).remove(args[1]));
  178. registerConsumer("list.removeindex", (args, sc) ->
  179. ((List) args[0]).remove(getInt(args[1])));
  180. registerFunction("list.contains", (args, sc) ->
  181. ((List) args[0]).contains(args[1]));
  182. registerFunction("list.getsize", (args, sc) ->
  183. new Fraction(((List) args[0]).size()));
  184. registerFunction("list.getindex", (args, sc) ->
  185. ((List) args[0]).get(getInt(args[1])));
  186. registerConsumer("list.setindex", (args, sc) ->
  187. ((List) args[0]).set(getInt(args[1]), args[2]));
  188. registerConsumer("list.clear", (args, sc) ->
  189. ((List) args[0]).clear());
  190. registerFunction("list.getindexof", (args, sc) ->
  191. new Fraction(((List) args[0]).indexOf(args[1])));
  192. registerConsumer("list.sort", (args, sc) ->
  193. sortList((List<Object>) args[0], sc));
  194. registerConsumer("list.reverse", (args, sc) ->
  195. Collections.reverse((List<Object>) args[0]));
  196. registerConsumer("list.shuffle", (args, sc) ->
  197. Collections.shuffle((List<Object>) args[0]));
  198. // -------------------------------------------------------------------------------
  199. // maps
  200. // -------------------------------------------------------------------------------
  201. registerConsumer("map.new", (args, sc) ->
  202. sc.setVar(args[0].toString(), new HashMap<>()));
  203. registerFunction("map.exists", (args, sc) ->
  204. args[0] instanceof Map);
  205. registerConsumer("map.add", (args, sc) ->
  206. ((HashMap) args[0]).put(args[1], args[2]));
  207. registerConsumer("map.remove", (args, sc) ->
  208. ((HashMap) args[0]).remove(args[1]));
  209. registerFunction("map.contains", (args, sc) ->
  210. ((HashMap) args[0]).containsKey(args[1]));
  211. registerFunction("map.getsize", (args, sc) ->
  212. new Fraction(((HashMap) args[0]).size()));
  213. registerFunction("map.get", (args, sc) ->
  214. ((HashMap) args[0]).get(args[1]));
  215. registerConsumer("map.clear", (args, sc) ->
  216. ((HashMap) args[0]).clear());
  217. // -------------------------------------------------------------------------------
  218. // sets
  219. // -------------------------------------------------------------------------------
  220. registerConsumer("set.new", (args, sc) ->
  221. sc.setVar(args[0].toString(), new HashSet<>()));
  222. registerFunction("set.exists", (args, sc) ->
  223. args[0] instanceof Set);
  224. registerConsumer("set.add", (args, sc) ->
  225. ((HashSet) args[0]).add(args[1]));
  226. registerConsumer("set.remove", (args, sc) ->
  227. ((HashSet) args[0]).remove(args[1]));
  228. registerFunction("set.contains", (args, sc) ->
  229. ((HashSet) args[0]).contains(args[1]));
  230. registerFunction("set.getsize", (args, sc) ->
  231. new Fraction(((HashSet) args[0]).size()));
  232. registerConsumer("set.tolist", (args, sc) ->
  233. sc.setVar(args[0].toString(), ((HashSet) args[1]).stream().collect(Collectors.toList())));
  234. // -------------------------------------------------------------------------------
  235. // time
  236. // -------------------------------------------------------------------------------
  237. registerFunction("time.get", (args, sc) ->
  238. new Fraction(System.currentTimeMillis()));
  239. registerFunction("time.nextday", (args, sc) ->
  240. getNextDay(args));
  241. registerFunction("time.getyear", (args, sc) ->
  242. getYear(args));
  243. registerFunction("time.getmonth", (args, sc) ->
  244. getMonth(args));
  245. registerFunction("time.getday", (args, sc) ->
  246. getDay(args));
  247. registerFunction("time.gethour", (args, sc) ->
  248. getHour(args));
  249. registerFunction("time.getminute", (args, sc) ->
  250. getMinute(args));
  251. registerFunction("time.getsecond", (args, sc) ->
  252. getSecond(args));
  253. // -------------------------------------------------------------------------------
  254. // text
  255. // -------------------------------------------------------------------------------
  256. registerFunction("text.number", (args, sc) ->
  257. fractionToDoubleString((Fraction) args[0]));
  258. registerFunction("text.class", (args, sc) ->
  259. args[0].getClass().getSimpleName());
  260. registerFunction("text.tolowercase", (args, sc) ->
  261. ScriptUtils.connect(args, 0).toLowerCase());
  262. registerAlias("tolowercase", "text.tolowercase");
  263. registerFunction("text.touppercase", (args, sc) ->
  264. ScriptUtils.connect(args, 0).toUpperCase());
  265. registerAlias("touppercase", "text.touppercase");
  266. registerConsumer("text.split", (args, sc) ->
  267. split(args, sc));
  268. registerAlias("split", "text.split");
  269. registerFunction("text.concatlist", (args, sc) ->
  270. ((List<Object>) args[0]).stream().limit(getInt(args[3]) + 1).skip(getInt(args[2])).map(o -> o.toString()).collect(Collectors.joining(args[1].toString())));
  271. registerAlias("concatlist", "text.concatlist");
  272. registerFunction("text.concat", (args, sc) ->
  273. ScriptUtils.connect(args, 0));
  274. registerAlias("concat", "text.concat");
  275. registerFunction("text", (args, sc) ->
  276. String.valueOf(args[0]));
  277. registerFunction("text.substring", (args, sc) ->
  278. args[0].toString().substring(getInt(args[1]), getInt(args[2])));
  279. registerFunction("text.length", (args, sc) ->
  280. args[0].toString().length());
  281. registerFunction("text.startswith", (args, sc) ->
  282. args[0].toString().startsWith(args[1].toString(), getInt(args[2])));
  283. registerFunction("text.endswith", (args, sc) ->
  284. args[0].toString().endsWith(args[1].toString()));
  285. registerFunction("text.contains", (args, sc) ->
  286. args[0].toString().contains(args[1].toString()));
  287. registerFunction("text.indexof", (args, sc) ->
  288. args[0].toString().indexOf(args[1].toString(), getInt(args[2])));
  289. registerFunction("text.lastindexof", (args, sc) ->
  290. args[0].toString().lastIndexOf(args[1].toString(), getInt(args[2])));
  291. registerFunction("text.replace", (args, sc) ->
  292. args[0].toString().replace(args[1].toString(), args[2].toString()));
  293. registerFunction("text.trim", (args, sc) ->
  294. args[0].toString().trim());
  295. registerFunction("text.charat", (args, sc) ->
  296. String.valueOf(args[0].toString().charAt(getInt(args[1]))));
  297. // -------------------------------------------------------------------------------
  298. // files
  299. // -------------------------------------------------------------------------------
  300. registerFunction("file.new", (args, sc) ->
  301. new File(args[0].toString()));
  302. registerFunction("file.exists", (args, sc) ->
  303. ((File) args[0]).exists());
  304. registerFunction("file.delete", (args, sc) ->
  305. ((File) args[0]).delete());
  306. registerFunction("file.getname", (args, sc) ->
  307. ((File) args[0]).getName());
  308. registerFunction("file.", (args, sc) ->
  309. ((File) args[0]).renameTo(new File(args[1].toString())));
  310. registerConsumer("file.getlist", (args, sc) ->
  311. sc.setVar(args[0].toString(), Arrays.asList(((File) args[0]).listFiles())));
  312. registerConsumer("file.read", (args, sc) ->
  313. readFile(args, sc));
  314. registerConsumer("file.write", (args, sc) ->
  315. writeFile(args, sc));
  316. // -------------------------------------------------------------------------------
  317. // commands without library
  318. // -------------------------------------------------------------------------------
  319. registerFunction("add", (args, sc) ->
  320. ((Fraction) args[0]).add((Fraction) args[1]));
  321. registerFunction("sub", (args, sc) ->
  322. ((Fraction) args[0]).sub((Fraction) args[1]));
  323. registerFunction("inc", (args, sc) ->
  324. increaseVar(args[0], sc, 1));
  325. registerFunction("dec", (args, sc) ->
  326. increaseVar(args[0], sc, -1));
  327. registerFunction("mul", (args, sc) ->
  328. ((Fraction) args[0]).mul((Fraction) args[1]));
  329. registerFunction("div", (args, sc) ->
  330. ((Fraction) args[0]).div((Fraction) args[1]));
  331. registerFunction("getvar", (args, sc) ->
  332. sc.getVar(args[0].toString()));
  333. registerConsumer("setvar", (args, sc) ->
  334. sc.setVar(args[0].toString(), args[1]));
  335. registerConsumer("removevar", (args, sc) ->
  336. sc.removeVar(args[0].toString()));
  337. registerConsumer("reset", (args, sc) ->
  338. sc.resetLoopCounter());
  339. registerConsumer("wait", (args, sc) ->
  340. { sc.resetLoopCounter(); throw new HoldCodeException(); });
  341. registerConsumer("goto", (args, sc) ->
  342. goTo(args, sc));
  343. registerConsumer("sgoto", (args, sc) ->
  344. scheduleGoto(args, sc));
  345. registerConsumer("gosub", (args, sc) ->
  346. sc.gotoLabelWithReturn(args[0].toString()));
  347. registerConsumer("return", (args, sc) ->
  348. sc.doReturn());
  349. registerConsumer("try", (args, sc) ->
  350. {sc.saveTryJumpLine(); sc.jump(1);});
  351. registerConsumer("catch", (args, sc) ->
  352. sc.resetTryJumpLine());
  353. registerConsumer("if", (args, sc) ->
  354. ifFunction(args, sc));
  355. registerConsumer("else", (args, sc) ->
  356. {});
  357. registerConsumer("while", (args, sc) ->
  358. whileFunction(args, sc));
  359. registerFunction("equal", (args, sc) ->
  360. isEqual(args));
  361. registerAlias("equals", "equal");
  362. registerFunction("less", (args, sc) ->
  363. ((Fraction) args[0]).compareTo((Fraction) args[1]) < 0);
  364. registerFunction("greater", (args, sc) ->
  365. ((Fraction) args[0]).compareTo((Fraction) args[1]) > 0);
  366. registerFunction("notequal", (args, sc) ->
  367. !isEqual(args));
  368. registerFunction("lessequal", (args, sc) ->
  369. ((Fraction) args[0]).compareTo((Fraction) args[1]) <= 0);
  370. registerFunction("greaterequal", (args, sc) ->
  371. ((Fraction) args[0]).compareTo((Fraction) args[1]) >= 0);
  372. registerFunction("invert", (args, sc) ->
  373. !((boolean) args[0]));
  374. registerFunction("and", (args, sc) ->
  375. Arrays.stream(args).allMatch(s -> s.equals(true)));
  376. registerFunction("or", (args, sc) ->
  377. Arrays.stream(args).anyMatch(s -> s.equals(true)));
  378. registerConsumer("waitfor", (args, sc) ->
  379. waitFor(args, sc));
  380. registerConsumer("term", (args, sc) ->
  381. { termSafe(sc); throw new HoldCodeException(); });
  382. }
  383. @SuppressWarnings("unchecked")
  384. protected Object parseFunction(Script sc, String function, Object[] args) throws HoldCodeException
  385. {
  386. try
  387. {
  388. return functions.get(function).apply(args, sc);
  389. }
  390. catch(HoldCodeException ex)
  391. {
  392. throw ex;
  393. }
  394. catch(Exception ex)
  395. {
  396. if(printStack)
  397. {
  398. ex.printStackTrace();
  399. }
  400. if(sc.getTryJumpLine() != -1)
  401. {
  402. sc.setVar("error", ex.getClass().getSimpleName());
  403. sc.gotoTryJumpLine();
  404. sc.resetTryJumpLine();
  405. return Void.TYPE;
  406. }
  407. logger.printException(ex, sc, sc.getActiveRealCodeLine());
  408. sc.resetLoopCounter();
  409. throw new HoldCodeException();
  410. }
  411. }
  412. // -----------------------------------------------------------------------------------
  413. // script controller
  414. // -----------------------------------------------------------------------------------
  415. public Script getScript(int id)
  416. {
  417. return scripts.get(id);
  418. }
  419. public boolean termUnsafe(Script sc)
  420. {
  421. if(sc == null)
  422. {
  423. return false;
  424. }
  425. sc.setInvalid();
  426. sc.onTerm();
  427. return scripts.remove(sc.getId()) != null;
  428. }
  429. public void termSafe(Script sc)
  430. {
  431. if(sc == null)
  432. {
  433. return;
  434. }
  435. sc.setInvalid();
  436. termQueue.add(sc.getId());
  437. }
  438. private void term()
  439. {
  440. if(!termQueue.isEmpty())
  441. {
  442. termQueue.forEach(i ->
  443. {
  444. Script sc = scripts.remove(i);
  445. if(sc != null)
  446. {
  447. sc.onTerm();
  448. }
  449. });
  450. termQueue.clear();
  451. }
  452. }
  453. public void termAllUnsafe()
  454. {
  455. scripts.values().forEach(sc ->
  456. {
  457. sc.onTerm();
  458. sc.setInvalid();
  459. });
  460. scripts.clear();
  461. }
  462. public Collection<Script> getScripts()
  463. {
  464. return scripts.values();
  465. }
  466. private Script startScript(Script script, Object... o)
  467. {
  468. try
  469. {
  470. script.initExpansion(o);
  471. scripts.put(idCounter, script);
  472. idCounter++;
  473. script.runCode();
  474. term();
  475. return script;
  476. }
  477. catch(PreScriptException ex)
  478. {
  479. logger.printException(ex, script, ex.getLine());
  480. return null;
  481. }
  482. }
  483. public Script startScript(String scriptName, String code)
  484. {
  485. return startScript(new Script(this, idCounter, scriptName, code));
  486. }
  487. public <T extends Script> T startScript(Class<T> c, String scriptName, String code, boolean b, Object... o)
  488. {
  489. try
  490. {
  491. return (T) startScript(c.getDeclaredConstructor(
  492. SnuviParser.class, int.class, String.class, String.class, boolean.class)
  493. .newInstance(this, idCounter, scriptName, code, b), o);
  494. }
  495. catch(IllegalAccessException | IllegalArgumentException | InstantiationException |
  496. InvocationTargetException | NoSuchMethodException | SecurityException ex)
  497. {
  498. if(ex instanceof InvocationTargetException)
  499. {
  500. Throwable t = ((InvocationTargetException) ex).getCause();
  501. if(t != null && t instanceof Exception)
  502. {
  503. logger.printException((Exception) t);
  504. return null;
  505. }
  506. }
  507. ex.printStackTrace();
  508. return null;
  509. }
  510. }
  511. // -----------------------------------------------------------------------------------
  512. // event
  513. // -----------------------------------------------------------------------------------
  514. public void callEvent(String name, Consumer<Script> before, Consumer<Script> after, Predicate<Script> check)
  515. {
  516. scripts.values().stream()
  517. .filter(sc -> sc.receiveEventBroadcast && !sc.isHalt() && !sc.isRunning())
  518. .filter(sc -> sc.isLoadedEvent(name))
  519. .filter(check)
  520. .forEach(sc ->
  521. {
  522. sc.setVar("event", name);
  523. if(before != null)
  524. {
  525. before.accept(sc);
  526. }
  527. sc.runCode();
  528. if(after != null)
  529. {
  530. after.accept(sc);
  531. }
  532. sc.eventVars.forEach(s -> sc.removeVar(s));
  533. sc.eventVars.clear();
  534. });
  535. term();
  536. }
  537. public void callEvent(String name, Consumer<Script> before, Consumer<Script> after)
  538. {
  539. callEvent(name, before, after, sc -> true);
  540. }
  541. public boolean callEvent(String name, Script sc, Consumer<Script> before, Consumer<Script> after, boolean check)
  542. {
  543. if(sc.isLoadedEvent(name) && !sc.isHalt() && !sc.isRunning() && check)
  544. {
  545. sc.setVar("event", name);
  546. if(before != null)
  547. {
  548. before.accept(sc);
  549. }
  550. sc.runCode();
  551. if(after != null)
  552. {
  553. after.accept(sc);
  554. }
  555. sc.eventVars.forEach(s -> sc.removeVar(s));
  556. sc.eventVars.clear();
  557. term();
  558. return true;
  559. }
  560. return false;
  561. }
  562. public boolean callEvent(String name, Script sc, Consumer<Script> before, Consumer<Script> after)
  563. {
  564. return callEvent(name, sc, before, after, true);
  565. }
  566. // -----------------------------------------------------------------------------------
  567. // functions
  568. // -----------------------------------------------------------------------------------
  569. private String fractionToDoubleString(Fraction f)
  570. {
  571. if(f.doubleValue() == f.longValue())
  572. {
  573. return String.valueOf(f.longValue());
  574. }
  575. return String.valueOf(f.doubleValue());
  576. }
  577. private void ifFunction(Object[] args, Script sc)
  578. {
  579. if(Arrays.stream(args).anyMatch(s -> !((boolean) s)))
  580. {
  581. sc.jumpNextDoElse();
  582. }
  583. else
  584. {
  585. sc.jump(1);
  586. }
  587. }
  588. private void whileFunction(Object[] args, Script sc)
  589. {
  590. if(Arrays.stream(args).allMatch(s -> ((boolean) s)))
  591. {
  592. sc.jump(1);
  593. }
  594. }
  595. @SuppressWarnings("unchecked")
  596. private void sortList(List<Object> args, Script sc)
  597. {
  598. Collections.sort(args, (Object o1, Object o2) -> ((Comparable) o1).compareTo(o2));
  599. }
  600. private void scheduleGoto(Object[] args, Script sc)
  601. {
  602. scheduler.scheduleTask(() ->
  603. {
  604. if(!sc.isValid() && !sc.isHalt())
  605. {
  606. return;
  607. }
  608. try
  609. {
  610. sc.gotoLabel(args[1].toString(), true);
  611. sc.runCode();
  612. }
  613. catch(Exception ex)
  614. {
  615. logger.printException(ex, sc, sc.getActiveRealCodeLine());
  616. }
  617. }, getInt(args[0]));
  618. }
  619. @SuppressWarnings("")
  620. private void waitFor(Object[] args, Script sc) throws UnsupportedOperationException
  621. {
  622. sc.resetLoopCounter();
  623. long l = getLong(args[0]);
  624. if(l < 0)
  625. {
  626. throw new IllegalArgumentException("time units can't be negative");
  627. }
  628. sc.setHalt(true);
  629. scheduler.scheduleTask(() ->
  630. {
  631. if(sc == null || !sc.isValid())
  632. {
  633. return;
  634. }
  635. sc.setHalt(false);
  636. sc.runCode();
  637. }, l);
  638. throw new HoldCodeException();
  639. }
  640. private Number increaseVar(Object var, Script sc, int value)
  641. {
  642. Fraction n = ((Fraction) sc.getVar(var.toString())).add(new Fraction(value));
  643. sc.setVar(var.toString(), n);
  644. return n;
  645. }
  646. private static boolean isEqual(Object[] args)
  647. {
  648. if(args[0] == null)
  649. {
  650. return args[1] == null;
  651. }
  652. else if(args[1] == null)
  653. {
  654. return false;
  655. }
  656. return args[0].equals(args[1]);
  657. }
  658. private void split(Object[] args, Script sc)
  659. {
  660. sc.setVar(args[0].toString(),
  661. Arrays.stream(ScriptUtils.connect(args, 2).split(args[1].toString()))
  662. .map(s -> Code.convertInput(null, s, false)).collect(Collectors.toList()));
  663. }
  664. private void goTo(Object[] args, Script sc)
  665. {
  666. if(args.length == 0)
  667. {
  668. sc.gotoLabel(null, false);
  669. return;
  670. }
  671. sc.gotoLabel(args[0].toString(), false);
  672. }
  673. private void readFile(Object[] args, Script sc)
  674. {
  675. try
  676. {
  677. sc.setVar(args[0].toString(), Files.readAllLines(((File) args[1]).toPath()));
  678. }
  679. catch(IOException ex)
  680. {
  681. throw new FileIOException(ex.getMessage());
  682. }
  683. }
  684. private void writeFile(Object[] args, Script sc)
  685. {
  686. try
  687. {
  688. File f = (File) args[0];
  689. if(!f.getParentFile().exists())
  690. {
  691. f.getParentFile().mkdirs();
  692. }
  693. if(!f.exists())
  694. {
  695. try
  696. {
  697. f.createNewFile();
  698. }
  699. catch(IOException ex)
  700. {
  701. throw new FileIOException(ex.getMessage());
  702. }
  703. }
  704. Files.write(Paths.get(f.toURI()), (List<String>) args[1], StandardCharsets.UTF_8);
  705. }
  706. catch(UnsupportedOperationException | SecurityException | IOException ex)
  707. {
  708. throw new FileIOException(ex.getMessage());
  709. }
  710. }
  711. // -----------------------------------------------------------------------------------
  712. // number handlers, cause primitive types transform into their classes all the time
  713. // -----------------------------------------------------------------------------------
  714. public static int getInt(Object o)
  715. {
  716. return ScriptUtils.getInt(o);
  717. }
  718. public static long getLong(Object o)
  719. {
  720. return ScriptUtils.getLong(o);
  721. }
  722. public static double getDouble(Object o)
  723. {
  724. return ScriptUtils.getDouble(o);
  725. }
  726. // -----------------------------------------------------------------------------------
  727. // time handler
  728. // -----------------------------------------------------------------------------------
  729. private Fraction getNextDay(Object[] args)
  730. {
  731. GregorianCalendar cal = GregorianCalendar.from(ZonedDateTime.now());
  732. cal.setTimeInMillis(getLong(args[0]));
  733. cal.add(Calendar.DAY_OF_YEAR, 1);
  734. cal.set(Calendar.HOUR, 0);
  735. cal.set(Calendar.SECOND, 0);
  736. cal.set(Calendar.MINUTE, 0);
  737. cal.set(Calendar.MILLISECOND, 0);
  738. return new Fraction(cal.getTimeInMillis());
  739. }
  740. private Fraction getYear(Object[] args)
  741. {
  742. GregorianCalendar cal = GregorianCalendar.from(ZonedDateTime.now());
  743. cal.setTimeInMillis(getLong(args[0]));
  744. return new Fraction(cal.get(Calendar.YEAR));
  745. }
  746. private Fraction getMonth(Object[] args)
  747. {
  748. GregorianCalendar cal = GregorianCalendar.from(ZonedDateTime.now());
  749. cal.setTimeInMillis(getLong(args[0]));
  750. return new Fraction(cal.get(Calendar.MONTH) + 1);
  751. }
  752. private Fraction getDay(Object[] args)
  753. {
  754. GregorianCalendar cal = GregorianCalendar.from(ZonedDateTime.now());
  755. cal.setTimeInMillis(getLong(args[0]));
  756. return new Fraction(cal.get(Calendar.DAY_OF_MONTH));
  757. }
  758. private Fraction getHour(Object[] args)
  759. {
  760. GregorianCalendar cal = GregorianCalendar.from(ZonedDateTime.now());
  761. cal.setTimeInMillis(getLong(args[0]));
  762. return new Fraction(cal.get(Calendar.HOUR_OF_DAY));
  763. }
  764. private Fraction getMinute(Object[] args)
  765. {
  766. GregorianCalendar cal = GregorianCalendar.from(ZonedDateTime.now());
  767. cal.setTimeInMillis(getLong(args[0]));
  768. return new Fraction(cal.get(Calendar.MINUTE));
  769. }
  770. private Fraction getSecond(Object[] args)
  771. {
  772. GregorianCalendar cal = GregorianCalendar.from(ZonedDateTime.now());
  773. cal.setTimeInMillis(getLong(args[0]));
  774. return new Fraction(cal.get(Calendar.SECOND));
  775. }
  776. }