// Copyright 2026 Synnax Labs, Inc.
//
// Use of this software is governed by the Business Source License included in the file
// licenses/BSL.txt.
//
// As of the Change Date specified in that file, in accordance with the Business Source
// License, use of this software will be governed by the Apache License, Version 2.0,
// included in the file licenses/APL.txt.

import { z } from "zod";

import { type Authority, type State, stateZ } from "@/control/types.gen";
import { type bounds } from "@/spatial";

export const ABSOLUTE_AUTHORITY: Authority = 255;
export const ZERO_AUTHORITY: Authority = 0;

export const AUTHORITY_BOUNDS: bounds.Bounds<Authority> = {
  lower: ZERO_AUTHORITY,
  upper: ABSOLUTE_AUTHORITY + 1,
};

export const filterTransfersByChannelKey =
  <R extends z.ZodType>(...resources: z.infer<R>[]) =>
  (transfers: Transfer<R>[]): Transfer<R>[] =>
    transfers.filter((t) => {
      let ok = false;
      if (t.to != null) ok = resources.includes(t.to.resource);
      if (t.from != null && !ok) ok = resources.includes(t.from.resource);
      return ok;
    });

interface Release<R extends z.ZodType> {
  from: State<R>;
  to?: null;
}

export const releaseZ = <R extends z.ZodType>(resource: R) =>
  z.object({
    from: stateZ(resource),
    to: z.null().optional(),
  });

interface Acquire<R extends z.ZodType> {
  from?: null;
  to: State<R>;
}

export const acquireZ = <R extends z.ZodType>(resource: R) =>
  z.object({
    from: z.null().optional(),
    to: stateZ(resource),
  });

export type Transfer<R extends z.ZodType> =
  | {
      from: State<R>;
      to: State<R>;
    }
  | Release<R>
  | Acquire<R>;

export const transferZ = <R extends z.ZodType>(resource: R) =>
  z.union([
    releaseZ(resource),
    acquireZ(resource),
    z.object({
      from: stateZ(resource),
      to: stateZ(resource),
    }),
  ]);