Add update instance

DefGuard · Sep 28, 2023 · 577a52e · 577a52e
1 parent 36f9f05
commit 577a52e
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 17 deletions.
diff --git a/src-tauri/src/main.rs b/src-tauri/src/main.rs
@@ -13,7 +13,7 @@ use tauri::{Manager, State};
 use tauri::SystemTrayEvent;
 mod tray;
 use crate::commands::{
-    all_instances, all_locations, connect, disconnect, location_stats, save_device_config,
+    all_instances, all_locations, connect, disconnect, location_stats, save_device_config, update_instance
 };
 use crate::tray::create_tray_menu;
 
@@ -36,6 +36,7 @@ fn main() {
             all_instances,
             connect,
             disconnect,
+            update_instance,
             location_stats,
         ])
         .on_window_event(|event| match event.event() {

diff --git a/src/pages/client/components/modals/AddInstanceModal/steps/AddInstanceInitStep.tsx b/src/pages/client/components/modals/AddInstanceModal/steps/AddInstanceInitStep.tsx
@@ -1,7 +1,7 @@
 import { zodResolver } from '@hookform/resolvers/zod';
 import { Body, fetch, Response } from '@tauri-apps/api/http';
+import { invoke } from '@tauri-apps/api/tauri';
 import dayjs from 'dayjs';
-import { isUndefined } from 'lodash-es';
 import { useMemo } from 'react';
 import { SubmitHandler, useForm } from 'react-hook-form';
 import { useNavigate } from 'react-router-dom';
@@ -61,9 +61,6 @@ export const AddInstanceModalInitStep = () => {
   const handleValidSubmit: SubmitHandler<FormFields> = async (values) => {
     const url = () => {
       const endpoint = '/api/v1/enrollment/start';
-      if (import.meta.env.DEV) {
-        return endpoint;
-      }
       let base: string;
       if (values.url[values.url.length - 1] === '/') {
         base = values.url.slice(0, -1);
@@ -75,7 +72,7 @@ export const AddInstanceModalInitStep = () => {
 
     const endpointUrl = url();
 
-    const headers = {
+    const headers: Record<string, string> = {
       'Content-Type': 'application/json',
     };
 
@@ -85,7 +82,7 @@ export const AddInstanceModalInitStep = () => {
 
     setModalState({ loading: true });
 
-    fetch(endpointUrl, {
+    fetch<EnrollmentStartResponse>(endpointUrl, {
       method: 'POST',
       headers,
       body: Body.json(data),
@@ -111,14 +108,29 @@ export const AddInstanceModalInitStep = () => {
         if (instance) {
           // update already registered instance instead
           headers['Cookie'] = authCookie;
-          const instanceInfo = await fetch(`${proxy_api_url}/enrollment/network_info`, {
+          fetch<CreateDeviceResponse>(`${proxy_api_url}/enrollment/network_info`, {
             method: 'POST',
             headers,
             body: Body.json({
               pubkey: instance.pubkey,
             }),
+          }).then(async (res) => {
+            console.log('here', res.data);
+            invoke<void>('update_instance', {
+              instanceId: instance.id,
+              response: res.data,
+            })
+              .then(() => {
+                toaster.success(
+                  LL.pages.enrollment.steps.deviceSetup.desktopSetup.messages.deviceConfigured(),
+                );
+                closeModal();
+              })
+              .catch((e) => {
+                console.error(e);
+                toaster.error(LL.pages.client.modals.addInstanceModal.messages.error());
+              });
           });
-          return;
         }
         // register new instance
         // is user in need of full enrollment ?

diff --git a/vite.config.ts b/vite.config.ts
@@ -10,14 +10,6 @@ export default defineConfig({
   server: {
     strictPort: true,
     port: 3000,
-    proxy: {
-      '/api': {
-        target: 'http://127.0.0.1:8080',
-        changeOrigin: true,
-        secure: false,
-        ws: true,
-      },
-    },
   },
   resolve: {
     alias: {