feat(cloud): Add Operations.waitOperation

Author: dnys1

packages/celest_cloud/CHANGELOG.md

@@ -1,5 +1,6 @@
 # 0.1.5-wip
 
+- feat: Add `Operations.waitOperation`
 - chore: Update license
 
 # 0.1.4

packages/celest_cloud/lib/src/cloud/operations/operations.dart

@@ -1,6 +1,7 @@
 import 'package:celest_cloud/src/cloud/base/base_service.dart';
 import 'package:celest_cloud/src/cloud/operations/operations_protocol.dart';
 import 'package:celest_cloud/src/grpc.dart';
+import 'package:celest_cloud/src/util/well_known.dart';
 import 'package:logging/logging.dart';
 
 final class Operations with BaseService {
@@ -37,4 +38,15 @@ final class Operations with BaseService {
     final request = CancelOperationRequest(name: name);
     return run('Operations.Cancel', request: request, action: _protocol.cancel);
   }
+
+  Future<Operation> wait(
+    String name, {
+    Duration? timeout,
+  }) async {
+    final request = WaitOperationRequest(
+      name: name,
+      timeout: timeout?.toProto(),
+    );
+    return run('Operations.Wait', request: request, action: _protocol.wait);
+  }
 }

packages/celest_cloud/lib/src/cloud/operations/operations_protocol.dart

@@ -4,4 +4,5 @@ abstract interface class OperationsProtocol {
   Future<Operation> get(GetOperationRequest request);
   Future<ListOperationsResponse> list(ListOperationsRequest request);
   Future<void> cancel(CancelOperationRequest request);
+  Future<Operation> wait(WaitOperationRequest request);
 }

packages/celest_cloud/lib/src/cloud/operations/operations_protocol.grpc.dart

@@ -24,4 +24,9 @@ final class OperationsProtocolGrpc implements OperationsProtocol {
   Future<ListOperationsResponse> list(ListOperationsRequest request) {
     return _client.listOperations(request);
   }
+
+  @override
+  Future<Operation> wait(WaitOperationRequest request) {
+    return _client.waitOperation(request);
+  }
 }

packages/celest_cloud/lib/src/cloud/operations/operations_protocol.http.dart

@@ -94,4 +94,33 @@ final class OperationsProtocolHttp
         typeRegistry: CelestCloud.typeRegistry,
       );
   }
+
+  @override
+  Future<Operation> wait(WaitOperationRequest request) async {
+    if (request.name == '') {
+      throw ArgumentError.value(request.name, 'name', 'must not be empty');
+    }
+    final url = Uri.parse('/v1alpha1/${request.name}:wait');
+    final req = http.Request('POST', url)
+      ..headers['content-type'] = 'application/json'
+      ..headers['accept'] = 'application/json'
+      ..body = jsonEncode(
+        request.toProto3Json(
+          typeRegistry: CelestCloud.typeRegistry,
+        ),
+      );
+    final res = await _client.send(req);
+    final body = await res.stream.toBytes();
+    if (res.statusCode != 200) {
+      throwError(
+        statusCode: res.statusCode,
+        bodyBytes: body,
+      );
+    }
+    return Operation()
+      ..mergeFromProto3Json(
+        JsonUtf8.decode(body),
+        typeRegistry: CelestCloud.typeRegistry,
+      );
+  }
 }

packages/celest_cloud/lib/src/util/well_known.dart

@@ -1,5 +1,7 @@
 import 'package:celest_cloud/src/proto.dart';
+import 'package:celest_cloud/src/proto/google/protobuf/duration.pb.dart' as pb;
 import 'package:celest_cloud/src/proto/google/protobuf/timestamp.pb.dart';
+import 'package:fixnum/fixnum.dart';
 
 extension DateTimeProto on DateTime {
   Timestamp toProto() {
@@ -8,3 +10,11 @@ extension DateTimeProto on DateTime {
     return ts;
   }
 }
+
+extension DurationProto on Duration {
+  pb.Duration toProto() {
+    return pb.Duration()
+      ..seconds = Int64(inSeconds)
+      ..nanos = inMicroseconds.remainder(Duration.microsecondsPerSecond) * 1000;
+  }
+}