Add WellKnownTypes to the Orchestrator

airbyte-container-orchestrator/Dockerfile

@@ -18,6 +18,9 @@ ENV AIRBYTE_ENTRYPOINT "/app/${APPLICATION}-${VERSION}/bin/${APPLICATION}"
 
 WORKDIR /app
 
+# Grab well-known types file
+COPY WellKnownTypes.json /app
+
 # Move orchestrator app
 ADD bin/${APPLICATION}-${VERSION}.tar /app
 

airbyte-container-orchestrator/build.gradle

@@ -1,3 +1,6 @@
+import groovy.json.JsonBuilder
+import groovy.yaml.YamlSlurper
+
 plugins {
     id 'application'
 }
@@ -54,6 +57,18 @@ application {
     applicationDefaultJvmArgs = ['-XX:+ExitOnOutOfMemoryError', '-XX:MaxRAMPercentage=75.0']
 }
 
+task generateWellKnownTypes() {
+    doLast {
+        def wellKnownTypesYaml = project(':airbyte-protocol').file('protocol-models/src/main/resources/airbyte_protocol/well_known_types.yaml').text
+        def parsedYaml = new YamlSlurper().parseText(wellKnownTypesYaml)
+        def wellKnownTypesJson = new JsonBuilder(parsedYaml).toPrettyString()
+        def targetFile = project.file("build/docker/WellKnownTypes.json")
+        targetFile.getParentFile().mkdirs()
+        targetFile.text = wellKnownTypesJson
+    }
+}
+
 tasks.named("buildDockerImage") {
     dependsOn copyGeneratedTar
+    dependsOn generateWellKnownTypes
 }

airbyte-workers/build.gradle

@@ -143,6 +143,7 @@ task generateWellKnownTypes() {
 
 tasks.named("buildDockerImage") {
     dependsOn copyGeneratedTar
+    dependsOn generateWellKnownTypes
 }
 
 // Ideally, we would have buildDockerImage depend on generateWellKnownTypes