Handle arguments of all types in built-ins

Author: stasm

fluent/src/builtins.js

@@ -11,14 +11,7 @@
  * `FluentType`.  Functions must return `FluentType` objects as well.
  */
 
-import { FluentNumber, FluentDateTime } from "./types.js";
-
-export default {
-  "NUMBER": ([arg], opts) =>
-    new FluentNumber(arg.valueOf(), merge(arg.opts, opts)),
-  "DATETIME": ([arg], opts) =>
-    new FluentDateTime(arg.valueOf(), merge(arg.opts, opts)),
-};
+import { FluentNone, FluentNumber, FluentDateTime } from "./types.js";
 
 function merge(argopts, opts) {
   return Object.assign({}, argopts, values(opts));
@@ -31,3 +24,25 @@ function values(opts) {
   }
   return unwrapped;
 }
+
+export
+function NUMBER([arg], opts) {
+  if (arg instanceof FluentNone) {
+    return arg;
+  }
+  if (arg instanceof FluentNumber) {
+    return new FluentNumber(arg.valueOf(), merge(arg.opts, opts));
+  }
+  return new FluentNone("NUMBER()");
+}
+
+export
+function DATETIME([arg], opts) {
+  if (arg instanceof FluentNone) {
+    return arg;
+  }
+  if (arg instanceof FluentDateTime) {
+    return new FluentDateTime(arg.valueOf(), merge(arg.opts, opts));
+  }
+  return new FluentNone("DATETIME()");
+}

fluent/src/resolver.js

@@ -37,7 +37,7 @@
 
 import { FluentType, FluentNone, FluentNumber, FluentDateTime }
   from "./types.js";
-import builtins from "./builtins.js";
+import * as builtins from "./builtins.js";
 
 // Prevent expansion of too long placeables.
 const MAX_PLACEABLE_LENGTH = 2500;

fluent/test/functions_builtin_test.js

@@ -22,13 +22,13 @@ suite('Built-in functions', function() {
       let msg;
 
       msg = bundle.getMessage('num-decimal');
-      assert.equal(bundle.format(msg), 'NaN');
+      assert.equal(bundle.format(msg), '{$arg}');
 
       msg = bundle.getMessage('num-percent');
-      assert.equal(bundle.format(msg), 'NaN');
+      assert.equal(bundle.format(msg), '{$arg}');
 
       msg = bundle.getMessage('num-bad-opt');
-      assert.equal(bundle.format(msg), 'NaN');
+      assert.equal(bundle.format(msg), '{$arg}');
     });
 
     test('number argument', function() {
@@ -50,13 +50,42 @@ suite('Built-in functions', function() {
       let msg;
 
       msg = bundle.getMessage('num-decimal');
-      assert.equal(bundle.format(msg, args), 'NaN');
+      assert.equal(bundle.format(msg, args), '{NUMBER()}');
 
       msg = bundle.getMessage('num-percent');
-      assert.equal(bundle.format(msg, args), 'NaN');
+      assert.equal(bundle.format(msg, args), '{NUMBER()}');
 
       msg = bundle.getMessage('num-bad-opt');
-      assert.equal(bundle.format(msg, args), 'NaN');
+      assert.equal(bundle.format(msg, args), '{NUMBER()}');
+    });
+
+    test('date argument', function() {
+      const date = new Date('2016-09-29');
+      const args = {arg: date};
+      let msg;
+
+      msg = bundle.getMessage('num-decimal');
+      assert.equal(bundle.format(msg, args), '{NUMBER()}');
+
+      msg = bundle.getMessage('num-percent');
+      assert.equal(bundle.format(msg, args), '{NUMBER()}');
+
+      msg = bundle.getMessage('num-bad-opt');
+      assert.equal(bundle.format(msg, args), '{NUMBER()}');
+    });
+
+    test('invalid argument', function() {
+      const args = {arg: []};
+      let msg;
+
+      msg = bundle.getMessage('num-decimal');
+      assert.equal(bundle.format(msg, args), '{$arg}');
+
+      msg = bundle.getMessage('num-percent');
+      assert.equal(bundle.format(msg, args), '{$arg}');
+
+      msg = bundle.getMessage('num-bad-opt');
+      assert.equal(bundle.format(msg, args), '{$arg}');
     });
   });
 
@@ -74,13 +103,13 @@ suite('Built-in functions', function() {
       let msg;
 
       msg = bundle.getMessage('dt-default');
-      assert.equal(bundle.format(msg), 'Invalid Date');
+      assert.equal(bundle.format(msg), '{$arg}');
 
       msg = bundle.getMessage('dt-month');
-      assert.equal(bundle.format(msg), 'Invalid Date');
+      assert.equal(bundle.format(msg), '{$arg}');
 
       msg = bundle.getMessage('dt-bad-opt');
-      assert.equal(bundle.format(msg), 'Invalid Date');
+      assert.equal(bundle.format(msg), '{$arg}');
     });
 
     test('Date argument', function () {
@@ -107,5 +136,47 @@ suite('Built-in functions', function() {
       //     Thu Sep 29 2016 02:00:00 GMT+0200 (CEST)
       assert.equal(bundle.format(msg, args), date.toString());
     });
+
+    test('number argument', function() {
+      let args = {arg: 1};
+      let msg;
+
+      msg = bundle.getMessage('dt-default');
+      assert.equal(bundle.format(msg, args), '{DATETIME()}');
+
+      msg = bundle.getMessage('dt-month');
+      assert.equal(bundle.format(msg, args), '{DATETIME()}');
+
+      msg = bundle.getMessage('dt-bad-opt');
+      assert.equal(bundle.format(msg, args), '{DATETIME()}');
+    });
+
+    test('string argument', function() {
+      let args = {arg: 'Foo'};
+      let msg;
+
+      msg = bundle.getMessage('dt-default');
+      assert.equal(bundle.format(msg, args), '{DATETIME()}');
+
+      msg = bundle.getMessage('dt-month');
+      assert.equal(bundle.format(msg, args), '{DATETIME()}');
+
+      msg = bundle.getMessage('dt-bad-opt');
+      assert.equal(bundle.format(msg, args), '{DATETIME()}');
+    });
+
+    test('invalid argument', function() {
+      let args = {arg: []};
+      let msg;
+
+      msg = bundle.getMessage('dt-default');
+      assert.equal(bundle.format(msg, args), '{$arg}');
+
+      msg = bundle.getMessage('dt-month');
+      assert.equal(bundle.format(msg, args), '{$arg}');
+
+      msg = bundle.getMessage('dt-bad-opt');
+      assert.equal(bundle.format(msg, args), '{$arg}');
+    });
   });
 });