From 31226e3c75f3d3eddd99355be273ad0765c33b29 Mon Sep 17 00:00:00 2001
From: Jens Langhammer <jens.langhammer@beryju.org>
Date: Sat, 3 Apr 2021 00:10:59 +0200
Subject: [PATCH] stages/prompt: migrate to web

Signed-off-by: Jens Langhammer <jens.langhammer@beryju.org>
---
 authentik/stages/prompt/forms.py              |  16 ---
 authentik/stages/prompt/models.py             |   7 +-
 web/src/pages/stages/PromptForm.ts            |   2 +-
 web/src/pages/stages/PromptListPage.ts        |   8 +-
 .../pages/stages/prompt/PromptStageForm.ts    | 106 ++++++++++++++++++
 5 files changed, 113 insertions(+), 26 deletions(-)
 delete mode 100644 authentik/stages/prompt/forms.py
 create mode 100644 web/src/pages/stages/prompt/PromptStageForm.ts

diff --git a/authentik/stages/prompt/forms.py b/authentik/stages/prompt/forms.py
deleted file mode 100644
index 9f3b22bde..000000000
--- a/authentik/stages/prompt/forms.py
+++ /dev/null
@@ -1,16 +0,0 @@
-"""Prompt forms"""
-from django import forms
-
-from authentik.stages.prompt.models import PromptStage
-
-
-class PromptStageForm(forms.ModelForm):
-    """Form to create/edit Prompt Stage instances"""
-
-    class Meta:
-
-        model = PromptStage
-        fields = ["name", "fields", "validation_policies"]
-        widgets = {
-            "name": forms.TextInput(),
-        }
diff --git a/authentik/stages/prompt/models.py b/authentik/stages/prompt/models.py
index c5a00095f..686931103 100644
--- a/authentik/stages/prompt/models.py
+++ b/authentik/stages/prompt/models.py
@@ -3,7 +3,6 @@ from typing import Type
 from uuid import uuid4
 
 from django.db import models
