test adding a stream on the server

Emyrk · Emyrk · commit 73151d1790a3 · 2025-06-06T09:19:23.000-05:00
diff --git a/provisionerd/proto/provisionerd.proto b/provisionerd/proto/provisionerd.proto
@@ -106,6 +106,34 @@ message CompletedJob {
     }
 }
 
+enum DataUploadType {
+	UPLOAD_TYPE_UNKNOWN = 0;
+	// UPLOAD_TYPE_MODULE_FILES is used to stream over terraform module files.
+	// These files are located in `.terraform/modules` and are used for dynamic
+	// parameters.
+	UPLOAD_TYPE_MODULE_FILES = 1;
+}
+
+message DataUpload {
+	DataUploadType upload_type = 1;
+	// data_hash is the sha256 of the payload to be uploaded.
+	// This is also used to uniquely identify the upload.
+	bytes data_hash = 2;
+	// file_size is the total size of the data being uploaded.
+	int64 file_size = 3;
+	// Number of chunks to be uploaded.
+	int32 chunks = 4;
+}
+
+// ChunkPiece is used to stream over large files (over the 4mb limit).
+message ChunkPiece {
+	bytes data = 1;
+	// full_data_hash should match the hash from the original
+	// DataUpload message
+	bytes full_data_hash = 2;
+	int32 piece_index = 3;
+}
+
 // LogSource represents the sender of the log.
 enum LogSource {
     PROVISIONER_DAEMON = 0;
@@ -153,6 +181,14 @@ message CommitQuotaResponse {
 
 message CancelAcquire {}
 
+message CompleteWithFilesRequest {
+    oneof type {
+        CompletedJob complete = 1;
+        DataUpload data_upload = 2;
+        ChunkPiece chunk_piece = 3;
+    }
+}
+
 service ProvisionerDaemon {
     // AcquireJob requests a job. Implementations should
     // hold a lock on the job until CompleteJob() is
@@ -180,4 +216,5 @@ service ProvisionerDaemon {
 
     // CompleteJob indicates a job has been completed.
     rpc CompleteJob(CompletedJob) returns (Empty);
+    rpc CompleteJobWithFiles(stream CompleteWithFilesRequest) returns (Empty);
 }
