Merge pull request #85 from SquirrelCorporation/feat-debug-settings

Add feature to delete and resync playbooks
SquirrelCorporation · Jul 5, 2024 · 6943c56 · 6943c56
2 parents 0da8307 + 77841cb
commit 6943c56
Show file tree

Hide file tree

Showing 5 changed files with 59 additions and 6 deletions.
diff --git a/client/src/components/Template/Title.tsx b/client/src/components/Template/Title.tsx
@@ -20,6 +20,7 @@ export enum SettingsSubTitleColors {
   SERVER = '#8e7d50',
   GIT = '#336048',
   LOCAL = '#4b4a4a',
+  DEBUG = '#b614b6',
 }
 
 export type PageContainerTitleProps = {

diff --git a/client/src/pages/Admin/Settings/components/AdvancedSettings.tsx b/client/src/pages/Admin/Settings/components/AdvancedSettings.tsx
@@ -8,14 +8,11 @@ import {
   deleteAnsibleLogs,
   deleteContainerStats,
   deleteDeviceStats,
+  deletePlaybooksAndResync,
   deleteServerLogs,
   postRestartServer,
 } from '@/services/rest/settings';
-import {
-  InfoCircleFilled,
-  RedoOutlined,
-  TableOutlined,
-} from '@ant-design/icons';
+import { BugFilled, InfoCircleFilled, RedoOutlined } from '@ant-design/icons';
 import { Button, Card, Flex, Popover, Space, Typography } from 'antd';
 import { DeleteOutline } from 'antd-mobile-icons';
 import React from 'react';
@@ -161,6 +158,37 @@ const AdvancedSettings: React.FC = () => {
           </Space>
         </Flex>
       </Card>
+      <Card
+        type="inner"
+        style={{ marginTop: 16 }}
+        title={
+          <Title.SubTitle
+            title={'Debug'}
+            backgroundColor={SettingsSubTitleColors.DEBUG}
+            icon={<BugFilled />}
+          />
+        }
+      >
+        <Flex vertical gap={32} style={{ width: '50%' }}>
+          <Space direction="horizontal" size="middle">
+            <Typography>
+              {' '}
+              <Popover content={'Delete all playbooks and resync'}>
+                <InfoCircleFilled />
+              </Popover>{' '}
+              Playbooks
+            </Typography>{' '}
+            <Button
+              danger
+              icon={<DeleteOutline />}
+              onClick={async () => await deletePlaybooksAndResync()}
+            >
+              Purge & Sync again
+            </Button>
+            <Space.Compact style={{ width: '100%' }} />
+          </Space>
+        </Flex>
+      </Card>
     </Card>
   );
 };

diff --git a/client/src/services/rest/settings.ts b/client/src/services/rest/settings.ts
@@ -88,6 +88,17 @@ export async function deleteDeviceStats(options?: Record<string, any>) {
   });
 }
 
+export async function deletePlaybooksAndResync(options?: Record<string, any>) {
+  return request<API.SimpleResult>(
+    `/api/settings/advanced/playbooks-and-resync`,
+    {
+      method: 'DELETE',
+      ...{},
+      ...(options || {}),
+    },
+  );
+}
+
 export async function postContainerStatsSettings(
   type: string,
   value: number,

diff --git a/server/src/routes/settings.ts b/server/src/routes/settings.ts
@@ -15,6 +15,7 @@ import {
   deleteContainerStats,
   deleteDeviceStats,
   deleteLogs,
+  deletePlaybooksModelAndResync,
   postRestartServer,
 } from '../services/settings/advanced';
 
@@ -36,5 +37,5 @@ router.delete('/advanced/logs', deleteLogs);
 router.delete('/advanced/ansible-logs', deleteAnsibleLogs);
 router.delete('/advanced/device-stats', deleteDeviceStats);
 router.delete('/advanced/container-stats', deleteContainerStats);
-
+router.delete('/advanced/playbooks-and-resync', deletePlaybooksModelAndResync);
 export default router;
diff --git a/server/src/services/settings/advanced.ts b/server/src/services/settings/advanced.ts
@@ -1,10 +1,15 @@
 import { SuccessResponse } from '../../core/api/ApiResponse';
+import {
+  COLLECTION_NAME as PlaybookCollectionName,
+  PlaybookModel,
+} from '../../data/database/model/Playbook';
 import AnsibleLogsRepo from '../../data/database/repository/AnsibleLogsRepo';
 import ContainerStatsRepo from '../../data/database/repository/ContainerStatsRepo';
 import DeviceStatRepo from '../../data/database/repository/DeviceStatRepo';
 import LogsRepo from '../../data/database/repository/LogsRepo';
 import asyncHandler from '../../helpers/AsyncHandler';
 import { restart } from '../../index';
+import PlaybooksRepositoryEngine from '../../integrations/playbooks-repository/PlaybooksRepositoryEngine';
 
 export const postRestartServer = asyncHandler(async (req, res) => {
   await restart();
@@ -29,3 +34,10 @@ export const deleteDeviceStats = asyncHandler(async (req, res) => {
   await DeviceStatRepo.deleteAll();
   new SuccessResponse('Device stats Purged Successfully').send(res);
 });
+
+export const deletePlaybooksModelAndResync = asyncHandler(async (req, res) => {
+  await PlaybookModel.db.collection(PlaybookCollectionName).drop();
+  await PlaybooksRepositoryEngine.registerRepositories();
+  await PlaybooksRepositoryEngine.syncAllRegistered();
+  new SuccessResponse('All data purged successfully').send(res);
+});