-from django.forms import ModelForm
 from django.utils.translation import gettext_lazy as _
 from django.views import View
 from rest_framework.fields import (
@@ -144,10 +143,8 @@ class PromptStage(Stage):
         return PromptStageView
 
     @property
-    def form(self) -> Type[ModelForm]:
-        from authentik.stages.prompt.forms import PromptStageForm
-
-        return PromptStageForm
+    def component(self) -> str:
+        return "ak-stage-prompt-form"
 
     class Meta:
 
diff --git a/web/src/pages/stages/PromptForm.ts b/web/src/pages/stages/PromptForm.ts
index 6caa3b9cf..0247cce0f 100644
--- a/web/src/pages/stages/PromptForm.ts
+++ b/web/src/pages/stages/PromptForm.ts
@@ -7,7 +7,7 @@ import { Form } from "../../elements/forms/Form";
 import { ifDefined } from "lit-html/directives/if-defined";
 import "../../elements/forms/HorizontalFormElement";
 
-@customElement("ak-stage-prompt-form")
+@customElement("ak-prompt-form")
 export class PromptForm extends Form<Prompt> {
 
     @property({attribute: false})
diff --git a/web/src/pages/stages/PromptListPage.ts b/web/src/pages/stages/PromptListPage.ts
index 525719fd3..786c4a443 100644
--- a/web/src/pages/stages/PromptListPage.ts
+++ b/web/src/pages/stages/PromptListPage.ts
@@ -68,8 +68,8 @@ export class PromptListPage extends TablePage<Prompt> {
                 <span slot="header">
                     ${gettext("Update Prompt")}
                 </span>
-                <ak-stage-prompt-form slot="form" .prompt=${item}>
-                </ak-stage-prompt-form>
+                <ak-prompt-form slot="form" .prompt=${item}>
+                </ak-prompt-form>
                 <button slot="trigger" class="pf-c-button pf-m-secondary">
                     ${gettext("Edit")}
                 </button>
@@ -98,8 +98,8 @@ export class PromptListPage extends TablePage<Prompt> {
             <span slot="header">
                 ${gettext("Create Prompt")}
             </span>
-            <ak-stage-prompt-form slot="form">
-            </ak-stage-prompt-form>
+            <ak-prompt-form slot="form">
+            </ak-prompt-form>
             <button slot="trigger" class="pf-c-button pf-m-primary">
                 ${gettext("Create")}
             </button>
diff --git a/web/src/pages/stages/prompt/PromptStageForm.ts b/web/src/pages/stages/prompt/PromptStageForm.ts
new file mode 100644
index 000000000..988272e06
--- /dev/null
+++ b/web/src/pages/stages/prompt/PromptStageForm.ts
@@ -0,0 +1,106 @@
+import { PoliciesApi, PromptStage, StagesApi } from "authentik-api";
+import { gettext } from "django";
+import { customElement, property } from "lit-element";
+import { html, TemplateResult } from "lit-html";
+import { DEFAULT_CONFIG } from "../../../api/Config";
+import { Form } from "../../../elements/forms/Form";
+import { ifDefined } from "lit-html/directives/if-defined";
+import "../../../elements/forms/HorizontalFormElement";
+import "../../../elements/forms/FormGroup";
+import { until } from "lit-html/directives/until";
+
+@customElement("ak-stage-prompt-form")
+export class PromptStageForm extends Form<PromptStage> {
+
+    set stageUUID(value: string) {
+        new StagesApi(DEFAULT_CONFIG).stagesPromptStagesRead({
+            stageUuid: value,
+        }).then(stage => {
+            this.stage = stage;
+        });
+    }
+
+    @property({attribute: false})
+    stage?: PromptStage;
+
+    getSuccessMessage(): string {
+        if (this.stage) {
+            return gettext("Successfully updated stage.");
+        } else {
+            return gettext("Successfully created stage.");
+        }
+    }
+
+    send = (data: PromptStage): Promise<PromptStage> => {
+        if (this.stage) {
+            return new StagesApi(DEFAULT_CONFIG).stagesPromptStagesUpdate({
+                stageUuid: this.stage.pk || "",
+                data: data
+            });
+        } else {
+            return new StagesApi(DEFAULT_CONFIG).stagesPromptStagesCreate({
+                data: data
+            });
+        }
+    };
+
+    renderForm(): TemplateResult {
+        return html`<form class="pf-c-form pf-m-horizontal">
+            <ak-form-element-horizontal
+                label=${gettext("Name")}
+                ?required=${true}
+                name="name">
+                <input type="text" value="${ifDefined(this.stage?.name || "")}" class="pf-c-form-control" required>
+            </ak-form-element-horizontal>
+            <ak-form-group .expanded=${true}>
+                <span slot="header">
+                    ${gettext("Stage-specific settings")}
+                </span>
+                <div slot="body" class="pf-c-form">
+                    <ak-form-element-horizontal
+                        label=${gettext("Fields")}
+                        ?required=${true}
+                        name="fields">
+                        <select name="users" class="pf-c-form-control" multiple>
+                            ${until(new StagesApi(DEFAULT_CONFIG).stagesPromptPromptsList({
+                                ordering: "field_name"
+                            }).then(prompts => {
+                                return prompts.results.map(prompt => {
+                                    const selected = Array.from(this.stage?.fields || []).some(su => {
+                                        return su == prompt.pk;
+                                    });
+                                    return html`<option value=${ifDefined(prompt.pk)} ?selected=${selected}>
+                                        ${gettext(`${prompt.fieldKey} ('${prompt.label}', Type ${prompt.type})`)}
+                                    </option>`;
+                                });
+                            }))}
+                        </select>
+                        <p class="pf-c-form__helper-text">${gettext("Hold control/command to select multiple items.")}</p>
+                    </ak-form-element-horizontal>
+                    <ak-form-element-horizontal
+                        label=${gettext("Validation Policies")}
+                        ?required=${true}
+                        name="validationPolicies">
+                        <select name="users" class="pf-c-form-control" multiple>
+                            ${until(new PoliciesApi(DEFAULT_CONFIG).policiesAllList({
+                                ordering: "name"
+                            }).then(policies => {
+                                return policies.results.map(policy => {
+                                    const selected = Array.from(this.stage?.validationPolicies || []).some(su => {
+                                        return su == policy.pk;
+                                    });
+                                    return html`<option value=${ifDefined(policy.pk)} ?selected=${selected}>
+                                        ${gettext(`${policy.name} (${policy.verboseName})`)}
+                                    </option>`;
+                                });
+                            }))}
+                        </select>
+                        <p class="pf-c-form__helper-text">${gettext("Selected policies are executed when the stage is submitted to validate the data.")}</p>
+                        <p class="pf-c-form__helper-text">${gettext("Hold control/command to select multiple items.")}</p>
+                    </ak-form-element-horizontal>
+                </div>
+            </ak-form-group>
+        </form>`;
+    }
+
+}