Nhogs/nestjs-neo4j

import { Neo4jModelService } from './neo4j.model.service';
import {
  neo4jSkipLimit,
  NODE,
  ORDER_BY,
  RETURN_PROPERTIES,
  SKIP_LIMIT,
  TIMESTAMP,
} from '../common';
import { Transaction } from 'neo4j-driver';
import { Query } from '../interface';

/**
 * Helper class to generate Node model service using Neo4j
 */
export abstract class Neo4jNodeModelService<T> extends Neo4jModelService<T> {
  create(
    properties: Partial<T>,
    options: { returns?: boolean } = {
      returns: true,
    },
  ) {
    const props = this.toNeo4j(properties);

    const CREATE = `CREATE ${NODE('n', this.label)}`;
    const SET = ` SET n=$props${
      props[this.timestamp] ? '' : TIMESTAMP('n', this.timestamp, ', ')
    }`;
    const RETURN = options?.returns ? RETURN_PROPERTIES('n', 'created') : ``;

    const query: Query<T> = {
      cypher: `${CREATE}${SET}${RETURN}`,
      parameters: { props },
    };

    return {
      query,
      runTx: (tx?: Transaction) => tx.run(query.cypher, query.parameters),
      run: async () => {
        const res = await this._run(query, { write: true });
        return res.map((r) => {
          return this.fromNeo4j(r.created);
        })[0];
      },
    };
  }

  merge(
    properties: Partial<T>,
    options: { returns?: boolean } = {
      returns: true,
    },
  ) {
    const props = this.toNeo4j(properties);

    const MERGE = `MERGE ${NODE('n', this.label, { props })}`;
    const SET = `${TIMESTAMP('n', this.timestamp, ' ON CREATE SET ')}`;
    const RETURN = options?.returns ? RETURN_PROPERTIES('n', 'merged') : ``;

    const query = {
      cypher: `${MERGE}${SET}${RETURN}`,
      parameters: { props },
    };

    return {
      query,
      runTx: (tx?: Transaction) => tx.run(query.cypher, query.parameters),
      run: async () => {
        const res = await this._run(query, { write: true });
        return res.map((r) => {
          return this.fromNeo4j(r.merged);
        });
      },
    };
  }

  update(
    match: Partial<T>,
    update: Partial<T>,
    options: { mutate?: boolean; returns?: boolean } = {
      mutate: true,
      returns: true,
    },
  ) {
    const props = this.toNeo4j(match);
    const updates = this.toNeo4j(update);

    const MATCH = `MATCH ${NODE('n', this.label, { props })}`;
    const SET = ` SET n ${options?.mutate ? '+' : ''}= $updates`;
    const RETURN = options?.returns ? RETURN_PROPERTIES('n', 'updated') : ``;

    const query = {
      cypher: `${MATCH}${SET}${RETURN}`,
      parameters: { props, updates },
    };

    return {
      query,
      runTx: (tx?: Transaction) => tx.run(query.cypher, query.parameters),
      run: async () => {
        const res = await this._run(query, { write: true });
        return res.map((r) => {
          return this.fromNeo4j(r.updated);
        });
      },
    };
  }

  delete(
    properties: Partial<T>,
    options: { returns?: boolean } = {
      returns: true,
    },
  ) {
    const props = this.toNeo4j(properties);

    const MATCH = `MATCH ${NODE('n', this.label, { props })}`;
    const WITH = options?.returns
      ? ` WITH n, properties(n) AS \`deleted\``
      : ``;
    const DELETE = ` DELETE n`;
    const RET = options?.returns ? ` RETURN \`deleted\`` : ``;

    const query = {
      cypher: `${MATCH}${WITH}${DELETE}${RET}`,
      parameters: { props },
    };

    return {
      query,
      runTx: (tx?: Transaction) => tx.run(query.cypher, query.parameters),
      run: async () => {
        const res = await this._run(query, { write: true });
        return res.map((r) => {
          return this.fromNeo4j(r.deleted);
        });
      },
    };
  }

  findAll(
    options: {
      skip?: number;
      limit?: number;
      orderBy?: string;
      descending?: boolean;
    } = {
      skip: 0,
      limit: 100,
    },
  ) {
    const MATCH = `MATCH ${NODE('n', this.label)}`;
    const RETURN = ` RETURN properties(n) AS \`matched\``;

    const query = {
      cypher: `${MATCH}${RETURN}${ORDER_BY(
        'n',
        options?.orderBy,
        options?.descending,
      )}${SKIP_LIMIT(options)}`,
      parameters: neo4jSkipLimit(options),
    };

    return {
      query,
      runTx: (tx?: Transaction) => tx.run(query.cypher, query.parameters),
      run: async () => {
        const res = await this._run(query);
        return res.map((r) => {
          return this.fromNeo4j(r.matched);
        });
      },
    };
  }

  findBy(
    properties: Partial<T>,
    options: {
      skip?: number;
      limit?: number;
      orderBy?: string;
      descending?: boolean;
    } = {
      skip: 0,
      limit: 100,
    },
  ) {
    const props = this.toNeo4j(properties);

    const MATCH = `MATCH ${NODE('n', this.label, { props })}`;
    const RETURN = ` RETURN properties(\`n\`) AS \`matched\``;

    const query = {
      cypher: `${MATCH}${RETURN}${ORDER_BY(
        'n',
        options?.orderBy,
        options?.descending,
      )}${SKIP_LIMIT(props)}`,
      parameters: { props, ...neo4jSkipLimit(options) },
    };

    return {
      query,
      runTx: (tx?: Transaction) => tx.run(query.cypher, query.parameters),
      run: async () => {
        const res = await this._run(query);
        return res.map((r) => {
          return this.fromNeo4j(r.matched);
        });
      },
    };
  }

  searchBy(
    prop: keyof T,
    search: string,
    options: {
      skip?: number;
      limit?: number;
    } = {
      skip: 0,
      limit: 100,
    },
  ) {
    const terms = search.replace(/\s+/g, ' ').trim().split(' ');

    const MATCH = `MATCH ${NODE('n', this.label)}`;
    const WITH = ` WITH n, split(n.\`${String(prop)}\`, ' ') as words`;

    const query = {
      cypher: `${MATCH}${WITH}
    WHERE ANY (term IN $terms WHERE ANY(word IN words WHERE word CONTAINS term))
    WITH n, words, 
    CASE WHEN apoc.text.join($terms, '') = apoc.text.join(words, '') THEN 100
    ELSE reduce(s = 0, st IN $terms | s + reduce(s2 = 0, w IN words | CASE WHEN (w = st) THEN (s2 + 4) ELSE CASE WHEN (w CONTAINS st) THEN (s2 +2) ELSE (s2) END END)) END AS score 
    ORDER BY score DESC${SKIP_LIMIT(
      options,
    )} RETURN properties(n) as \`matched\`, score`,
      parameters: {
        terms: terms,
        ...neo4jSkipLimit(options),
      },
    };

    return {
      query,
      runTx: (tx?: Transaction) => tx.run(query.cypher, query.parameters),
      run: async () => {
        const res = await this._run(query);
        return res.map(
          (r) => <[T, number]>[this.fromNeo4j(r.matched), Number(r.score)],
        );
      },
    };
  }
}