From 963b2d89ba3d47e64087970c8872159917c76a88 Mon Sep 17 00:00:00 2001
From: InfiniteStash <117855276+InfiniteStash@users.noreply.github.com>
Date: Wed, 8 Jan 2025 21:40:09 +0100
Subject: [PATCH] Add note about remaining edit update counts (#895)

---
 frontend/src/graphql/queries/Config.gql       |  1 +
 frontend/src/graphql/types.ts                 |  6 ++
 frontend/src/pages/edits/Edit.tsx             |  5 ++
 .../pages/edits/components/UpdateCount.tsx    | 22 +++++++
 graphql/schema/types/config.graphql           |  1 +
 pkg/api/resolver.go                           |  1 +
 pkg/models/generated_exec.go                  | 60 +++++++++++++++++++
 pkg/models/generated_models.go                |  1 +
 8 files changed, 97 insertions(+)
 create mode 100644 frontend/src/pages/edits/components/UpdateCount.tsx

diff --git a/frontend/src/graphql/queries/Config.gql b/frontend/src/graphql/queries/Config.gql
index ee872d826..37451d7ab 100644
--- a/frontend/src/graphql/queries/Config.gql
+++ b/frontend/src/graphql/queries/Config.gql
@@ -1,5 +1,6 @@
 query Config {
   getConfig {
+    edit_update_limit
     host_url
     require_invite
     require_activation
diff --git a/frontend/src/graphql/types.ts b/frontend/src/graphql/types.ts
index d61398bc1..a12ea93bf 100644
--- a/frontend/src/graphql/types.ts
+++ b/frontend/src/graphql/types.ts
@@ -1728,6 +1728,7 @@ export enum SortDirectionEnum {
 
 export type StashBoxConfig = {
   __typename: "StashBoxConfig";
+  edit_update_limit: Scalars["Int"]["output"];
   guidelines_url: Scalars["String"]["output"];
   host_url: Scalars["String"]["output"];
   min_destructive_voting_period: Scalars["Int"]["output"];
@@ -14531,6 +14532,7 @@ export type ConfigQuery = {
   __typename: "Query";
   getConfig: {
     __typename: "StashBoxConfig";
+    edit_update_limit: number;
     host_url: string;
     require_invite: boolean;
     require_activation: boolean;
@@ -53360,6 +53362,10 @@ export const ConfigDocument = {
             selectionSet: {
               kind: "SelectionSet",
               selections: [
+                {
+                  kind: "Field",
+                  name: { kind: "Name", value: "edit_update_limit" },
+                },
                 { kind: "Field", name: { kind: "Name", value: "host_url" } },
                 {
                   kind: "Field",
diff --git a/frontend/src/pages/edits/Edit.tsx b/frontend/src/pages/edits/Edit.tsx
index c37f86c7f..dd376016a 100644
--- a/frontend/src/pages/edits/Edit.tsx
+++ b/frontend/src/pages/edits/Edit.tsx
@@ -1,6 +1,7 @@
 import { FC, useState } from "react";
 import { Button } from "react-bootstrap";
 import { useParams, Link } from "react-router-dom";
+import { UpdateCount } from "./components/UpdateCount";
 
 import {
   useEdit,
@@ -80,6 +81,10 @@ const EditComponent: FC = () => {
   const buttons = (isAdmin || isSelf(edit.user?.id)) &&
     edit.status === VoteStatusEnum.PENDING && (
       <div className="d-flex justify-content-end">
+        <UpdateCount
+          updatable={edit.updatable}
+          updateCount={edit.update_count}
+        />
         {edit.updatable && (
           <Link to={createHref(ROUTE_EDIT_UPDATE, edit)} className="me-2">
             <Button variant="primary" disabled={mutating}>
diff --git a/frontend/src/pages/edits/components/UpdateCount.tsx b/frontend/src/pages/edits/components/UpdateCount.tsx
new file mode 100644
index 000000000..96ca6ba03
--- /dev/null
+++ b/frontend/src/pages/edits/components/UpdateCount.tsx
@@ -0,0 +1,22 @@
+import { FC } from "react";
+import { useConfig } from "src/graphql";
+
+interface Props {
+  updatable: boolean;
+  updateCount: number;
+}
+
+export const UpdateCount: FC<Props> = ({ updatable, updateCount }) => {
+  const { data: config } = useConfig();
+
+  const updateLimit = config?.getConfig.edit_update_limit;
+  if (!updatable || !updateLimit) return null;
+
+  const updates = updateLimit - updateCount;
+  return (
+    <small className="text-muted align-content-center me-3">
+      Edit can be updated{" "}
+      {updates === 1 ? "one more time" : `${updates} more times`}
+    </small>
+  );
+};
diff --git a/graphql/schema/types/config.graphql b/graphql/schema/types/config.graphql
index 072cf9de4..3737dbae4 100644
--- a/graphql/schema/types/config.graphql
+++ b/graphql/schema/types/config.graphql
@@ -9,4 +9,5 @@ type StashBoxConfig {
   vote_cron_interval: String!
   guidelines_url: String!
   require_scene_draft: Boolean!
+  edit_update_limit: Int!
 }
diff --git a/pkg/api/resolver.go b/pkg/api/resolver.go
index 83951750a..31912c5ff 100644
--- a/pkg/api/resolver.go
+++ b/pkg/api/resolver.go
@@ -129,5 +129,6 @@ func (r *queryResolver) GetConfig(ctx context.Context) (*models.StashBoxConfig,
 		VoteCronInterval:           config.GetVoteCronInterval(),
 		GuidelinesURL:              config.GetGuidelinesURL(),
 		RequireSceneDraft:          config.GetRequireSceneDraft(),
+		EditUpdateLimit:            config.GetEditUpdateLimit(),
 	}, nil
 }
diff --git a/pkg/models/generated_exec.go b/pkg/models/generated_exec.go
index 78bc31486..9c10725ba 100644
--- a/pkg/models/generated_exec.go
+++ b/pkg/models/generated_exec.go
@@ -562,6 +562,7 @@ type ComplexityRoot struct {
 	}
 
 	StashBoxConfig struct {
+		EditUpdateLimit            func(childComplexity int) int
 		GuidelinesURL              func(childComplexity int) int
 		HostURL                    func(childComplexity int) int
 		MinDestructiveVotingPeriod func(childComplexity int) int
@@ -3875,6 +3876,13 @@ func (e *executableSchema) Complexity(typeName, field string, childComplexity in
 
 		return e.complexity.Site.ValidTypes(childComplexity), true
 
+	case "StashBoxConfig.edit_update_limit":
+		if e.complexity.StashBoxConfig.EditUpdateLimit == nil {
+			break
+		}
+
+		return e.complexity.StashBoxConfig.EditUpdateLimit(childComplexity), true
+
 	case "StashBoxConfig.guidelines_url":
 		if e.complexity.StashBoxConfig.GuidelinesURL == nil {
 			break
@@ -4661,6 +4669,7 @@ var sources = []*ast.Source{
   vote_cron_interval: String!
   guidelines_url: String!
   require_scene_draft: Boolean!
+  edit_update_limit: Int!
 }
 `, BuiltIn: false},
 	{Name: "../../graphql/schema/types/draft.graphql", Input: `type DraftSubmissionStatus {
@@ -26116,6 +26125,8 @@ func (ec *executionContext) fieldContext_Query_getConfig(_ context.Context, fiel
 				return ec.fieldContext_StashBoxConfig_guidelines_url(ctx, field)
 			case "require_scene_draft":
 				return ec.fieldContext_StashBoxConfig_require_scene_draft(ctx, field)
+			case "edit_update_limit":
+				return ec.fieldContext_StashBoxConfig_edit_update_limit(ctx, field)
 			}
 			return nil, fmt.Errorf("no field named %q was found under type StashBoxConfig", field.Name)
 		},
@@ -31411,6 +31422,50 @@ func (ec *executionContext) fieldContext_StashBoxConfig_require_scene_draft(_ co
 	return fc, nil
 }
 
+func (ec *executionContext) _StashBoxConfig_edit_update_limit(ctx context.Context, field graphql.CollectedField, obj *StashBoxConfig) (ret graphql.Marshaler) {
+	fc, err := ec.fieldContext_StashBoxConfig_edit_update_limit(ctx, field)
+	if err != nil {
+		return graphql.Null
+	}
+	ctx = graphql.WithFieldContext(ctx, fc)
+	defer func() {
+		if r := recover(); r != nil {
+			ec.Error(ctx, ec.Recover(ctx, r))
+			ret = graphql.Null
+		}
+	}()
+	resTmp, err := ec.ResolverMiddleware(ctx, func(rctx context.Context) (interface{}, error) {
+		ctx = rctx // use context from middleware stack in children
+		return obj.EditUpdateLimit, nil
+	})
+	if err != nil {
+		ec.Error(ctx, err)
+		return graphql.Null
+	}
+	if resTmp == nil {
+		if !graphql.HasFieldError(ctx, fc) {
+			ec.Errorf(ctx, "must not be null")
+		}
+		return graphql.Null
+	}
+	res := resTmp.(int)
+	fc.Result = res
+	return ec.marshalNInt2int(ctx, field.Selections, res)
+}
+
+func (ec *executionContext) fieldContext_StashBoxConfig_edit_update_limit(_ context.Context, field graphql.CollectedField) (fc *graphql.FieldContext, err error) {
+	fc = &graphql.FieldContext{
+		Object:     "StashBoxConfig",
+		Field:      field,
+		IsMethod:   false,
+		IsResolver: false,
+		Child: func(ctx context.Context, field graphql.CollectedField) (*graphql.FieldContext, error) {
+			return nil, errors.New("field of type Int does not have child fields")
+		},
+	}
+	return fc, nil
+}
+
 func (ec *executionContext) _Studio_id(ctx context.Context, field graphql.CollectedField, obj *Studio) (ret graphql.Marshaler) {
 	fc, err := ec.fieldContext_Studio_id(ctx, field)
 	if err != nil {
@@ -49323,6 +49378,11 @@ func (ec *executionContext) _StashBoxConfig(ctx context.Context, sel ast.Selecti
 			if out.Values[i] == graphql.Null {
 				out.Invalids++
 			}
+		case "edit_update_limit":
+			out.Values[i] = ec._StashBoxConfig_edit_update_limit(ctx, field, obj)
+			if out.Values[i] == graphql.Null {
+				out.Invalids++
+			}
 		default:
 			panic("unknown field " + strconv.Quote(field.Name))
 		}
diff --git a/pkg/models/generated_models.go b/pkg/models/generated_models.go
index c49ef9ccc..91d07df0e 100644
--- a/pkg/models/generated_models.go
+++ b/pkg/models/generated_models.go
@@ -703,6 +703,7 @@ type StashBoxConfig struct {
 	VoteCronInterval           string `json:"vote_cron_interval"`
 	GuidelinesURL              string `json:"guidelines_url"`
 	RequireSceneDraft          bool   `json:"require_scene_draft"`
+	EditUpdateLimit            int    `json:"edit_update_limit"`
 }
 
 type StringCriterionInput struct {