import { getDocumentType, RelationshipType, type AnyDocument } from "./types";

export function displayName(relationshipType: RelationshipType) {
  return relationshipType.charAt(0).toUpperCase() + relationshipType.slice(1);
}

export function relationshipsForDocument(doc: AnyDocument): RelationshipType[] {
  switch (getDocumentType(doc)) {
    case "session":
      return [
        RelationshipType.Scenes,
        RelationshipType.Secrets,
        RelationshipType.Locations,
        RelationshipType.Npcs,
        RelationshipType.Monsters,
        RelationshipType.Treasures,
      ];
    default:
      return [];
  }
}