Merge pull request #6 from cloudquery/mnorbury-adding-cli-skeleton

Adding initial CLI structure

Author: mnorbury

.gitignore

@@ -28,3 +28,6 @@ replay_pid*
 .DS_Store
 # Ignore Gradle build output directory
 build
+
+# Intellij
+.idea

lib/build.gradle

@@ -1,5 +1,6 @@
 plugins {
     id 'java-library'
+    id "io.freefair.lombok" version "8.1.0"
 }
 
 repositories {
@@ -19,9 +20,12 @@ dependencies {
 
     // This dependency is used internally, and not exposed to consumers on their own compile classpath.
     implementation 'com.google.guava:guava:31.1-jre'
+    implementation 'info.picocli:picocli:4.7.4'
+    implementation 'com.google.guava:guava:31.1-jre'
     implementation "io.grpc:grpc-protobuf:1.57.0"
     implementation "io.grpc:grpc-stub:1.57.0"
     implementation "io.grpc:grpc-services:1.57.0"
+    implementation "io.grpc:grpc-testing:1.15.1"
     implementation 'com.cloudquery:plugin-pb-java:0.0.2'
 }
 
@@ -38,6 +42,6 @@ testing {
 // Apply a specific Java toolchain to ease working on different environments.
 java {
     toolchain {
-        languageVersion = JavaLanguageVersion.of(18)
+        languageVersion = JavaLanguageVersion.of(20)
     }
 }

lib/src/main/java/cloudquery/plugin/sdk/DiscoveryServer.java

@@ -0,0 +1,22 @@
+package io.cloudquery.internal.servers.discovery.v1;
+
+import cloudquery.discovery.v1.DiscoveryGrpc.DiscoveryImplBase;
+import cloudquery.discovery.v1.DiscoveryOuterClass.GetVersions.Request;
+import cloudquery.discovery.v1.DiscoveryOuterClass.GetVersions.Response;
+import io.grpc.stub.StreamObserver;
+
+import java.util.List;
+
+public class DiscoverServer extends DiscoveryImplBase {
+    private final List<Integer> versions;
+
+    public DiscoverServer(List<Integer> versions) {
+        this.versions = versions;
+    }
+
+    @Override
+    public void getVersions(Request request, StreamObserver<Response> responseObserver) {
+        responseObserver.onNext(Response.newBuilder().addAllVersions(versions).build());
+        responseObserver.onCompleted();
+    }
+}

lib/src/main/java/cloudquery/plugin/sdk/PluginServer.java

@@ -0,0 +1,12 @@
+package io.cloudquery.internal.servers.plugin.v3;
+
+import cloudquery.plugin.v3.PluginGrpc.PluginImplBase;
+import io.cloudquery.plugin.Plugin;
+
+public class PluginServer extends PluginImplBase {
+    private final Plugin plugin;
+
+    public PluginServer(Plugin plugin) {
+        this.plugin = plugin;
+    }
+}

lib/src/main/java/io/cloudquery/internal/servers/discovery/v1/DiscoverServer.java

@@ -0,0 +1,18 @@
+package io.cloudquery.plugin;
+
+import lombok.Builder;
+import lombok.Getter;
+import lombok.NonNull;
+
+@Builder(builderMethodName = "innerBuilder")
+@Getter
+public class Plugin {
+    public static PluginBuilder builder(String name, String version) {
+        return innerBuilder().name(name).verion(version);
+    }
+
+    @NonNull
+    private final String name;
+    @NonNull
+    private final String verion;
+}

lib/src/main/java/io/cloudquery/internal/servers/plugin/v3/PluginServer.java

@@ -0,0 +1,22 @@
+package io.cloudquery.server;
+
+import picocli.CommandLine.ITypeConverter;
+
+public class AddressConverter implements ITypeConverter<AddressConverter.Address> {
+    public static class AddressParseException extends RuntimeException {
+
+    }
+    public  record Address(String host, int port) {
+    }
+
+    @Override
+    public Address convert(String rawAddress) throws Exception {
+        String[] components = rawAddress.split(":");
+        if (components.length != 2) {
+            throw new AddressParseException();
+        }
+        return new Address(components[0], Integer.parseInt(components[1]));
+    }
+
+
+}

lib/src/main/java/io/cloudquery/plugin/Plugin.java

@@ -0,0 +1,7 @@
+package io.cloudquery.server;
+
+import picocli.CommandLine;
+
+@CommandLine.Command
+public class DocCommand {
+}

lib/src/main/java/io/cloudquery/server/AddressConverter.java

@@ -0,0 +1,33 @@
+package io.cloudquery.server;
+
+import io.cloudquery.plugin.Plugin;
+import lombok.AccessLevel;
+import lombok.Builder;
+import lombok.NonNull;
+import picocli.CommandLine;
+
+import java.util.ArrayList;
+import java.util.List;
+
+@Builder(access = AccessLevel.PUBLIC)
+public class PluginServe {
+    @NonNull
+    private final Plugin plugin;
+    @Builder.Default
+    private List<String> args = new ArrayList<>();
+    private boolean destinationV0V1Server;
+    private String sentryDSN;
+    private boolean testListener;
+    //TODO: Allow a test listener to be passed in
+    // 	testListenerConn      *bufconn.Listener
+
+    public void Serve() throws ServerException {
+        int exitStatus = new CommandLine(new RootCommand()).
+                addSubcommand("serve", new ServeCommand(plugin)).
+                addSubcommand("doc", new DocCommand()).
+                execute(args.toArray(new String[]{}));
+        if (exitStatus != 0) {
+            throw new ServerException("error processing command line exit status = "+exitStatus);
+        }
+    }
+}