books/frappe/model/naming.ts

import frappe from 'frappe';
import NumberSeries from 'frappe/models/NumberSeries';
import { getRandomString } from 'frappe/utils';
import { BaseError } from 'frappe/utils/errors';
import { Field, Schema } from 'schemas/types';
import Doc from './doc';

export function getNumberSeries(schema: Schema): Field | undefined {
  const numberSeries = schema.fields.find(
    (f) => f.fieldname === 'numberSeries'
  );
  return numberSeries;
}

export function isNameAutoSet(schemaName: string): boolean {
  const schema = frappe.schemaMap[schemaName]!;
  if (schema.naming === 'autoincrement') {
    return true;
  }

  const numberSeries = getNumberSeries(schema);
  if (numberSeries) {
    return true;
  }

  return false;
}

export async function setName(doc: Doc) {
  // if is server, always name again if autoincrement or other
  if (doc.schema.naming === 'autoincrement') {
    doc.name = await getNextId(doc.schemaName);
    return;
  }

  // Current, per doc number series
  const numberSeries = doc.numberSeries as string | undefined;
  if (numberSeries !== undefined) {
    doc.name = await getSeriesNext(numberSeries, doc.schemaName);
    return;
  }

  if (doc.name) {
    return;
  }

  // name === doctype for Single
  if (doc.schema.isSingle) {
    doc.name = doc.schema.name;
    return;
  }

  // assign a random name by default
  // override doc to set a name
  if (!doc.name) {
    doc.name = getRandomString();
  }
}

export async function getNextId(schemaName: string) {
  // get the last inserted row
  const lastInserted = await getLastInserted(schemaName);
  let name = 1;
  if (lastInserted) {
    let lastNumber = parseInt(lastInserted.name as string);
    if (isNaN(lastNumber)) lastNumber = 0;
    name = lastNumber + 1;
  }
  return (name + '').padStart(9, '0');
}

export async function getLastInserted(schemaName: string) {
  const lastInserted = await frappe.db.getAll(schemaName, {
    fields: ['name'],
    limit: 1,
    orderBy: 'creation',
    order: 'desc',
  });
  return lastInserted && lastInserted.length ? lastInserted[0] : null;
}

export async function getSeriesNext(prefix: string, schemaName: string) {
  let series: NumberSeries;

  try {
    series = (await frappe.doc.getDoc('NumberSeries', prefix)) as NumberSeries;
  } catch (e) {
    const { statusCode } = e as BaseError;
    if (!statusCode || statusCode !== 404) {
      throw e;
    }

    await createNumberSeries(prefix, schemaName);
    series = (await frappe.doc.getDoc('NumberSeries', prefix)) as NumberSeries;
  }

  return await series.next(schemaName);
}

export async function createNumberSeries(
  prefix: string,
  referenceType: string,
  start = 1001
) {
  const exists = await frappe.db.exists('NumberSeries', prefix);
  if (exists) {
    return;
  }

  const series = frappe.doc.getNewDoc('NumberSeries', {
    name: prefix,
    start,
    referenceType,
  });

  await series.insert();
}