ocaml for nix

runtimes/default.nix

@@ -29,4 +29,5 @@ args: {
     "php" = import ./php.nix args;
     "perl" = import ./perl.nix args;
     "octave" = import ./octave.nix args;
+    "ocaml" = import ./ocaml.nix args;
 }

runtimes/ocaml.nix

@@ -0,0 +1,35 @@
+{pkgs, piston, ...}:
+let
+    pkg = pkgs.ocaml; 
+in piston.mkRuntime {
+    language = "ocaml";
+    version = pkg.version;
+
+    aliases = [
+        "ml"
+    ];
+
+    compile = ''
+        rename 's/$/\.ml/' "$@" # Add .ml extension
+        ${pkg}/bin/ocamlc -o out *.ml
+    '';
+
+    run = ''
+        shift
+        ./out "$@"
+    '';
+
+    tests = [
+        (piston.mkTest {
+            files = {
+                "test.ml" = ''
+                    print_string "OK\n";
+                '';
+            };
+            args = [];
+            stdin = "";
+            packages = [];
+            main = "test.ml";
+        })
+    ];
+}