From a3bad8cb0b562208172bbb8ecad5e795861468fd Mon Sep 17 00:00:00 2001
From: Matthieu Moquet <matthieu@moquet.net>
Date: Tue, 10 Dec 2024 11:30:03 +0100
Subject: [PATCH] feat: parse filters from query string

---
 src/components/alerts/template.tsx | 12 +++++---
 src/components/alerts/types.ts     |  4 +--
 src/components/alerts/utils.ts     | 45 ++++++++++++++++++++++++++++++
 tsconfig.json                      |  2 +-
 4 files changed, 56 insertions(+), 7 deletions(-)

diff --git a/src/components/alerts/template.tsx b/src/components/alerts/template.tsx
index 7bdd484..d628c7c 100644
--- a/src/components/alerts/template.tsx
+++ b/src/components/alerts/template.tsx
@@ -2,7 +2,7 @@
 
 import { useEffect, useState } from 'react'
 import { notFound } from 'next/navigation'
-import { useQueryState } from 'nuqs'
+import { parseAsArrayOf, useQueryState } from 'nuqs'
 import { ListFilter, LoaderCircle, RefreshCcw, TriangleAlert } from 'lucide-react'
 import { ViewConfig } from '@/config/types'
 import { useAlerts } from '@/contexts/alerts'
@@ -13,8 +13,8 @@ import { Alert } from '@/types/alertmanager'
 import { Tooltip, TooltipContent, TooltipProvider, TooltipTrigger } from '@/components/ui/tooltip'
 import { AlertGroups } from './alert-groups'
 import { AlertModal } from './alert-modal'
-import { Group } from './types'
-import { alertFilter, alertSort } from './utils'
+import { Group, LabelFilter } from './types'
+import { alertFilter, alertSort, parseAsFilter } from './utils'
 import {
   Select,
   SelectContent,
@@ -41,6 +41,7 @@ export function AlertsTemplate(props: Props) {
   const [flattenedAlerts, setFlattenedAlerts] = useState<Alert[]>([])
   const [view, setView] = useState<ViewConfig | null>(null)
   const [alertGroups, setAlertGroups] = useState<Group[]>([])
+  const [filters] = useQueryState('filters', parseAsArrayOf(parseAsFilter, ';'))
 
   useHotkeys('r', () => refreshAlerts(), []);
 
@@ -55,10 +56,13 @@ export function AlertsTemplate(props: Props) {
 
     const groupBy = group !== '' ? group : view.groupBy
 
+    // Merge view filters with query filters
+    const alertFilters = view.filters.concat(filters || [])
+
     // Flatten, filter & sort alerts
     const flatAlerts = Object.values(alerts).
       reduce((acc, val) => acc.concat(val), []).
-      filter(alertFilter(view.filters))
+      filter(alertFilter(alertFilters))
     flatAlerts.sort(alertSort)
 
     setFlattenedAlerts(flatAlerts)
diff --git a/src/components/alerts/types.ts b/src/components/alerts/types.ts
index 4ad57c8..db51fe6 100644
--- a/src/components/alerts/types.ts
+++ b/src/components/alerts/types.ts
@@ -10,6 +10,6 @@ export type Filter = (alert: Alert) => boolean
 export type LabelFilter = {
   label: string
   value: string | string[]
-  exclude?: boolean
-  regex?: boolean
+  exclude: boolean
+  regex: boolean
 }
diff --git a/src/components/alerts/utils.ts b/src/components/alerts/utils.ts
index 2b9c067..ae457cf 100644
--- a/src/components/alerts/utils.ts
+++ b/src/components/alerts/utils.ts
@@ -1,8 +1,53 @@
 import { Alert } from "@/types/alertmanager"
 import { LabelFilter } from "./types"
+import { createParser } from "nuqs"
 
 const showOnlyActive = true
 
+const filterRegex = /(?<label>\w+)(?<pattern>|!=|=~|!~|=)(?<value>[^!=~]+)/
+
+export function parseFilter(filterStr: string): LabelFilter {
+  const match = filterStr.match(filterRegex)
+
+  const label = match?.groups?.label || ''
+  const regex = match?.groups?.pattern === '=~' || match?.groups?.pattern === '!~'
+  const exclude = match?.groups?.pattern === '!=' || match?.groups?.pattern === '!~'
+
+  let value: string | string[] = match?.groups?.value || ''
+  if (typeof value === 'string' && value.includes(',')) {
+    value = value.split(',')
+  }
+
+  return { label, exclude, regex, value }
+}
+
+export function filterToString(filter: LabelFilter): string {
+  const label = filter.label
+  const value = Array.isArray(filter.value) ? filter.value.join(',') : filter.value
+
+  if (filter.regex && filter.exclude) {
+    return `${label}!~${value}`
+  }
+  if (filter.regex) {
+    return `${label}=~${value}`
+  }
+  if (filter.exclude) {
+    return `${label}!=${value}`
+  }
+
+  return `${label}=${value}`
+}
+
+// Custom nuqs parser
+export const parseAsFilter = createParser<LabelFilter>({
+  parse(queryValue) {
+    return parseFilter(queryValue)
+  },
+  serialize(value) {
+    return filterToString(value)
+  }
+})
+
 export function flattenAlerts(alerts: Record<string, Alert[]>): Alert[] {
   return Object.values(alerts).reduce((acc, val) => acc.concat(val), [])
 }
diff --git a/tsconfig.json b/tsconfig.json
index c133409..8f371e8 100644
--- a/tsconfig.json
+++ b/tsconfig.json
@@ -1,6 +1,6 @@
 {
   "compilerOptions": {
-    "target": "ES2017",
+    "target": "ES2018",
     "lib": ["dom", "dom.iterable", "esnext"],
     "allowJs": true,
     "skipLibCheck": true,