Developers

import { jsonClient } from "ghostlyn/http";
import {
  dateRangeUtc,
  rollup,
  type SolidtimeProject,
  type SolidtimeClient,
} from "./summary.ts";

export const instructions = `solidtime tracks time entries for an organization, by person, client and
project. Call summary with a start and end date, inclusive, and an IANA
timezone, to get totals for that range. Name organization only when the
member belongs to more than one solidtime organization.`;

const client = jsonClient({ credential: "solidtime" });

type Membership = {
  id: string;
  role: string;
  organization: { id: string; name: string };
};

type PageMeta = { last_page?: number };

async function pageAll<T>(path: string): Promise<T[]> {
  const results: T[] = [];
  let page = 1;
  for (;;) {
    const sep = path.includes("?") ? "&" : "?";
    const res = await client("GET", `${path}${sep}page=${page}`);
    const data = (res.data as T[] | undefined) ?? [];
    results.push(...data);
    const meta = res.meta as PageMeta | undefined;
    const lastPage = meta?.last_page ?? 1;
    if (page >= lastPage) break;
    page++;
  }
  return results;
}

function pickMembership(
  memberships: Membership[],
  organization: string | undefined
): Membership {
  if (memberships.length === 0) {
    throw new Error("No solidtime organization membership found.");
  }
  const names = memberships.map(m => m.organization.name).join(", ");
  if (organization) {
    const match = memberships.find(
      m =>
        m.organization.name === organization ||
        m.organization.id === organization
    );
    if (!match) {
      throw new Error(
        `No solidtime organization named "${organization}". You belong to: ${names}.`
      );
    }
    return match;
  }
  if (memberships.length === 1) return memberships[0];
  throw new Error(
    `You belong to more than one solidtime organization. Name one: ${names}.`
  );
}

/**
 * Summarizes tracked solidtime hours by person, client within each person and
 * project within each client, plus team totals per client, for an inclusive
 * date range.
 *
 * @messageTemplate Summarizing solidtime hours
 * @readOnly
 */
export async function summary({
  start,
  end,
  timezone,
  organization,
}: {
  /** Inclusive start date, YYYY-MM-DD. */
  start: string;
  /** Inclusive end date, YYYY-MM-DD. */
  end: string;
  /** IANA timezone the range is local to, such as Europe/Helsinki. */
  timezone: string;
  /** Organization name or id, needed only with more than one membership. */
  organization?: string;
}) {
  const memberships = await client("GET", "/api/v1/users/me/memberships");
  const membership = pickMembership(
    (memberships.data as Membership[] | undefined) ?? [],
    organization
  );
  const orgId = membership.organization.id;
  const restrictedToSelf = membership.role === "employee";

  const range = dateRangeUtc(start, end, timezone);
  const params = new URLSearchParams({
    group: "user",
    sub_group: "project",
    start: range.start,
    end: range.end,
  });
  if (restrictedToSelf) params.set("member_id", membership.id);

  const [aggregate, projects, clients] = await Promise.all([
    client(
      "GET",
      `/api/v1/organizations/${orgId}/time-entries/aggregate?${params}`
    ),
    pageAll<SolidtimeProject>(`/api/v1/organizations/${orgId}/projects`),
    pageAll<SolidtimeClient>(`/api/v1/organizations/${orgId}/clients`),
  ]);

  const result = rollup(
    aggregate.data as Parameters<typeof rollup>[0],
    projects,
    clients
  );

  return {
    organization: membership.organization.name,
    start,
    end,
    restrictedToSelf,
    ...result,
  };
}