import { CliTerseError } from '@alwaysai/alwayscli';

export type ModelId = {
  publisher: string;
  name: string;
};

export const ModelId = {
  parse(id: string) {
    const errorMessage = `Expected model ID to be of the form "publisher/name"`;
    const splits = id.split('/');
    if (splits.length !== 2) {
      throw new CliTerseError(errorMessage);
    }
    for (const split of splits) {
      if (!split) {
        throw new CliTerseError(errorMessage);
      }
    }
    return {
      publisher: splits[0],
      name: splits[1]
    };
  },
  serialize({ publisher, name }: ModelId) {
    return `${publisher}/${name}`;
  }
};