From aed5a4db1dfab85aff0c118a4a55008be7042854 Mon Sep 17 00:00:00 2001
From: sawka <mike.sawka@gmail.com>
Date: Sun, 26 Mar 2023 13:36:33 -0700
Subject: [PATCH] multiple updates in request

---
 db/migrations/000016_webptypos.up.sql |  2 +-
 pkg/pcloud/pcloud.go                  | 27 ++++++++++++++++++++++++---
 2 files changed, 25 insertions(+), 4 deletions(-)

diff --git a/db/migrations/000016_webptypos.up.sql b/db/migrations/000016_webptypos.up.sql
index 69ee7486..e21618b0 100644
--- a/db/migrations/000016_webptypos.up.sql
+++ b/db/migrations/000016_webptypos.up.sql
@@ -2,7 +2,7 @@ CREATE TABLE webptypos (
     screenid varchar(36) NOT NULL,
     lineid varchar(36) NOT NULL,
     ptypos bigint NOT NULL,
-    PRIMARY KEY (screenid, cmdid)
+    PRIMARY KEY (screenid, lineid)
 );
 
 CREATE INDEX idx_screenupdate_ids ON screenupdate (screenid, lineid);
diff --git a/pkg/pcloud/pcloud.go b/pkg/pcloud/pcloud.go
index a5a83f75..d457fd51 100644
--- a/pkg/pcloud/pcloud.go
+++ b/pkg/pcloud/pcloud.go
@@ -22,10 +22,12 @@ const PCloudEndpoint = "https://api.getprompt.dev/central"
 const PCloudEndpointVarName = "PCLOUD_ENDPOINT"
 const APIVersion = 1
 const MaxPtyUpdateSize = (128 * 1024) + 1
+const MaxUpdatesPerReq = 10
 
 const TelemetryUrl = "/telemetry"
 const NoTelemetryUrl = "/no-telemetry"
 const CreateWebScreenUrl = "/auth/create-web-screen"
+const WebShareUpdateUrl = "/auth/web-share-update"
 
 type AuthInfo struct {
 	UserId   string `json:"userid"`
@@ -177,7 +179,7 @@ func defaultError(err error, estr string) error {
 	return errors.New(estr)
 }
 
-func makeWebScreenUpdate(ctx context.Context, update sstore.ScreenUpdateType) (*WebShareUpdateType, error) {
+func makeWebShareUpdate(ctx context.Context, update *sstore.ScreenUpdateType) (*WebShareUpdateType, error) {
 	rtn := &WebShareUpdateType{
 		ScreenId:   update.ScreenId,
 		LineId:     update.LineId,
@@ -286,7 +288,7 @@ func makeWebScreenUpdate(ctx context.Context, update sstore.ScreenUpdateType) (*
 	return rtn, nil
 }
 
-func finalizeWebScreenUpdate(ctx context.Context, screenUpdate sstore.ScreenUpdateType, webUpdate *WebShareUpdateType) error {
+func finalizeWebScreenUpdate(ctx context.Context, screenUpdate *sstore.ScreenUpdateType, webUpdate *WebShareUpdateType) error {
 	switch screenUpdate.UpdateType {
 	case sstore.UpdateType_PtyPos:
 		dataEof := len(webUpdate.PtyData.Data) < MaxPtyUpdateSize
@@ -312,7 +314,26 @@ func finalizeWebScreenUpdate(ctx context.Context, screenUpdate sstore.ScreenUpda
 	return nil
 }
 
-func DoWebScreenUpdate(ctx context.Context, update sstore.ScreenUpdateType) error {
+func DoWebScreenUpdates(ctx context.Context, authInfo AuthInfo, updateArr []*sstore.ScreenUpdateType) error {
+	var webUpdates []*WebShareUpdateType
+	for _, update := range updateArr {
+		webUpdate, err := makeWebShareUpdate(ctx, update)
+		if err != nil {
+			return fmt.Errorf("error create web-share update updateid:%d: %v", update.UpdateId, err)
+		}
+		if webUpdate == nil {
+			continue
+		}
+		webUpdates = append(webUpdates, webUpdate)
+	}
+	req, err := makeAuthPostReq(ctx, WebShareUpdateUrl, authInfo, webUpdates)
+	if err != nil {
+		return fmt.Errorf("cannot create auth-post-req for %s: %v", WebShareUpdateUrl, err)
+	}
+	_, err = doRequest(req, nil)
+	if err != nil {
+		return err
+	}
 	return nil
 }