@plutonhq/core-frontend 0.0.1

package/src/services/backups.ts

@@ -0,0 +1,231 @@
+import { useMutation, useQuery, useQueryClient } from '@tanstack/react-query';
+import { toast } from 'react-toastify';
+import { API_URL } from '../utils/constants';
+
+// Generate Download
+export async function generateBackupDownload({ backupId }: { backupId: string; planId: string }) {
+   // const header = new Headers({ 'Content-Type': 'application/json', Accept: 'application/json' });
+   const res = await fetch(`${API_URL}/backups/${backupId}/action/download`, {
+      method: 'POST',
+      credentials: 'include',
+      // headers: header,
+   });
+   // Check if response is ok
+   const data = await res.json();
+   if (!data.success) {
+      throw new Error(data.error);
+   }
+   return data;
+}
+
+export function useDownloadBackup() {
+   const queryClient = useQueryClient();
+
+   return useMutation({
+      mutationFn: generateBackupDownload,
+      onSuccess: (res, payload) => {
+         console.log('res :', payload, res);
+         queryClient.invalidateQueries({ queryKey: ['plan', payload.planId] });
+      },
+   });
+}
+
+// Get Download
+export async function getBackupDownload(id: string) {
+   // const header = new Headers({ 'Content-Type': 'application/json', Accept: 'application/json' });
+   const res = await fetch(`${API_URL}/backups/${id}/action/download`, {
+      method: 'GET',
+      credentials: 'include',
+      // headers: header,
+   });
+   // Check if response is ok
+   if (!res.ok) {
+      const data = await res.json();
+      throw new Error(data.error);
+   }
+
+   const filename = res.headers.get('content-disposition')?.split('filename=')[1] || `backup-${id}.tar`;
+
+   // Use streams API
+   const reader = res.body?.getReader();
+   const stream = new ReadableStream({
+      async start(controller) {
+         while (true) {
+            const { done, value } = await reader!.read();
+            if (done) break;
+            controller.enqueue(value);
+         }
+         controller.close();
+         reader!.releaseLock();
+      },
+   });
+
+   // Create download from stream
+   const blob = await new Response(stream).blob();
+   const url = window.URL.createObjectURL(blob);
+   const link = document.createElement('a');
+   link.href = url;
+   link.download = filename;
+   document.body.appendChild(link);
+   link.click();
+   link.remove();
+   window.URL.revokeObjectURL(url);
+}
+
+export function useGetBackupDownload() {
+   return useMutation({
+      mutationFn: getBackupDownload,
+      onSuccess: (res) => {
+         console.log('# Backup Downloaded! :', res);
+      },
+      onError: (res) => {
+         console.log('# Download Failed! :', res);
+      },
+   });
+}
+
+// Cancel Download
+export async function cancelBackupDownload({ backupId }: { backupId: string; planId: string }) {
+   const res = await fetch(`${API_URL}/backups/${backupId}/action/download`, {
+      method: 'DELETE',
+      credentials: 'include',
+      // headers: header,
+   });
+   // Check if response is ok
+   const data = await res.json();
+   if (!data.success) {
+      throw new Error(data.error);
+   }
+   return data;
+}
+export function useCancelBackupDownload() {
+   const queryClient = useQueryClient();
+   return useMutation({
+      mutationFn: cancelBackupDownload,
+      onSuccess: (res, payload) => {
+         console.log('res :', payload, res);
+         queryClient.invalidateQueries({ queryKey: ['plan', payload.planId] });
+      },
+   });
+}
+
+// Remove Backup
+export async function deleteBackup(id: string) {
+   const header = new Headers({ 'Content-Type': 'application/json', Accept: 'application/json' });
+   const res = await fetch(`${API_URL}/backups/${id}`, {
+      method: 'DELETE',
+      credentials: 'include',
+      headers: header,
+   });
+   const data = await res.json();
+   if (!data.success) {
+      throw new Error(data.error);
+   }
+   return data;
+}
+
+export function useDeleteBackup() {
+   return useMutation({
+      mutationFn: deleteBackup,
+      onSuccess: (res) => {
+         console.log('# Backup Removed! :', res);
+      },
+   });
+}
+
+// Get Backup Progress
+export async function getBackupProgress({ id, sourceId, sourceType, planId }: { id: string; sourceId: string; sourceType: string; planId: string }) {
+   const url = new URL(`${API_URL}/backups/${id}/progress?sourceId=${sourceId}&sourceType=${sourceType}&planId=${planId}`);
+
+   const res = await fetch(url.toString(), {
+      method: 'GET',
+      credentials: 'include',
+   });
+   const data = await res.json();
+   return data;
+}
+
+export function useGetBackupProgress(payload: { id: string; sourceId: string; sourceType: string; planId: string }) {
+   const queryClient = useQueryClient();
+   return useQuery({
+      queryKey: ['progress-backup', payload.id],
+      queryFn: () => getBackupProgress(payload),
+      refetchOnMount: true,
+      retry: false,
+      refetchInterval(query) {
+         // console.log('query :', query.state?.data);
+         const progressData = query.state?.data;
+
+         // Check if backup is finished by looking for a "finished" phase event
+         const isFinished = progressData?.events?.some((event: any) => event.phase === 'finished' && event.completed === true);
+
+         if (isFinished) {
+            const planId = progressData?.planId || payload.planId;
+            if (planId) {
+               console.log('Invalidate Plan and Reload It :', planId);
+               queryClient.invalidateQueries({ queryKey: ['plan', planId] });
+            }
+            toast.success('Process Complete!');
+            return false;
+         }
+
+         return 1000;
+      },
+   });
+}
+
+export function useGetBackupProgressOnce(payload: { id: string; sourceId: string; sourceType: string; planId: string }) {
+   return useQuery({
+      queryKey: ['progress-backup', payload.id],
+      queryFn: () => getBackupProgress(payload),
+      refetchOnMount: true,
+      retry: false,
+   });
+}
+
+// Cancel Backup
+export async function cancelBackup({ planId, backupId }: { planId: string; backupId: string }) {
+   const res = await fetch(`${API_URL}/backups/${backupId}/action/cancel?planId=${planId}`, {
+      method: 'POST',
+      credentials: 'include',
+      // headers: header,
+   });
+   // Check if response is ok
+   const data = await res.json();
+   if (!data.success) {
+      throw new Error(data.error);
+   }
+   return data;
+}
+export function useCancelBackup() {
+   return useMutation({
+      mutationFn: cancelBackup,
+      onSuccess: (res, payload) => {
+         console.log('res :', payload, res);
+      },
+   });
+}
+
+// Get Snapshot Files
+export async function getSnapshotFiles({ backupId }: { backupId: string }) {
+   const res = await fetch(`${API_URL}/backups/${backupId}/files`, {
+      method: 'GET',
+      credentials: 'include',
+      // headers: header,
+   });
+   // Check if response is ok
+   const data = await res.json();
+   if (!data.success) {
+      throw new Error(data.error);
+   }
+   return data;
+}
+export function useGetSnapshotFiles(payload: { backupId: string }) {
+   return useQuery({
+      queryKey: ['snapshot-files', payload.backupId],
+      queryFn: () => getSnapshotFiles(payload),
+      refetchOnMount: true,
+      staleTime: 1000 * 60 * 60, // Cache the data for 60 minutes
+      retry: false,
+   });
+}

