From 6d86067ceaa2d47dbd0e73a488b72a4be7c9e30b Mon Sep 17 00:00:00 2001
From: Jens Langhammer <jens.langhammer@beryju.org>
Date: Thu, 29 Dec 2022 10:51:43 +0100
Subject: [PATCH] web/elements: add grouping and descriptions to search select

Signed-off-by: Jens Langhammer <jens.langhammer@beryju.org>
---
 web/src/elements/SearchSelect.ts | 89 +++++++++++++++++++++++++-------
 1 file changed, 70 insertions(+), 19 deletions(-)

diff --git a/web/src/elements/SearchSelect.ts b/web/src/elements/SearchSelect.ts
index 07a6365f0..99fc02bb8 100644
--- a/web/src/elements/SearchSelect.ts
+++ b/web/src/elements/SearchSelect.ts
@@ -1,3 +1,4 @@
+import { groupBy } from "@goauthentik/common/utils";
 import { AKElement } from "@goauthentik/elements/Base";
 
 import { t } from "@lingui/macro";
@@ -44,23 +45,33 @@ export class SearchSelect<T> extends AKElement {
     @property({ attribute: false })
     renderElement!: (element: T) => string;
 
+    @property({ attribute: false })
+    renderDescription?: (element: T) => TemplateResult;
+
     @property({ attribute: false })
     value!: (element: T | undefined) => unknown;
 
     @property({ attribute: false })
-    selected!: (element: T) => boolean;
+    selected?: (element: T) => boolean;
 
     firstUpdated(): void {
         this.fetchObjects(this.query).then((objects) => {
             this.objects = objects;
             this.objects.forEach((obj) => {
-                if (this.selected(obj)) {
+                if (this.selected && this.selected(obj)) {
                     this.selectedObject = obj;
                 }
             });
         });
     }
 
+    @property({ attribute: false })
+    groupBy: (items: T[]) => [string, T[]][] = (items: T[]): [string, T[]][] => {
+        return groupBy(items, () => {
+            return "";
+        });
+    };
+
     menuId: string;
 
     constructor() {
@@ -86,6 +97,50 @@ export class SearchSelect<T> extends AKElement {
      */
     renderMenu(): void {
         const pos = this.getBoundingClientRect();
+        let groupedItems = this.groupBy(this.objects);
+        let shouldRenderGroups = true;
+        if (groupedItems.length === 1) {
+            if (groupedItems[0].length < 1 || groupedItems[0][0] === "") {
+                shouldRenderGroups = false;
+            }
+        }
+        if (groupedItems.length === 0) {
+            shouldRenderGroups = false;
+            groupedItems = [["", []]];
+        }
+        const renderGroup = (items: T[]): TemplateResult => {
+            return html`${items.map((obj) => {
+                let desc = undefined;
+                if (this.renderDescription) {
+                    desc = this.renderDescription(obj);
+                }
+                return html`
+                    <li>
+                        <button
+                            class="pf-c-dropdown__menu-item ${desc === undefined
+                                ? ""
+                                : "pf-m-description"}"
+                            role="option"
+                            @click=${() => {
+                                this.selectedObject = obj;
+                                this.open = false;
+                            }}
+                        >
+                            ${desc === undefined
+                                ? this.renderElement(obj)
+                                : html`
+                                      <div class="pf-c-dropdown__menu-item-main">
+                                          ${this.renderElement(obj)}
+                                      </div>
+                                      <div class="pf-c-dropdown__menu-item-description">
+                                          ${desc}
+                                      </div>
+                                  `}
+                        </button>
+                    </li>
+                `;
+            })}`;
+        };
         render(
             html`<div
                 class="pf-c-dropdown pf-m-expanded"
@@ -97,7 +152,7 @@ export class SearchSelect<T> extends AKElement {
                 <ul class="pf-c-dropdown__menu pf-m-static" role="listbox">
                     ${this.blankable
                         ? html`
-                              <li role="presentation">
+                              <li>
                                   <button
                                       class="pf-c-dropdown__menu-item"
                                       role="option"
@@ -111,22 +166,18 @@ export class SearchSelect<T> extends AKElement {
                               </li>
                           `
                         : html``}
-                    ${this.objects.map((obj) => {
-                        return html`
-                            <li role="presentation">
-                                <button
-                                    class="pf-c-dropdown__menu-item"
-                                    role="option"
-                                    @click=${() => {
-                                        this.selectedObject = obj;
-                                        this.open = false;
-                                    }}
-                                >
-                                    ${this.renderElement(obj)}
-                                </button>
-                            </li>
-                        `;
-                    })}
+                    ${shouldRenderGroups
+                        ? html`${groupedItems.map(([group, items]) => {
+                              return html`
+                                  <section class="pf-c-dropdown__group">
+                                      <h1 class="pf-c-dropdown__group-title">${group}</h1>
+                                      <ul>
+                                          ${renderGroup(items)}
+                                      </ul>
+                                  </section>
+                              `;
+                          })}`
+                        : html`${renderGroup(groupedItems[0][1])}`}
                 </ul>
             </div>`,
             document.body,