From 79a19548fcef75461aa6820cb8935641ffdc6735 Mon Sep 17 00:00:00 2001
From: Joseph Rajewski <83741749+paladin-devops@users.noreply.github.com>
Date: Thu, 10 Aug 2023 16:56:42 -0400
Subject: [PATCH] addons: Update state interface for add-on updates.

---
 internal/server/boltdbstate/add_on.go | 2 +-
 pkg/serverstate/serverstate.go        | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/internal/server/boltdbstate/add_on.go b/internal/server/boltdbstate/add_on.go
index 79db8c8191b..1ff39528dcd 100644
--- a/internal/server/boltdbstate/add_on.go
+++ b/internal/server/boltdbstate/add_on.go
@@ -36,7 +36,7 @@ func (s *State) AddOnPut(ctx context.Context, addOn *pb.AddOn) (*pb.AddOn, error
 	return nil, status.Errorf(codes.Unimplemented, "Add On Unimplemented")
 }
 
-func (s *State) AddOnUpdate(ctx context.Context, addOn *pb.AddOn) (*pb.AddOn, error) {
+func (s *State) AddOnUpdate(ctx context.Context, addOn *pb.AddOn, existingAddOn *pb.Ref_AddOn) (*pb.AddOn, error) {
 	return nil, status.Errorf(codes.Unimplemented, "Add On Unimplemented")
 }
 
diff --git a/pkg/serverstate/serverstate.go b/pkg/serverstate/serverstate.go
index ff135ee1ca1..ee5cc92c2d8 100644
--- a/pkg/serverstate/serverstate.go
+++ b/pkg/serverstate/serverstate.go
@@ -236,6 +236,7 @@ type Interface interface {
 
 	AddOnPut(ctx context.Context, addOn *pb.AddOn) (*pb.AddOn, error)
 	AddOnGet(ctx context.Context, addOn *pb.Ref_AddOn) (*pb.AddOn, error)
+	AddOnUpdate(ctx context.Context, addOn *pb.AddOn, existingAddOn *pb.Ref_AddOn) (*pb.AddOn, error)
 	AddOnDelete(ctx context.Context, addOn *pb.Ref_AddOn) error
 	AddOnList(ctx context.Context, request *pb.ListAddOnsRequest) ([]*pb.AddOn, *pb.PaginationResponse, error)
 }