package/src/services/devices.ts

@@ -0,0 +1,161 @@
+import { useMutation, useQuery, useQueryClient } from '@tanstack/react-query';
+import { API_URL } from '../utils/constants';
+
+// Get System Metrics
+export async function getSystemMetrics(id: string) {
+   const url = new URL(`${API_URL}/devices/${id}/metrics`);
+   const res = await fetch(url.toString(), {
+      method: 'GET',
+      credentials: 'include',
+   });
+   const data = await res.json();
+   if (!data.success) {
+      throw new Error(data.error);
+   }
+   return data;
+}
+
+export function useGetSystemMetrics(id: string) {
+   return useQuery({
+      queryKey: ['device-metrics', id],
+      queryFn: () => getSystemMetrics(id),
+      refetchOnMount: true,
+      refetchOnWindowFocus: false,
+      retry: false,
+   });
+}
+
+// Get All Devices
+export async function getAllDevices() {
+   const url = new URL(`${API_URL}/devices`);
+
+   const res = await fetch(url.toString(), {
+      method: 'GET',
+      credentials: 'include',
+   });
+   const data = await res.json();
+   if (!data.success) {
+      throw new Error(data.error);
+   }
+   return data;
+}
+
+export function useGetDevices() {
+   return useQuery({
+      queryKey: ['devices'],
+      queryFn: () => getAllDevices(),
+      refetchOnMount: true,
+      retry: false,
+   });
+}
+
+// Get Single Device
+export async function getDevice(id: string, metrics: boolean) {
+   const url = new URL(`${API_URL}/devices/${id}?metrics=${metrics ? 'true' : 'false'}`);
+
+   const res = await fetch(url.toString(), {
+      method: 'GET',
+      credentials: 'include',
+   });
+   const data = await res.json();
+   if (!data.success) {
+      const error = new Error(data.error) as Error & { status?: number };
+      error.status = res.status;
+      throw error;
+   }
+   return data;
+}
+
+export function useGetDevice(id: string, metrics = false) {
+   return useQuery({
+      queryKey: ['device', id],
+      queryFn: () => getDevice(id, metrics),
+      refetchOnMount: true,
+      refetchOnWindowFocus: false,
+      retry: false,
+   });
+}
+
+// Update Device
+export async function updateDevice(updatePayload: { id: string; data: Record<string, any> }) {
+   const header = new Headers({ 'Content-Type': 'application/json', Accept: 'application/json' });
+   const res = await fetch(`${API_URL}/devices/${updatePayload.id}`, {
+      method: 'PUT',
+      credentials: 'include',
+      headers: header,
+      body: JSON.stringify(updatePayload.data),
+   });
+   const data = await res.json();
+   console.log('data :', data);
+   if (!data.success) {
+      throw new Error(data.error);
+   }
+   return data;
+}
+
+export function useUpdateDevice() {
+   const queryClient = useQueryClient();
+   return useMutation({
+      mutationFn: updateDevice,
+      onSuccess: (res) => {
+         // TODO: Should Display a Notification Bubble.
+         console.log('# Device Updated! :', res);
+         queryClient.invalidateQueries({ queryKey: ['devices'] });
+      },
+   });
+}
+
+// Update Dependent
+export async function updateDependent(payload: { id: string; dependent: string; newVersion: string }) {
+   const header = new Headers({ 'Content-Type': 'application/json', Accept: 'application/json' });
+   const res = await fetch(`${API_URL}/devices/${payload.id}/action/update/${payload.dependent}?version=${payload.newVersion}`, {
+      method: 'POST',
+      credentials: 'include',
+      headers: header,
+   });
+   const data = await res.json();
+   console.log('data :', data);
+   if (!data.success) {
+      throw new Error(data.error);
+   }
+   return data;
+}
+
+export function useUpdateDependent() {
+   const queryClient = useQueryClient();
+   return useMutation({
+      mutationFn: updateDependent,
+      onSuccess: (res) => {
+         // TODO: Should Display a Notification Bubble.
+         console.log('# Device Updated! :', res);
+         queryClient.invalidateQueries({ queryKey: ['devices'] });
+      },
+   });
+}
+
+// Browse Directory
+export async function browseDir(payload: { deviceId: string; path?: string }) {
+   const url = new URL(`${API_URL}/devices/${payload.deviceId}/browse`);
+   if (payload.path) {
+      url.searchParams.append('path', encodeURIComponent(payload.path));
+   }
+
+   const res = await fetch(url.toString(), {
+      method: 'GET',
+      credentials: 'include',
+   });
+   const data = await res.json();
+   if (!data.success) {
+      throw new Error(data.error);
+   }
+   return data;
+}
+
+export function useBrowseDir(payload: { deviceId: string; path?: string }) {
+   return useQuery({
+      queryKey: ['dirPath', payload.deviceId, payload.path],
+      queryFn: () => browseDir(payload),
+      refetchOnMount: true,
+      retry: false,
+   });
+}

package/src/services/index.ts

@@ -0,0 +1,11 @@
+/**
+ * Services exports
+ */
+
+export * from './backups';
+export * from './devices';
+export * from './plans';
+export * from './restores';
+export * from './settings';
+export * from './storage';
+export * from './users';