feat: API for Tags

Author: svc-shorpo

.changeset/six-lemons-compete.md

@@ -0,0 +1,5 @@
+---
+'@hyperdx/api': patch
+---
+
+Add tags to Dashboards and LogViews

packages/api/src/models/dashboard.ts

@@ -78,6 +78,7 @@ export interface IDashboard {
   query: string;
   team: ObjectId;
   charts: Chart[];
+  tags: string[];
 }
 
 const DashboardSchema = new Schema<IDashboard>(
@@ -89,6 +90,10 @@ const DashboardSchema = new Schema<IDashboard>(
     query: String,
     team: { type: mongoose.Schema.Types.ObjectId, ref: 'Team' },
     charts: { type: mongoose.Schema.Types.Mixed, required: true },
+    tags: {
+      type: [String],
+      default: [],
+    },
   },
   {
     timestamps: true,

packages/api/src/models/logView.ts

@@ -8,6 +8,7 @@ export interface ILogView {
   name: string;
   query: string;
   team: ObjectId;
+  tags: string[];
 }
 
 const LogViewSchema = new Schema<ILogView>(
@@ -22,6 +23,10 @@ const LogViewSchema = new Schema<ILogView>(
     },
     team: { type: mongoose.Schema.Types.ObjectId, ref: 'Team' },
     creator: { type: mongoose.Schema.Types.ObjectId, ref: 'User' },
+    tags: {
+      type: [String],
+      default: [],
+    },
   },
   {
     timestamps: true,

packages/api/src/routers/api/__tests__/team.test.ts

@@ -45,4 +45,38 @@ Object {
 }
 `);
   });
+
+  it('GET /team/tags - no tags', async () => {
+    const { agent } = await getLoggedInAgent(server);
+
+    const resp = await agent.get('/team/tags').expect(200);
+
+    expect(resp.body.data).toMatchInlineSnapshot(`Array []`);
+  });
+
+  it('GET /team/tags', async () => {
+    const { agent } = await getLoggedInAgent(server);
+    await agent
+      .post('/dashboards')
+      .send({
+        name: 'Test',
+        charts: [],
+        query: '',
+        tags: ['test'],
+      })
+      .expect(200);
+    await agent
+      .post('/log-views')
+      .send({
+        name: 'Test',
+        query: '',
+        tags: ['test2'],
+      })
+      .expect(200);
+    const resp = await agent.get('/team/tags').expect(200);
+    expect(resp.body.data).toMatchInlineSnapshot(`Array [
+  "test",
+  "test2",
+    ]`);
+  });
 });

packages/api/src/routers/api/dashboards.ts

@@ -97,6 +97,7 @@ router.post(
       name: z.string(),
       charts: z.array(zChart),
       query: z.string(),
+      tags: z.array(z.string()).optional(),
     }),
   }),
   async (req, res, next) => {
@@ -106,12 +107,13 @@ router.post(
         return res.sendStatus(403);
       }
 
-      const { name, charts, query } = req.body ?? {};
+      const { name, charts, query, tags } = req.body ?? {};
       // Create new dashboard from name and charts
       const newDashboard = await new Dashboard({
         name,
         charts,
         query,
+        tags,
         team: teamId,
       }).save();
 
@@ -131,6 +133,7 @@ router.put(
       name: z.string(),
       charts: z.array(zChart),
       query: z.string(),
+      tags: z.array(z.string()).optional(),
     }),
   }),
   async (req, res, next) => {
@@ -144,7 +147,7 @@ router.put(
         return res.sendStatus(400);
       }
 
-      const { name, charts, query } = req.body ?? {};
+      const { name, charts, query, tags } = req.body ?? {};
       // Update dashboard from name and charts
       const oldDashboard = await Dashboard.findById(dashboardId);
       const updatedDashboard = await Dashboard.findByIdAndUpdate(
@@ -153,6 +156,7 @@ router.put(
           name,
           charts,
           query,
+          tags,
         },
         { new: true },
       );

packages/api/src/routers/api/logViews.ts

@@ -9,7 +9,7 @@ router.post('/', async (req, res, next) => {
   try {
     const teamId = req.user?.team;
     const userId = req.user?._id;
-    const { query, name } = req.body;
+    const { query, name, tags } = req.body;
     if (teamId == null) {
       return res.sendStatus(403);
     }
@@ -18,6 +18,7 @@ router.post('/', async (req, res, next) => {
     }
     const logView = await new LogView({
       name,
+      tags,
       query: `${query}`,
       team: teamId,
       creator: userId,
@@ -64,7 +65,7 @@ router.patch('/:id', async (req, res, next) => {
   try {
     const teamId = req.user?.team;
     const { id: logViewId } = req.params;
-    const { query } = req.body;
+    const { query, tags } = req.body;
     if (teamId == null) {
       return res.sendStatus(403);
     }
@@ -76,6 +77,7 @@ router.patch('/:id', async (req, res, next) => {
       logViewId,
       {
         query,
+        tags,
       },
       { new: true },
     );

packages/api/src/routers/api/team.ts

@@ -2,13 +2,17 @@ import crypto from 'crypto';
 import express from 'express';
 import isemail from 'isemail';
 import pick from 'lodash/pick';
+import ms from 'ms';
 import { serializeError } from 'serialize-error';
 
 import * as config from '@/config';
 import { getTeam, rotateTeamApiKey } from '@/controllers/team';
 import { findUserByEmail, findUsersByTeam } from '@/controllers/user';
+import Dashboard from '@/models/dashboard';
+import LogView from '@/models/logView';
 import TeamInvite from '@/models/teamInvite';
 import logger from '@/utils/logger';
+import { SimpleCache } from '@/utils/redis';
 
 const router = express.Router();
 
@@ -144,4 +148,43 @@ router.patch('/apiKey', async (req, res, next) => {
   }
 });
 
+router.get('/tags', async (req, res, next) => {
+  try {
+    const teamId = req.user?.team;
+    if (teamId == null) {
+      throw new Error(`User ${req.user?._id} not associated with a team`);
+    }
+    const simpleCache = new SimpleCache(
+      `tags:${teamId}`,
+      ms('10m'),
+      async () => {
+        const _tags: string[] = [];
+        const dashboards = await Dashboard.find(
+          { team: teamId },
+          { tags: 1 },
+        ).lean();
+        dashboards?.forEach(d => {
+          if (d?.tags?.length) {
+            _tags?.push(...d.tags);
+          }
+        });
+        const logViews = await LogView.find(
+          { team: teamId },
+          { tags: 1 },
+        ).lean();
+        logViews?.forEach(lv => {
+          if (lv?.tags?.length) {
+            _tags?.push(...lv.tags);
+          }
+        });
+        return _tags;
+      },
+    );
+    const tags = await simpleCache.get();
+    return res.json({ data: tags });
+  } catch (e) {
+    next(e);
+  }
+});
+
 export default router;

packages/app/src/api.ts

@@ -572,23 +572,29 @@ const api = {
     return useMutation<
       any,
       HTTPError,
-      { name: string; query: string; charts: any[] }
-    >(async ({ name, charts, query }) =>
+      { name: string; query: string; charts: any[]; tags?: string[] }
+    >(async ({ name, charts, query, tags }) =>
       server(`dashboards`, {
         method: 'POST',
-        json: { name, charts, query },
+        json: { name, charts, query, tags },
       }).json(),
     );
   },
   useUpdateDashboard() {
     return useMutation<
       any,
       HTTPError,
-      { id: string; name: string; query: string; charts: any[] }
-    >(async ({ id, name, charts, query }) =>
+      {
+        id: string;
+        name: string;
+        query: string;
+        charts: any[];
+        tags?: string[];
+      }
+    >(async ({ id, name, charts, query, tags }) =>
       server(`dashboards/${id}`, {
         method: 'PUT',
-        json: { name, charts, query },
+        json: { name, charts, query, tags },
       }).json(),
     );
   },
@@ -651,6 +657,11 @@ const api = {
       retry: 1,
     });
   },
+  useTags() {
+    return useQuery<{ data: string[] }, HTTPError>(`team/tags`, () =>
+      server(`team/tags`).json<{ data: string[] }>(),
+    );
+  },
   useSaveWebhook() {
     return useMutation<
       any,

packages/app/src/types.ts

@@ -41,6 +41,7 @@ export type LogView = {
   name: string;
   query: string;
   alerts?: Alert[];
+  tags: string[];
 };
 
 export type Dashboard = {
@@ -51,6 +52,7 @@ export type Dashboard = {
   charts: Chart[];
   alerts?: Alert[];
   query?: string;
+  tags: string[];
 };
 
 export type AlertType = 'presence' | 'absence';