about summary refs log tree commit diff stats
path: root/js
diff options
context:
space:
mode:
Diffstat (limited to 'js')
-rw-r--r--js/functional.js462
1 files changed, 462 insertions, 0 deletions
diff --git a/js/functional.js b/js/functional.js
new file mode 100644
index 0000000..ec5a277
--- /dev/null
+++ b/js/functional.js
@@ -0,0 +1,462 @@
+// probably best not to *actually* use these, and instead check out
+//
+// https://folktale.origamitower.com/
+// https://ramdajs.com/
+// https://github.com/fantasyland
+
+// always :: a -> b -> a
+const always = curry((a, b) => a);
+
+// compose :: ((y -> z), (x -> y),  ..., (a -> b)) -> a -> z
+const compose = (...fns) => (...args) => fns.reduceRight((res, fn) => [fn.call(null, ...res)], args)[0];
+
+// curry :: ((a, b, ...) -> c) -> a -> b -> ... -> c
+function curry(fn) {
+  const arity = fn.length;
+
+  return function $curry(...args) {
+    if (args.length < arity) {
+      return $curry.bind(null, ...args);
+    }
+
+    return fn.call(null, ...args);
+  };
+}
+
+// either :: (a -> c) -> (b -> c) -> Either a b -> c
+const either = curry((f, g, e) => {
+  if (e.isLeft) {
+    return f(e.$value);
+  }
+
+  return g(e.$value);
+});
+
+// identity :: x -> x
+const identity = x => x;
+
+// inspect :: a -> String
+const inspect = (x) => {
+  if (x && typeof x.inspect === 'function') {
+    return x.inspect();
+  }
+
+  function inspectFn(f) {
+    return f.name ? f.name : f.toString();
+  }
+
+  function inspectTerm(t) {
+    switch (typeof t) {
+      case 'string':
+        return `'${t}'`;
+      case 'object': {
+        const ts = Object.keys(t).map(k => [k, inspect(t[k])]);
+        return `{${ts.map(kv => kv.join(': ')).join(', ')}}`;
+      }
+      default:
+        return String(t);
+    }
+  }
+
+  function inspectArgs(args) {
+    return Array.isArray(args) ? `[${args.map(inspect).join(', ')}]` : inspectTerm(args);
+  }
+
+  return (typeof x === 'function') ? inspectFn(x) : inspectArgs(x);
+};
+
+// left :: a -> Either a b
+const left = a => new Left(a);
+
+// liftA2 :: (Applicative f) => (a1 -> a2 -> b) -> f a1 -> f a2 -> f b
+const liftA2 = curry((fn, a1, a2) => a1.map(fn).ap(a2));
+
+// liftA3 :: (Applicative f) => (a1 -> a2 -> a3 -> b) -> f a1 -> f a2 -> f a3 -> f b
+const liftA3 = curry((fn, a1, a2, a3) => a1.map(fn).ap(a2).ap(a3));
+
+// maybe :: b -> (a -> b) -> Maybe a -> b
+const maybe = curry((v, f, m) => {
+  if (m.isNothing) {
+    return v;
+  }
+
+  return f(m.$value);
+});
+
+// nothing :: Maybe a
+const nothing = Maybe.of(null);
+
+// reject :: a -> Task a b
+const reject = a => Task.rejected(a);
+
+
+const createCompose = curry((F, G) => class Compose {
+  constructor(x) {
+    this.$value = x;
+  }
+
+  [util.inspect.custom]() {
+    return `Compose(${inspect(this.$value)})`;
+  }
+
+  // ----- Pointed (Compose F G)
+  static of(x) {
+    return new Compose(F(G(x)));
+  }
+
+  // ----- Functor (Compose F G)
+  map(fn) {
+    return new Compose(this.$value.map(x => x.map(fn)));
+  }
+
+  // ----- Applicative (Compose F G)
+  ap(f) {
+    return f.map(this.$value);
+  }
+});
+
+class Either {
+  constructor(x) {
+    this.$value = x;
+  }
+
+  // ----- Pointed (Either a)
+  static of(x) {
+    return new Right(x);
+  }
+}
+
+class Left extends Either {
+  get isLeft() {
+    return true;
+  }
+
+  get isRight() {
+    return false;
+  }
+
+  static of(x) {
+    throw new Error('`of` called on class Left (value) instead of Either (type)');
+  }
+
+  [util.inspect.custom]() {
+    return `Left(${inspect(this.$value)})`;
+  }
+
+  // ----- Functor (Either a)
+  map() {
+    return this;
+  }
+
+  // ----- Applicative (Either a)
+  ap() {
+    return this;
+  }
+
+  // ----- Monad (Either a)
+  chain() {
+    return this;
+  }
+
+  join() {
+    return this;
+  }
+
+  // ----- Traversable (Either a)
+  sequence(of) {
+    return of(this);
+  }
+
+  traverse(of, fn) {
+    return of(this);
+  }
+}
+
+class Right extends Either {
+  get isLeft() {
+    return false;
+  }
+
+  get isRight() {
+    return true;
+  }
+
+  static of(x) {
+    throw new Error('`of` called on class Right (value) instead of Either (type)');
+  }
+
+  [util.inspect.custom]() {
+    return `Right(${inspect(this.$value)})`;
+  }
+
+  // ----- Functor (Either a)
+  map(fn) {
+    return Either.of(fn(this.$value));
+  }
+
+  // ----- Applicative (Either a)
+  ap(f) {
+    return f.map(this.$value);
+  }
+
+  // ----- Monad (Either a)
+  chain(fn) {
+    return fn(this.$value);
+  }
+
+  join() {
+    return this.$value;
+  }
+
+  // ----- Traversable (Either a)
+  sequence(of) {
+    return this.traverse(of, identity);
+  }
+
+  traverse(of, fn) {
+    fn(this.$value).map(Either.of);
+  }
+}
+
+class Identity {
+  constructor(x) {
+    this.$value = x;
+  }
+
+  [util.inspect.custom]() {
+    return `Identity(${inspect(this.$value)})`;
+  }
+
+  // ----- Pointed Identity
+  static of(x) {
+    return new Identity(x);
+  }
+
+  // ----- Functor Identity
+  map(fn) {
+    return Identity.of(fn(this.$value));
+  }
+
+  // ----- Applicative Identity
+  ap(f) {
+    return f.map(this.$value);
+  }
+
+  // ----- Monad Identity
+  chain(fn) {
+    return this.map(fn).join();
+  }
+
+  join() {
+    return this.$value;
+  }
+
+  // ----- Traversable Identity
+  sequence(of) {
+    return this.traverse(of, identity);
+  }
+
+  traverse(of, fn) {
+    return fn(this.$value).map(Identity.of);
+  }
+}
+
+class IO {
+  constructor(fn) {
+    this.unsafePerformIO = fn;
+  }
+
+  [util.inspect.custom]() {
+    return 'IO(?)';
+  }
+
+  // ----- Pointed IO
+  static of(x) {
+    return new IO(() => x);
+  }
+
+  // ----- Functor IO
+  map(fn) {
+    return new IO(compose(fn, this.unsafePerformIO));
+  }
+
+  // ----- Applicative IO
+  ap(f) {
+    return this.chain(fn => f.map(fn));
+  }
+
+  // ----- Monad IO
+  chain(fn) {
+    return this.map(fn).join();
+  }
+
+  join() {
+    return new IO(() => this.unsafePerformIO().unsafePerformIO());
+  }
+}
+
+class List {
+  constructor(xs) {
+    this.$value = xs;
+  }
+
+  [util.inspect.custom]() {
+    return `List(${inspect(this.$value)})`;
+  }
+
+  concat(x) {
+    return new List(this.$value.concat(x));
+  }
+
+  // ----- Pointed List
+  static of(x) {
+    return new List([x]);
+  }
+
+  // ----- Functor List
+  map(fn) {
+    return new List(this.$value.map(fn));
+  }
+
+  // ----- Traversable List
+  sequence(of) {
+    return this.traverse(of, identity);
+  }
+
+  traverse(of, fn) {
+    return this.$value.reduce(
+      (f, a) => fn(a).map(b => bs => bs.concat(b)).ap(f),
+      of(new List([])),
+    );
+  }
+}
+
+class Map {
+  constructor(x) {
+    this.$value = x;
+  }
+
+  [util.inspect.custom]() {
+    return `Map(${inspect(this.$value)})`;
+  }
+
+  insert(k, v) {
+    const singleton = {};
+    singleton[k] = v;
+    return Map.of(Object.assign({}, this.$value, singleton));
+  }
+
+  reduceWithKeys(fn, zero) {
+    return Object.keys(this.$value)
+      .reduce((acc, k) => fn(acc, this.$value[k], k), zero);
+  }
+
+  // ----- Functor (Map a)
+  map(fn) {
+    return this.reduceWithKeys(
+      (m, v, k) => m.insert(k, fn(v)),
+      new Map({}),
+    );
+  }
+
+  // ----- Traversable (Map a)
+  sequence(of) {
+    return this.traverse(of, identity);
+  }
+
+  traverse(of, fn) {
+    return this.reduceWithKeys(
+      (f, a, k) => fn(a).map(b => m => m.insert(k, b)).ap(f),
+      of(new Map({})),
+    );
+  }
+}
+
+class Maybe {
+  get isNothing() {
+    return this.$value === null || this.$value === undefined;
+  }
+
+  get isJust() {
+    return !this.isNothing;
+  }
+
+  constructor(x) {
+    this.$value = x;
+  }
+
+  [util.inspect.custom]() {
+    return this.isNothing ? 'Nothing' : `Just(${inspect(this.$value)})`;
+  }
+
+  // ----- Pointed Maybe
+  static of(x) {
+    return new Maybe(x);
+  }
+
+  // ----- Functor Maybe
+  map(fn) {
+    return this.isNothing ? this : Maybe.of(fn(this.$value));
+  }
+
+  // ----- Applicative Maybe
+  ap(f) {
+    return this.isNothing ? this : f.map(this.$value);
+  }
+
+  // ----- Monad Maybe
+  chain(fn) {
+    return this.map(fn).join();
+  }
+
+  join() {
+    return this.isNothing ? this : this.$value;
+  }
+
+  // ----- Traversable Maybe
+  sequence(of) {
+    return this.traverse(of, identity);
+  }
+
+  traverse(of, fn) {
+    return this.isNothing ? of(this) : fn(this.$value).map(Maybe.of);
+  }
+}
+
+class Task {
+  constructor(fork) {
+    this.fork = fork;
+  }
+
+  [util.inspect.custom]() {
+    return 'Task(?)';
+  }
+
+  static rejected(x) {
+    return new Task((reject, _) => reject(x));
+  }
+
+  // ----- Pointed (Task a)
+  static of(x) {
+    return new Task((_, resolve) => resolve(x));
+  }
+
+  // ----- Functor (Task a)
+  map(fn) {
+    return new Task((reject, resolve) => this.fork(reject, compose(resolve, fn)));
+  }
+
+  // ----- Applicative (Task a)
+  ap(f) {
+    return this.chain(fn => f.map(fn));
+  }
+
+  // ----- Monad (Task a)
+  chain(fn) {
+    return new Task((reject, resolve) => this.fork(reject, x => fn(x).fork(reject, resolve)));
+  }
+
+  join() {
+    return this.chain(identity);
+  }
+}
\ No newline at end of file