memory-orm/src/rule.ts

client side ORM + map reduce

import _get from 'lodash/get'
import _uniq from 'lodash/uniq'
import _property from 'lodash/property'
import _snakeCase from 'lodash/snakeCase'

import * as Mem from './userdata'
import { Model } from './model'
import { List } from './list'
import { Struct } from './struct'
import {
  Cache,
  DEPLOY,
  CLASS,
  SCHEMA,
  SCOPE,
  DEFAULT_RULE_TYPE,
  QUERY_WITH_SCOPE,
  ID,
  NameBase,
  PathCmd,
  SortCmd,
  OrderCmd,
  RelationCmd,
} from './type'
import { Set } from './set'
import { Map } from './map'
import { Query } from './query'
import { Finder } from './finder'
import { PureObject, State } from './mem'

function rename(base: string): NameBase {
  base = _snakeCase(base).replace(/s$/, '')
  const name = Mem.Name[base]
  if (name) {
    return name
  }

  const list = `${base}s`
  const o = (Mem.Name[list] = Mem.Name[base] = PureObject())
  o.base = base
  o.list = list
  o.id = `${base}_id`
  o.ids = `${base}_ids`
  o.relations = []
  o.deploys = []
  o.depends = []
  return o
}

function method({ prototype }: any, key: string, o: Object) {
  Object.defineProperty(prototype, key, o)
}

export class Rule<A extends DEFAULT_RULE_TYPE> {
  $name: NameBase
  state: Cache
  all: QUERY_WITH_SCOPE<A>

  model!: CLASS<A[0]>
  list!: CLASS<List<A>>
  set!: CLASS<Set<A>>
  map!: CLASS<Map<A>>

  constructor(
    base: string,
    {
      model = class model extends Model {} as any,
      list = class list extends List<A> {} as any,
      set = class set extends Set<A> {} as any,
      map = class map extends Map<A> {} as any,
      scope,
      scope_without_cache,
      schema,
      deploy,
    }: {
      model?: CLASS<A[0]>
      list?: CLASS<List<A>>
      set?: CLASS<Set<A>>
      map?: CLASS<Map<A>>
      scope?: SCOPE<A>
      scope_without_cache?: SCOPE<A>
      schema?: SCHEMA<A>
      deploy?: DEPLOY<A[0]>
    } = {}
  ) {
    this.$name = rename(base)
    this.state = State.base(this.$name.list)

    this.all = Query.build<A>(this.state) as Query<A>
    this.all.$sort['_reduce.list'] = {}
    this.all._cache = {}
    this.all._finder = new Finder<A>()

    this.depend_on(this.$name.list)

    this.model = model
    this.list = list
    this.set = set
    this.map = map

    if (scope_without_cache) {
      this.scope_without_cache(scope_without_cache)
    }
    if (scope) {
      this.scope(scope)
    }
    if (deploy) {
      this.deploy(deploy)
    }
    if (schema) {
      this.schema(schema)
    }
  }

  schema(cb: SCHEMA<A>) {
    cb.call(this)
    this.model.$name = this.list.$name = this.set.$name = this.map.$name = this.$name
    this.all._finder.join(this as any)

    Mem.Set[this.$name.base] = new this.set(this)
    Mem.Query[this.$name.list] = this.all
    Mem.Finder[this.$name.list] = this.all._finder
    return this
  }

  key_by(keys?: string | string[] | { (): string }) {
    const get = (() => {
      if (undefined === keys) {
        return function (this: any) {
          return this._id
        }
      }
      if (keys instanceof Function) {
        return keys
      }
      if (keys instanceof String) {
        return _property(keys)
      }
      if (keys instanceof Array) {
        return _property(keys)
      }
      throw new Error(`unimplemented ${keys}`)
    })()

    method(this.model, 'id', {
      enumerable: true,
      get,
    })
  }

  deploy(cb: DEPLOY<A[0]>) {
    this.$name.deploys.push(cb)
  }

  depend_on(parent) {
    Mem.Name[parent].depends.push(parent)
  }

  scope_without_cache(cb: SCOPE<A>) {
    const cmd = cb(this.all)
    for (const key in cmd) {
      ;(this.all as any)[key] = cmd[key]
    }
  }

  scope(cb: SCOPE<A>) {
    const cmd = cb(this.all)
    for (const key in cmd) {
      this.use_cache(key, cmd[key])
    }
  }

  property(type, o) {
    Object.defineProperties(this[type].prototype, o)
  }

  default_scope(scope: (all: Query<A>) => Query<A>) {
    this.all._copy(scope(this.all))
    const base = State.base(this.$name.list)
    base.$sort = this.all.$sort
  }

  shuffle() {
    this.default_scope((all) => all.shuffle())
  }

  sort(...sort: SortCmd) {
    this.default_scope((all) => all.sort(...sort))
  }

  order(keys: string | string[], order: OrderCmd) {
    this.default_scope((all) => (all as any).order(keys, order))
  }

