RM-237: Claimer switchpanel behavior

claimsettings.go

@@ -2,18 +2,42 @@ package database
 
 import (
 	"context"
+
 	"github.com/jackc/pgx/v4"
 	"github.com/jackc/pgx/v4/pgxpool"
 )
 
+// SwitchPanelClaimBehavior defines behavior when switching a claimed ticket to a panel the claimer can't access
+type SwitchPanelClaimBehavior int
+
+const (
+	// SwitchPanelAutoUnclaim automatically unclaims the ticket if the claimer
+	// doesn't have access to the new panel (default behavior)
+	SwitchPanelAutoUnclaim SwitchPanelClaimBehavior = iota
+
+	// SwitchPanelBlockSwitch prevents switching to a panel if the claimer
+	// doesn't have access to it
+	SwitchPanelBlockSwitch
+
+	// SwitchPanelRemoveOnUnclaim allows the switch, but removes the claimer's
+	// access to the ticket when they unclaim
+	SwitchPanelRemoveOnUnclaim
+
+	// SwitchPanelKeepAccess allows the switch and keeps the claimer's access
+	// to the ticket even after unclaiming
+	SwitchPanelKeepAccess
+)
+
 type ClaimSettings struct {
-	SupportCanView bool `json:"support_can_view"`
-	SupportCanType bool `json:"support_can_type"`
+	SupportCanView            bool                     `json:"support_can_view"`
+	SupportCanType            bool                     `json:"support_can_type"`
+	SwitchPanelClaimBehavior  SwitchPanelClaimBehavior `json:"switch_panel_claim_behavior"`
 }
 
 var defaultClaimSettings = ClaimSettings{
-	SupportCanView: true,
-	SupportCanType: false,
+	SupportCanView:           true,
+	SupportCanType:           false,
+	SwitchPanelClaimBehavior: SwitchPanelAutoUnclaim,
 }
 
 type ClaimSettingsTable struct {
@@ -32,14 +56,15 @@ CREATE TABLE IF NOT EXISTS claim_settings(
 	"guild_id" int8 NOT NULL,
 	"support_can_view" bool NOT NULL,
 	"support_can_type" bool NOT NULL,
+	"switch_panel_claim_behavior" int2 NOT NULL DEFAULT 0,
 	PRIMARY KEY("guild_id")
 );
 `
 }
 
 func (c *ClaimSettingsTable) Get(ctx context.Context, guildId uint64) (settings ClaimSettings, e error) {
-	query := `SELECT "support_can_view", "support_can_type" FROM claim_settings WHERE "guild_id" = $1;`
-	if err := c.QueryRow(ctx, query, guildId).Scan(&settings.SupportCanView, &settings.SupportCanType); err != nil {
+	query := `SELECT "support_can_view", "support_can_type", "switch_panel_claim_behavior" FROM claim_settings WHERE "guild_id" = $1;`
+	if err := c.QueryRow(ctx, query, guildId).Scan(&settings.SupportCanView, &settings.SupportCanType, &settings.SwitchPanelClaimBehavior); err != nil {
 		if err == pgx.ErrNoRows {
 			settings = defaultClaimSettings
 		} else {
@@ -52,11 +77,23 @@ func (c *ClaimSettingsTable) Get(ctx context.Context, guildId uint64) (settings
 
 func (c *ClaimSettingsTable) Set(ctx context.Context, guildId uint64, settings ClaimSettings) (err error) {
 	query := `
-INSERT INTO claim_settings("guild_id", "support_can_view", "support_can_type") VALUES($1, $2, $3)
+INSERT INTO claim_settings("guild_id", "support_can_view", "support_can_type", "switch_panel_claim_behavior") VALUES($1, $2, $3, $4)
 	ON CONFLICT("guild_id") DO UPDATE SET
 	"support_can_view" = $2,
-	"support_can_type" = $3;`
+	"support_can_type" = $3,
+	"switch_panel_claim_behavior" = $4;`
+
+	_, err = c.Exec(ctx, query, guildId, settings.SupportCanView, settings.SupportCanType, settings.SwitchPanelClaimBehavior)
+	return
+}
+
+func (c *ClaimSettingsTable) SetSwitchPanelClaimBehavior(ctx context.Context, guildId uint64, behavior SwitchPanelClaimBehavior) (err error) {
+	query := `
+INSERT INTO claim_settings("guild_id", "support_can_view", "support_can_type", "switch_panel_claim_behavior")
+VALUES($1, $2, $3, $4)
+ON CONFLICT("guild_id") DO UPDATE SET "switch_panel_claim_behavior" = $4;`
 
-	_, err = c.Exec(ctx, query, guildId, settings.SupportCanView, settings.SupportCanType)
+	defaults := defaultClaimSettings
+	_, err = c.Exec(ctx, query, guildId, defaults.SupportCanView, defaults.SupportCanType, behavior)
 	return
 }