feat: add withAgent to context

[skip ci]

context/context.go

@@ -10,6 +10,7 @@ import (
 	"github.com/flanksource/duty/types"
 	"github.com/flanksource/kommons"
 	"github.com/jackc/pgx/v5/pgxpool"
+	"github.com/samber/lo"
 	"go.opentelemetry.io/otel/attribute"
 	"go.opentelemetry.io/otel/trace"
 	"gorm.io/gorm"
@@ -96,6 +97,22 @@ func (k Context) User() *models.Person {
 	return v.(*models.Person)
 }
 
+// WithAgent sets the current session's agent in the context
+func (k Context) WithAgent(agent models.Agent) Context {
+	k.GetSpan().SetAttributes(attribute.String("agent-id", agent.ID.String()))
+	return Context{
+		Context: k.WithValue("agent", agent),
+	}
+}
+
+func (k Context) Agent() *models.Agent {
+	v := k.Value("agent")
+	if v == nil {
+		return nil
+	}
+	return lo.ToPtr(v.(models.Agent))
+}
+
 func (k Context) WithTrace() Context {
 	k.Context = k.Context.WithTrace()
 	return k

views/018_playbooks.sql

@@ -35,6 +35,8 @@ FOR EACH ROW
 EXECUTE PROCEDURE notify_playbook_update();
 
 -- List of all the playbooks that can be run by an agent
+DROP VIEW IF EXISTS playbooks_for_agent;
+
 CREATE OR REPLACE VIEW 
 playbooks_for_agent AS
 WITH interim AS (
@@ -50,7 +52,7 @@ SELECT
   interim.agent_name,
   agents.person_id,
   agents.id as agent_id,
-  json_agg(interim.id) AS playbook_ids
+  jsonb_agg(interim.id) AS playbook_ids
 FROM
   interim
   INNER JOIN agents ON interim.agent_name :: TEXT = agents.name