feat(backend): implement job storage; tie everything up

backend/internal/grpcsrv/commander/commander.go

@@ -15,11 +15,18 @@ import (
 type Commander struct {
 	proto.UnimplementedCommanderServer
 	agents map[string]Agent
+	jobber Jobber
 }
 
-func New() *Commander {
+type Jobber interface {
+	InitJob(ctx context.Context, agentID string, job models.JobForInsert) (int64, error)
+	UpdateJobInDB(ctx context.Context, jid int64, msg models.JobForUpdate) (models.Job, error)
+}
+
+func New(jobber Jobber) *Commander {
 	return &Commander{
 		agents: make(map[string]Agent),
+		jobber: jobber,
 	}
 }
 
@@ -27,11 +34,13 @@ type Agent struct {
 	bidi   grpc.BidiStreamingServer[proto.FinishedCommand, proto.Command]
 	in     chan *proto.Command
 	jobs   map[int64]Job
-	jobber interface {
-		InitJob(ctx context.Context) (int64, error)
-		UpdateJobInDB(ctx context.Context, jid int64, msg models.JobForUpdate) (models.Job, error)
-	}
-	ctx context.Context
+	jobber Jobber
+	ctx    context.Context
+	aid    string
+
+	Token    string // agent id
+	Label    string
+	Services []string
 }
 type JobOut struct {
 	fc  models.Job
@@ -47,15 +56,23 @@ func (self *Commander) GetAgent(aid string) (agent Agent, ok bool) {
 	return
 }
 
+func (self *Commander) Agents() []Agent {
+	result := make([]Agent, 0, len(self.agents))
+	for _, a := range self.agents {
+		result = append(result, a)
+	}
+	return result
+}
+
 func (self *Agent) AddJob(job models.JobForInsert) (int64, error) {
-	jid, err := self.jobber.InitJob(self.ctx)
+	jid, err := self.jobber.InitJob(self.ctx, self.aid, job)
 	if err != nil {
 		return 0, err
 	}
 	self.in <- &proto.Command{
-		Id:      0,
-		Command: []string{},
-		Stdin:   new(string),
+		Id:      jid,
+		Command: job.Command,
+		Stdin:   job.Stdin,
 	}
 	return jid, err
 }
@@ -75,7 +92,14 @@ func (self *Commander) Stream(bidi grpc.BidiStreamingServer[proto.FinishedComman
 		return fmt.Errorf("agentid metadata missing")
 	}
 	aid := aidVals[0]
-	agent := newAgent(bidi)
+
+	var label string
+	labelVals := md["label"]
+	if len(labelVals) > 0 {
+		label = labelVals[0]
+	}
+
+	agent := newAgent(bidi, self.jobber, aid, label)
 	self.agents[aid] = agent
 	return agent.run()
 }
@@ -121,13 +145,16 @@ func (self *Agent) send() error {
 	// self.jobs[]
 }
 
-func newAgent(bidi grpc.BidiStreamingServer[proto.FinishedCommand, proto.Command]) Agent {
+func newAgent(bidi grpc.BidiStreamingServer[proto.FinishedCommand, proto.Command], jobber Jobber, aid string, label string) Agent {
 	return Agent{
-		bidi,
-		make(chan *proto.Command),
-		make(map[int64]Job),
-		nil,
-		bidi.Context(),
+		bidi:   bidi,
+		in:     make(chan *proto.Command),
+		jobs:   make(map[int64]Job),
+		jobber: jobber,
+		ctx:    bidi.Context(),
+		aid:    aid,
+		Label:  label,
+		Token:  aid,
 	}
 }