uik/signals: add sendSignal mutation to directly schedule signal messages (#4501)

* graphql: add SendSignal mutation and input type

* graphql: add feature flag check to SendSignal mutation

* test: add TestSendSignal for builtin-slack-channel mutation

* test: update TestSearchTemplate_Only_SQLGeneration to handle 'only' parameter correctly

* graphql: fix indentation for sendSignal mutation definition

Author: mastercactapus

graphql2/generated.go

@@ -0,0 +1,9 @@
+extend type Mutation {
+  sendSignal(input: SendSignalInput!): Boolean!
+}
+
+input SendSignalInput {
+  dest: DestinationInput!
+  serviceID: ID!
+  params: StringMap
+}

graphql2/graph/signals.graphqls

@@ -3,9 +3,12 @@ package graphqlapp
 import (
 	context "context"
 	"database/sql"
+	"encoding/json"
 	"time"
 
 	"github.com/target/goalert/assignment"
+	"github.com/target/goalert/expflag"
+	"github.com/target/goalert/gadb"
 	"github.com/target/goalert/graphql2"
 	"github.com/target/goalert/keyring"
 	"github.com/target/goalert/permission"
@@ -14,6 +17,7 @@ import (
 	"github.com/target/goalert/user"
 	"github.com/target/goalert/util/sqlutil"
 	"github.com/target/goalert/validation"
+	"github.com/target/goalert/validation/validate"
 
 	"github.com/pkg/errors"
 )
@@ -36,6 +40,43 @@ func (a *Mutation) ReEncryptKeyringsAndConfig(ctx context.Context) (bool, error)
 	return true, nil
 }
 
+func (a *Mutation) SendSignal(ctx context.Context, input graphql2.SendSignalInput) (bool, error) {
+	err := permission.LimitCheckAny(ctx, permission.User)
+	if err != nil {
+		return false, err
+	}
+
+	if !expflag.ContextHas(ctx, expflag.UnivKeys) {
+		return false, errors.New("feature not enabled")
+	}
+
+	svcID, err := validate.ParseUUID("ServiceID", input.ServiceID)
+	if err != nil {
+		return false, err
+	}
+
+	destID, err := a.NCStore.MapDestToID(ctx, a.DB, *input.Dest)
+	if err != nil {
+		return false, err
+	}
+
+	data, err := json.Marshal(input.Params)
+	if err != nil {
+		return false, err
+	}
+
+	err = gadb.New(a.DB).IntKeyInsertSignalMessage(ctx, gadb.IntKeyInsertSignalMessageParams{
+		DestID:    destID,
+		ServiceID: svcID,
+		Params:    data,
+	})
+	if err != nil {
+		return false, err
+	}
+
+	return true, nil
+}
+
 func (a *Mutation) SetFavorite(ctx context.Context, input graphql2.SetFavoriteInput) (bool, error) {
 	var err error
 	if input.Favorite {

graphql2/graphqlapp/mutation.go

@@ -182,20 +182,22 @@ func TestSearchTemplate_Only_SQLGeneration(t *testing.T) {
 	// Verify that our QueryArgs include the only parameter
 	args := data.QueryArgs()
 
-	var onlyArg *interface{}
+	var onlyArg interface{}
+	found := false
 	for _, arg := range args {
 		if arg.Name == "only" {
-			onlyArg = &arg.Value
+			onlyArg = arg.Value
+			found = true
 			break
 		}
 	}
 
-	if onlyArg == nil {
+	if !found {
 		t.Fatal("QueryArgs() should include 'only' parameter")
 	}
 
 	// Verify the argument is not nil (UUIDArray should convert the slice)
-	if *onlyArg == nil {
+	if onlyArg == nil {
 		t.Error("'only' parameter should not be nil when Only slice is provided")
 	}
 }

graphql2/models_gen.go

@@ -0,0 +1,38 @@
+package smoke
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+	"github.com/target/goalert/expflag"
+	"github.com/target/goalert/test/smoke/harness"
+)
+
+// TestSendSignal tests the sendSignal mutation with a builtin-slack-channel destination.
+func TestSendSignal(t *testing.T) {
+	t.Parallel()
+
+	const sql = `
+		insert into escalation_policies (id, name) values
+			({{uuid "ep"}}, 'esc policy');
+		insert into services (id, name, escalation_policy_id) values
+			({{uuid "svc"}}, 'service', {{uuid "ep"}});
+	`
+
+	h := harness.NewHarnessWithFlags(t, sql, "nc-duplicate-table", expflag.FlagSet{expflag.UnivKeys})
+	defer h.Close()
+
+	chanID := h.Slack().Channel("chan1").ID()
+
+	resp := h.GraphQLQuery2(fmt.Sprintf(`mutation {
+		sendSignal(input: {
+			serviceID: "%s",
+			dest: {type: "builtin-slack-channel", args: {slack_channel_id: "%s"}},
+			params: {message: "test-signal-message"}
+		})
+	}`, h.UUID("svc"), chanID))
+	require.Empty(t, resp.Errors)
+
+	h.Slack().Channel("chan1").ExpectMessage("test-signal-message")
+}

service/service_test.go

@@ -786,6 +786,7 @@ export interface Mutation {
   promoteSecondaryToken: boolean
   reEncryptKeyringsAndConfig: boolean
   sendContactMethodVerification: boolean
+  sendSignal: boolean
   setAlertNoiseReason: boolean
   setConfig: boolean
   setFavorite: boolean
@@ -1065,6 +1066,12 @@ export interface SendContactMethodVerificationInput {
   contactMethodID: string
 }
 
+export interface SendSignalInput {
+  dest: DestinationInput
+  params?: null | StringMap
+  serviceID: string
+}
+
 export interface Service {
   alertStats: AlertStats
   alertsByStatus: AlertsByStatus