  relation_to_one(key: string, target: string, ik: ID, else_id?: ID) {
    this.$name.relations.push(key)
    method(this.model, key, {
      enumerable: true,
      get() {
        const id = _get(this, ik)
        return Mem.Query[target].find(id, else_id!)
      },
    })
  }

  relation_to_many(key: string, target: string, ik: ID, cmd: string, qk: ID) {
    const { all } = this
    this.use_cache(key, (id) => Mem.Query[target].distinct(false)[cmd]({ [qk]: id }))

    this.$name.relations.push(key)
    method(this.model, key, {
      enumerable: true,
      get() {
        return all[key](this[ik])
      },
    })
  }

  relation_tree(key: string, ik: ID) {
    const { all } = this
    this.use_cache(key, (id: string, n: number) => {
      if (n) {
        const q = all.where({ [ik]: id })
        return all[key](q.ids, n - 1)
      } else {
        return all.where({ id })
      }
    })

    this.$name.relations.push(key)
    method(this.model, key, {
      enumerable: true,
      value(this: Model | Struct, n: number) {
        return all[key]([this.id], n)
      },
    })
  }

  relation_graph(key: string, ik: string) {
    const { all } = this
    this.use_cache(key, (id: string[], n: number) => {
      const q = all.where({ id })
      if (n) {
        const ids: any[] = []
        for (const a of q.pluck(ik) as List<any>) {
          if (a != null) {
            for (let k of a) {
              if (k != null) {
                ids.push(k)
              }
            }
          }
        }

        return all[key](_uniq(ids), n - 1)
      } else {
        return q
      }
    })

    this.$name.relations.push(key)
    method(this.model, key, {
      enumerable: true,
      value(this: Model | Struct, n: number) {
        return all[key]([this.id], n)
      },
    })
  }

  use_cache(key: string, val: any) {
    if (val instanceof Function) {
      ;(this.all as any)[key] = (...args: string[]) => {
        const name = `${key}:${JSON.stringify(args)}`
        return this.all._cache[name] || (this.all._cache[name] = val(...args))
      }
    } else {
      ;(this.all as any)[key] = val
    }
  }

  path(keys: string[], option: PathCmd = { key: 'id' }) {
    const { key } = option
    const { base, list } = this.$name
    let tail_key = keys[keys.length - 1]
    if ('*' === tail_key) {
      this.belongs_to(base)
      this.has_many(list)
      keys.pop()
      tail_key = keys[keys.length - 1]
    }

    for (const key of keys) {
      this.belongs_to(key)
    }

    this.deploy(({ o, reduce }) => {
      if ('string' !== typeof o[key]) {
        throw new Error(`id [${o[key]}] must be string.`)
      }
      const subids = o[key]!.split('-')
      o[`${key}x`] = subids[subids.length - 1]
      for (let idx = 0; idx < keys.length; idx++) {
        const sub_key = keys[idx]
        o[`${sub_key}_id`] = subids.slice(0, idx + 1).join('-')
      }

      // '*' support.
      if (base && keys.length + 1 < subids.length) {
        o[`${base}_id`] = subids.slice(0, -1).join('-')
      }

      reduce(key, { navi: subids })
    })

    const { all } = this
    const pk = `${tail_key}_id`
    method(this.model, 'siblings', {
      get() {
        return all.where({ [pk]: this[pk] })
      },
    })
  }

  belongs_to(to: string, option: RelationCmd = {}) {
    const name = rename(to)
    const { key = name.id, target = name.list, miss } = option
    this.relation_to_one(name.base, target, key, miss)
  }

  habtm(to, option: RelationCmd = {}) {
    const name = rename(to)
    if (option.reverse) {
      const { key = this.$name.ids, target = to } = option
      this.relation_to_many(name.list, target, 'id', 'in', key)
    } else {
      const { key = name.ids, target = name.list } = option
      this.relation_to_many(name.list, target, key, 'where', 'id')
    }
  }

  has_many(to, option: RelationCmd = {}) {
    const name = rename(to)
    const { key = this.$name.id, target = name.list } = option
    this.relation_to_many(name.list, target, 'id', 'where', key)
  }

  tree(option: RelationCmd = {}) {
    const fk = this.$name.id
    this.relation_tree('nodes', fk)
    this.belongs_to(this.$name.base, option)

    Object.defineProperties(this.all, {
      leaf: {
        get(this: Query<A>) {
          const not_leaf = _uniq(this.pluck(fk))
          return this.where((o) => !not_leaf.includes(o.id))
        },
      },
    })
  }

  graph(option: RelationCmd = {}) {
    const { directed, cost } = option
    const ik = this.$name.ids
    this.relation_to_many(this.$name.list, this.$name.list, ik, 'where', 'id')
    this.relation_graph('path', ik)
    if (!directed) {
      return true // todo
    }
    return false
  }
}