From 53c9df02c5e57142c8e9b5cb52db3b830d0969ab Mon Sep 17 00:00:00 2001
From: Siddhant Khare <Siddhantkhare2694@gmail.com>
Date: Sun, 14 Apr 2024 05:49:52 +0000
Subject: [PATCH] feat(tools/prism): Restructure tokens for easier
 highlighting.

Restructure the tokens in the language definition to use the default PrismJS tokens for easier customization of highlighting colors.
---
 src/tools/prism/language-definition.ts | 31 +++++++++++++++-----------
 1 file changed, 18 insertions(+), 13 deletions(-)

diff --git a/src/tools/prism/language-definition.ts b/src/tools/prism/language-definition.ts
index 5a39e09..18d3186 100644
--- a/src/tools/prism/language-definition.ts
+++ b/src/tools/prism/language-definition.ts
@@ -1,26 +1,29 @@
 import { OpenFgaDslThemeTokenType } from "../../theme/theme.typings";
 
 export const languageDefinition = {
-  [OpenFgaDslThemeTokenType.MODULE]: {
+  module: {
     pattern: /(module\s+)[\w_-]+/i,
-    lookbehind: true,
+    alias: OpenFgaDslThemeTokenType.MODULE,
   },
-  [OpenFgaDslThemeTokenType.TYPE]: {
+  type: {
     pattern: /(\btype\s+)[\w_-]+/i,
-    lookbehind: true,
+    alias: OpenFgaDslThemeTokenType.TYPE,
   },
-  [OpenFgaDslThemeTokenType.EXTEND]: {
+  "extend-type": {
     pattern: /(\bextend type\s+)[\w_-]+/i,
-    lookbehind: true,
+    alias: OpenFgaDslThemeTokenType.EXTEND,
   },
-  [OpenFgaDslThemeTokenType.RELATION]: {
+  relation: {
     pattern: /(\bdefine\s+)[\w_-]+/i,
-    lookbehind: true,
+    alias: OpenFgaDslThemeTokenType.RELATION,
   },
-  [OpenFgaDslThemeTokenType.DIRECTLY_ASSIGNABLE]: /\[.*]|self/,
-  [OpenFgaDslThemeTokenType.CONDITION]: {
+  "directly-assignable": {
+    pattern: /\[.*]|self/,
+    alias: OpenFgaDslThemeTokenType.DIRECTLY_ASSIGNABLE,
+  },
+  condition: {
     pattern: /(\bcondition\s+)[\w_-]+/i,
-    lookbehind: true,
+    alias: OpenFgaDslThemeTokenType.CONDITION,
   },
   "condition-params": {
     pattern: /\(.*\)\s*{/,
@@ -29,10 +32,12 @@ export const languageDefinition = {
       "condition-param-type": /\b(string|int|map|uint|list|timestamp|bool|duration|double|ipaddress)\b/,
     },
   },
-  [OpenFgaDslThemeTokenType.COMMENT]: {
+  comment: {
     pattern: /(^\s*|\s+)#.*/,
+    alias: OpenFgaDslThemeTokenType.COMMENT,
   },
-  [OpenFgaDslThemeTokenType.KEYWORD]: {
+  keyword: {
     pattern: /\b(type|relations|define|and|or|but not|from|as|model|schema|condition|module|extend)\b/,
+    alias: OpenFgaDslThemeTokenType.KEYWORD,
   },
 };