openreplay/frontend/app/mstore/projectsStore.ts

import { makeAutoObservable, runInAction } from 'mobx';
import Project from './types/project';
import GDPR from './types/gdpr';
import { GLOBAL_HAS_NO_RECORDINGS, SITE_ID_STORAGE_KEY } from 'App/constants/storageKeys';
import { projectsService } from "App/services";

export default class ProjectsStore {
  list: Project[] = [];
  instance: Project | null = null;
  siteId: string | null = null;
  active: Project | null = null;
  sitesLoading = true;
  loading = false;

  constructor() {
    const storedSiteId = localStorage.getItem(SITE_ID_STORAGE_KEY);
    this.siteId = storedSiteId ?? null;
    makeAutoObservable(this);
  }

  get isMobile() {
    return this.instance ? ['ios', 'android'].includes(this.instance.platform) : false;
  }

  getSiteId = () => {
    return {
      siteId: this.siteId,
      active: this.active,
    };
  }

  initProject = (project: Partial<Project>) => {
    this.instance = new Project(project);
  }

  setSitesLoading = (loading: boolean) => {
    this.sitesLoading = loading;
  }

  setLoading = (loading: boolean) => {
    this.loading = loading;
  }

  setSiteId = (siteId: string) => {
    localStorage.setItem(SITE_ID_STORAGE_KEY, siteId.toString());
    this.siteId = siteId;
    this.active = this.list.find((site) => site.id! === siteId) ?? null;
  }

  editGDPR = (gdprData: Partial<GDPR>) => {
    if (this.instance) {
      this.instance.gdpr.edit(gdprData);
    }
  }

  editInstance = (instance: Partial<Project>) => {
    if (!this.instance) return;
    Object.keys(instance).forEach((key) => {
      if (key in this.instance) {
        this.instance[key] = instance[key];
      } else {
        console.error(`Project: Unknown key ${key}`);
      }
    })
  }

  fetchGDPR = async (siteId: string) => {
    try {
      const response = await projectsService.fetchGDPR(siteId)
      runInAction(() => {
        if (this.instance) {
          Object.assign(this.instance.gdpr, response.data);
        }
      });
    } catch (error) {
      console.error('Failed to fetch GDPR:', error);
    }
  }

  saveGDPR = async (siteId: string) => {
    if (!this.instance) return;
    try {
      const gdprData = this.instance.gdpr.toData();
      const response = await projectsService.saveGDPR(siteId, gdprData);
      this.editGDPR(response.data);
    } catch (error) {
      console.error('Failed to save GDPR:', error);
    }
  }

  fetchList = async (siteIdFromPath?: string) => {
    this.setSitesLoading(true);
    try {
      const response = await projectsService.fetchList();
      runInAction(() => {
        this.list = response.data.map((data) => new Project(data));
        const siteIds = this.list.map(site => site.id);
        let siteId = this.siteId;
        const siteExists = siteId ? siteIds.includes(siteId) : false;

        if (siteIdFromPath && siteIds.includes(siteIdFromPath)) {
          siteId = siteIdFromPath;
        } else if (!siteId || !siteExists) {
          siteId = siteIds.includes(this.siteId)
            ? this.siteId
            : response.data[0].projectId;
        }

        const hasRecordings = this.list.some(site => site.recorded);
        if (!hasRecordings) {
          localStorage.setItem(GLOBAL_HAS_NO_RECORDINGS, 'true');
        } else {
          localStorage.removeItem(GLOBAL_HAS_NO_RECORDINGS);
        }
        if (siteId) {
          this.setSiteId(siteId);
        }
      });
    } catch (error) {
      console.error('Failed to fetch site list:', error);
    } finally {
      this.setSitesLoading(false);
    }
  }

  save = async (projectData: Partial<Project>) => {
    this.setLoading(true);
    try {
      const response = await projectsService.saveProject(projectData);
      runInAction(() => {
        const newSite = new Project(response.data);
        const index = this.list.findIndex(site => site.id === newSite.id);
        if (index !== -1) {
          this.list[index] = newSite;
        } else {
          this.list.push(newSite);
        }
        this.setSiteId(newSite.id);
        this.active = newSite;
      });
    } catch (error) {
      console.error('Failed to save site:', error);
    } finally {
      this.setLoading(false);
    }
  }

  updateProjectRecordingStatus = (siteId: string, status: boolean) => {
    const site = this.list.find(site => site.id === siteId);
    if (site) {
      site.recorded = status;
      const hasRecordings = this.list.some(site => site.recorded);
      if (!hasRecordings) {
        localStorage.setItem(GLOBAL_HAS_NO_RECORDINGS, 'true');
      } else {
        localStorage.removeItem(GLOBAL_HAS_NO_RECORDINGS);
      }
    }
  }

  removeProject = async (projectId: string) => {
    this.setLoading(true);
    try {
      await projectsService.removeProject(projectId);
      runInAction(() => {
        this.list = this.list.filter(site => site.id !== projectId);
        if (this.siteId === projectId) {
          this.setSiteId(this.list[0].id!);
        }
      })
    } catch (e) {
      console.error('Failed to remove project:', e);
    } finally {
      this.setLoading(false);
    }
  }

  updateProject = async (projectId: string, projectData: Partial<Project>) => {
    this.setLoading(true);
    try {
      const response = await projectsService.updateProject(projectId, projectData);
      runInAction(() => {
        const updatedSite = new Project(response.data);
        const index = this.list.findIndex(site => site.id === updatedSite.id);
        if (index !== -1) {
          this.list[index] = updatedSite;
        }
      });
    } catch (error) {
      console.error('Failed to update site:', error);
    } finally {
      this.setLoading
    }
  }
}