Modrinth/apps/frontend/src/store/moderation.ts

import { defineStore, createPinia } from "pinia";
import piniaPluginPersistedstate from "pinia-plugin-persistedstate";

export interface ModerationQueue {
  items: string[];
  total: number;
  completed: number;
  skipped: number;
  lastUpdated: Date;
}

const EMPTY_QUEUE: Partial<ModerationQueue> = {
  items: [],

  // TODO: Consider some form of displaying this in the checklist, maybe at the end
  total: 0,
  completed: 0,
  skipped: 0,
};

function createEmptyQueue(): ModerationQueue {
  return { ...EMPTY_QUEUE, lastUpdated: new Date() } as ModerationQueue;
}

const pinia = createPinia();
pinia.use(piniaPluginPersistedstate);

export const useModerationStore = defineStore("moderation", {
  state: () => ({
    currentQueue: createEmptyQueue(),
  }),

  getters: {
    queueLength: (state) => state.currentQueue.items.length,
    hasItems: (state) => state.currentQueue.items.length > 0,
    progress: (state) => {
      if (state.currentQueue.total === 0) return 0;
      return (state.currentQueue.completed + state.currentQueue.skipped) / state.currentQueue.total;
    },
  },

  actions: {
    setQueue(projectIDs: string[]) {
      this.currentQueue = {
        items: [...projectIDs],
        total: projectIDs.length,
        completed: 0,
        skipped: 0,
        lastUpdated: new Date(),
      };
    },

    setSingleProject(projectId: string) {
      this.currentQueue = {
        items: [projectId],
        total: 1,
        completed: 0,
        skipped: 0,
        lastUpdated: new Date(),
      };
    },

    completeCurrentProject(projectId: string, status: "completed" | "skipped" = "completed") {
      if (status === "completed") {
        this.currentQueue.completed++;
      } else {
        this.currentQueue.skipped++;
      }

      this.currentQueue.items = this.currentQueue.items.filter((id: string) => id !== projectId);
      this.currentQueue.lastUpdated = new Date();

      return this.currentQueue.items.length > 0;
    },

    getCurrentProjectId(): string | null {
      return this.currentQueue.items[0] || null;
    },

    resetQueue() {
      this.currentQueue = createEmptyQueue();
    },
  },

  persist: {
    key: "moderation-store",
    serializer: {
      serialize: JSON.stringify,
      deserialize: (value: string) => {
        const parsed = JSON.parse(value);
        if (parsed.currentQueue?.lastUpdated) {
          parsed.currentQueue.lastUpdated = new Date(parsed.currentQueue.lastUpdated);
        }
        return parsed;
      },
    },
  },
});