feat: Get Starlark Run APIC endpoint (#1455)

## Description: We need a way to retrieve the latest Starlark run info: package id, serialized params... The same parameters that are passed to the RunStarlark* APIC methods. ## Is this change user facing? YES ## References (if applicable): https://www.notion.so/kurtosistech/Run-Config-07b2dbdab74d45d59be76061500ee435 --------- Co-authored-by: Anders Schwartz <[email protected]>
kurtosis-tech · Oct 5, 2023 · 503cb8d · 503cb8d
1 parent 39e94d9
commit 503cb8d
Show file tree

Hide file tree

Showing 25 changed files with 1,474 additions and 258 deletions.
diff --git a/api/golang/core/kurtosis_core_rpc_api_bindings/api_container_service.pb.go b/api/golang/core/kurtosis_core_rpc_api_bindings/api_container_service.pb.go
diff --git a/api/golang/core/kurtosis_core_rpc_api_bindings/api_container_service_grpc.pb.go b/api/golang/core/kurtosis_core_rpc_api_bindings/api_container_service_grpc.pb.go
diff --git a/...e_rpc_api_bindings/kurtosis_core_rpc_api_bindingsconnect/api_container_service.connect.go b/...e_rpc_api_bindings/kurtosis_core_rpc_api_bindingsconnect/api_container_service.connect.go
diff --git a/api/golang/core/lib/enclaves/enclave_context.go b/api/golang/core/lib/enclaves/enclave_context.go
@@ -408,6 +408,15 @@ func (enclaveCtx *EnclaveContext) ConnectServices(ctx context.Context, connect k
 	return nil
 }
 
+// Docs available at https://docs.kurtosis.com/#getstarlarkrun
+func (enclaveCtx *EnclaveContext) GetStarlarkRun(ctx context.Context) (*kurtosis_core_rpc_api_bindings.GetStarlarkRunResponse, error) {
+	response, err := enclaveCtx.client.GetStarlarkRun(ctx, &emptypb.Empty{})
+	if err != nil {
+		return nil, stacktrace.Propagate(err, "An error occurred while getting the last starlark run")
+	}
+	return response, nil
+}
+
 // ====================================================================================================
 //
 //	Private helper methods

diff --git a/api/golang/engine/kurtosis_engine_rpc_api_bindings/engine_service.pb.go b/api/golang/engine/kurtosis_engine_rpc_api_bindings/engine_service.pb.go
diff --git a/api/golang/engine/kurtosis_engine_rpc_api_bindings/engine_service_grpc.pb.go b/api/golang/engine/kurtosis_engine_rpc_api_bindings/engine_service_grpc.pb.go
diff --git a/api/protobuf/core/api_container_service.proto b/api/protobuf/core/api_container_service.proto
@@ -50,6 +50,9 @@ service ApiContainerService {
 
   // User services port forwarding
   rpc ConnectServices(ConnectServicesArgs) returns (ConnectServicesResponse) {};
+
+  // Get last Starlark run
+  rpc GetStarlarkRun(google.protobuf.Empty) returns (GetStarlarkRunResponse) {};
 }
 
 // ==============================================================================================
@@ -513,3 +516,30 @@ message ConnectServicesArgs {
 
 message ConnectServicesResponse {
 }
+
+// ==============================================================================================
+//                               Get Run Starlark
+// ==============================================================================================
+
+enum RestartPolicy {
+  NEVER = 0;
+  ALWAYS = 1;
+}
+
+message GetStarlarkRunResponse {
+  string package_id = 1;
+
+  string serialized_script = 2;
+
+  string serialized_params = 3;
+
+  int32 parallelism = 4;
+
+  string relative_path_to_main_file = 5;
+
+  string main_function_name = 6;
+
+  repeated KurtosisFeatureFlag experimental_features = 7;
+
+  RestartPolicy restart_policy = 8;
+}