Поддержка функций-расширений для строк

Author: aNNiMON

src/main/java/com/annimon/ownlang/lib/StringValue.java

@@ -17,17 +17,30 @@ public StringValue(String value) {
         this.value = value;
     }
 
-    public Value access(Value property) {
-        switch (property.asString()) {
+    public Value access(Value propertyValue) {
+        final String prop = propertyValue.asString();
+        switch (prop) {
             // Properties
             case "length":
                 return NumberValue.of(length());
 
             // Functions
             case "trim":
                 return new FunctionValue(args -> new StringValue(value.trim()));
+
+            default:
+                if (Functions.isExists(prop)) {
+                    final Function f = Functions.get(prop);
+                    return new FunctionValue(args -> {
+                        final Value[] newArgs = new Value[args.length + 1];
+                        newArgs[0] = this;
+                        System.arraycopy(args, 0, newArgs, 1, args.length);
+                        return f.execute(newArgs);
+                    });
+                }
+                break;
         }
-        throw new UnknownPropertyException(property.asString());
+        throw new UnknownPropertyException(prop);
     }
 
     public int length() {

src/test/resources/other/stringFunctions.own

@@ -7,4 +7,15 @@ def testLength() {
 def testTrim() {
   s = "  test   "
   assertEquals("test", s.trim())
+}
+
+def testExtensionFunction() {
+  use "std"
+  s = "1es1"
+  assertEquals("test", s.replace("1", "t"))
+}
+
+def testExtensionCustomFunction() {
+  def repeat(str, num) = str * num
+  assertEquals("****", "*".repeat(4))
 }