gen-tests.gradle

import org.jetbrains.kotlin.gradle.plugin.KotlinPlatformType

import java.nio.file.Files
import java.nio.file.Paths
import java.util.stream.Collectors

buildscript {
  dependencies {
    classpath deps.plugins.kotlin
  }
  repositories {
    mavenCentral()
  }
}

def PROTOS = Files.find(
      Paths.get("${rootProject.rootDir}/wire-library/wire-tests/src/commonTest/proto/java"),
      Integer.MAX_VALUE,
      { path, attrs -> attrs.isRegularFile() }
    )
    .map { path ->
      Paths.get("${rootProject.rootDir}/wire-library/wire-tests/src/commonTest/proto/java").relativize(path)
    }
    .filter { path -> !path.startsWith("kotlin") }
    .map { path -> path.toString() }
    .filter { path -> path.endsWith(".proto") }
    .sorted()
    .collect(Collectors.toList())

configurations {
  wire.with {
    def platformAttr = Attribute.of('org.jetbrains.kotlin.platform.type', KotlinPlatformType.class)
    attributes.attribute(platformAttr, KotlinPlatformType.jvm)
  }
}

dependencies {
  wire deps.wire.compiler
}

// JAVA

task generateJavaTests(type: JavaExec) {
  group = 'Generate Tests'
  description = 'Generates Java classes from the test protos'
  classpath = configurations.wire
  main = 'com.squareup.wire.WireCompiler'
  args = [
      '--proto_path=wire-library/wire-tests/src/commonTest/proto/java',
      '--java_out=wire-library/wire-tests/src/jvmJavaTest/proto-java',
      '--emit_applied_options',
      '--excludes=squareup.options.*',
  ] + PROTOS
}

// NO OPTIONS

task generateNoOptionsTests(type: JavaExec) {
  group = 'Generate Tests'
  description = 'Generates Java classes with no options from the test protos'
  classpath = configurations.wire
  main = 'com.squareup.wire.WireCompiler'
  args = [
      '--proto_path=wire-library/wire-tests/src/commonTest/proto/java',
      '--java_out=wire-library/wire-tests/src/jvmJavaNoOptionsTest/proto-java',
      '--excludes=google.protobuf.*',
      '--excludes=squareup.options.*',
      '--excludes=com.squareup.wire.ModelEvaluation',
  ] + PROTOS
}

// COMPACT

task generateCompactTests(type: JavaExec) {
  group = 'Generate Tests'
  description = 'Generates compact Java classes from the test protos'
  classpath = configurations.wire
  main = 'com.squareup.wire.WireCompiler'
  args = [
      '--proto_path=wire-library/wire-tests/src/commonTest/proto/java',
      '--java_out=wire-library/wire-tests/src/jvmJavaCompactTest/proto-java',
      '--compact',
      'all_types.proto'
  ]
}

// GSON

task generateGsonAdapterCompactJavaTests(type: JavaExec) {
  group = 'Generate GsonAdapter Compact Java Tests'
  description = 'Generates compat Java classes from the GsonAdapter test protos'
  classpath = configurations.wire
  main = 'com.squareup.wire.WireCompiler'
  args = [
          '--proto_path=wire-library/wire-tests/src/commonTest/shared/proto/proto2',
          '--java_out=wire-library/wire-gson-support/src/test/java',
          '--compact',
          'all_types_proto2.proto'
  ]
}

task generateGsonAdapterJavaTests(type: JavaExec) {
  group = 'Generate GsonAdapter Java Tests'
  description = 'Generates Java classes from the GsonAdapter test protos'
  classpath = configurations.wire
  main = 'com.squareup.wire.WireCompiler'
  args = [
          '--proto_path=wire-library/wire-tests/src/commonTest/shared/proto',
          '--proto_path=wire-library/wire-tests/src/commonTest/shared/proto/proto2',
          '--java_out=wire-library/wire-gson-support/src/test/java',
          'collection_types.proto',
          'dinosaur_java.proto',
          'period_java.proto',
          'keyword_java.proto',
  ]
}

task generateGsonAdapterInteropKotlinTests(type: JavaExec) {
  group = 'Generate GsonAdapter Java interop Kotlin Tests'
  description = 'Generates Java interop Kotlin classes from the GsonAdapter test protos'
  classpath = configurations.wire
  main = 'com.squareup.wire.WireCompiler'
  args = [
          '--proto_path=wire-library/wire-tests/src/commonTest/shared/proto',
          '--proto_path=wire-library/wire-tests/src/commonTest/shared/proto/proto2',
          '--kotlin_out=wire-library/wire-gson-support/src/test/java',
          '--java_interop',
          'dinosaur_java_interop_kotlin.proto',
          'period_java_interop_kotlin.proto',
          'keyword_kotlin.proto',
  ]
}

task generateGsonAdapterKotlinTests(type: JavaExec) {
  group = 'Generate GsonAdapter Kotlin Tests'
  description = 'Generates Kotlin classes from the GsonAdapter test protos'
  classpath = configurations.wire
  main = 'com.squareup.wire.WireCompiler'
  args = [
          '--proto_path=wire-library/wire-tests/src/commonTest/shared/proto',
          '--proto_path=wire-library/wire-tests/src/commonTest/shared/proto/proto2',
          '--kotlin_out=wire-library/wire-gson-support/src/test/java',
          'dinosaur_kotlin.proto',
          'period_kotlin.proto',
  ]
}

task generateGsonTests(dependsOn: [generateGsonAdapterCompactJavaTests,
    generateGsonAdapterJavaTests, generateGsonAdapterInteropKotlinTests,
    generateGsonAdapterKotlinTests]) {
  group = 'Generate Tests'
  description = 'Generates Java and Kotlin classes that use Gson from the test protos'
}


// INCLUDES / EXCLUDES

task generateIncludesExcludesTests(type: JavaExec) {
  group = 'Generate Tests'
  description = 'Generates Java classes with included and excluded protos from the test protos'
  classpath = configurations.wire
  main = 'com.squareup.wire.WireCompiler'
  args = [
      '--proto_path=wire-library/wire-tests/src/commonTest/proto/java',
      '--java_out=wire-library/wire-tests/src/jvmJavaPrunedTest/proto-java',
      '--includes=squareup.protos.roots.A,squareup.protos.roots.H',
      '--excludes=squareup.protos.roots.B',
      'roots.proto'
  ]
}

// ANDROID

task generateAndroidTests(type: JavaExec) {
  group = 'Generate Tests'
  description = 'Generates Java classes for Android from the test protos'
  classpath = configurations.wire
  main = 'com.squareup.wire.WireCompiler'
  args = [
      '--proto_path=wire-library/wire-tests/src/commonTest/proto/java',
      '--java_out=wire-library/wire-tests/src/jvmJavaAndroidTest/proto-java',
      '--android',
      'person.proto'
  ]
}

// ANDROID COMPACT

task generateAndroidCompactTests(type: JavaExec) {
  group = 'Generate Tests'
  description = 'Generates compact Java classes for Android from the test protos'
  classpath = configurations.wire
  main = 'com.squareup.wire.WireCompiler'
  args = [
      '--proto_path=wire-library/wire-tests/src/commonTest/proto/java',
      '--java_out=wire-library/wire-tests/src/jvmJavaAndroidCompactTest/proto-java',
      '--android',
      '--compact',
      'person.proto'
  ]
}

// KOTLIN

task generateKotlinTests(type: JavaExec) {
  group = 'Generate Tests'
  description = 'Generates Kotlin classes from the test protos'
  classpath = configurations.wire
  main = 'com.squareup.wire.WireCompiler'
  args = [
      '--proto_path=wire-library/wire-tests/src/commonTest/proto/kotlin',
      '--kotlin_out=wire-library/wire-tests/src/commonTest/proto-kotlin',
      '--emit_applied_options',
      '--kotlin_box_oneofs_min_size=8',
      'all_types.proto',
      'custom_options.proto',
      'deprecated.proto',
      'deprecated_enum.proto',
      'edge_cases.proto',
      'external_message.proto',
      'foreign.proto',
      'form.proto',
      'map.proto',
      'no_fields.proto',
      'one_of.proto',
      'optional_enum.proto',
      'option_redacted.proto',
      'packed_encoding.proto',
      'person.proto',
      'redacted_one_of.proto',
      'redacted_test.proto',
      'same_name_enum.proto',
      'simple_message.proto',
      'to_string.proto',
      'unknown_fields.proto',
      'uses_any.proto',
  ]
}

// KOTLIN PROTO3

task generateProto3KotlinTests(type: JavaExec) {
  group = 'Generate Tests'
  description = 'Generates Kotlin classes from the test proto3s'
  classpath = configurations.wire
  main = 'com.squareup.wire.WireCompiler'
  args = [
          '--proto_path=wire-library/wire-tests/src/commonTest/proto3/kotlin',
          '--kotlin_out=wire-library/wire-tests/src/commonTest/proto-kotlin',
          'all_types.proto',
          'person.proto',
  ]
}

// Java PROTO3

task generateProto3JavaTests(type: JavaExec) {
  group = 'Generate Tests'
  description = 'Generates Kotlin classes from the test proto3s'
  classpath = configurations.wire
  main = 'com.squareup.wire.WireCompiler'
  args = [
          '--proto_path=wire-library/wire-tests/src/commonTest/proto3/java',
          '--java_out=wire-library/wire-tests/src/jvmJavaTest/proto-java',
          'all_types.proto',
          'person.proto',
  ]
}

// KOTLIN SERVICES

task generateKotlinServicesTests(type: JavaExec) {
  group = 'Generate Tests'
  description = 'Generates Kotlin classes from the test protos'
  classpath = configurations.wire
  main = 'com.squareup.wire.WireCompiler'
  args = [
      '--proto_path=wire-library/wire-tests/src/commonTest/proto/kotlin',
      '--kotlin_out=wire-library/wire-tests/src/jvmKotlinInteropTest/proto-kotlin',
      'service_kotlin.proto',
      'service_without_package.proto',
  ]
}

// KOTLIN ANDROID

task generateKotlinAndroidTests(type: JavaExec) {
  group = 'Generate Tests'
  description = 'Generates Kotlin classes for Android from the test protos'
  classpath = configurations.wire
  main = 'com.squareup.wire.WireCompiler'
  args = [
      '--proto_path=wire-library/wire-tests/src/commonTest/proto/kotlin',
      '--kotlin_out=wire-library/wire-tests/src/jvmKotlinAndroidTest/proto-kotlin',
      '--android',
      'person.proto'
  ]
}

// KOTLIN JAVA INTEROP

task generateKotlinJavaInteropTests(type: JavaExec) {
  group = 'Generate Tests'
  description = 'Generates Kotlin classes with Java interop from the test protos'
  classpath = configurations.wire
  main = 'com.squareup.wire.WireCompiler'
  args = [
      '--proto_path=wire-library/wire-tests/src/commonTest/proto/kotlin',
      '--kotlin_out=wire-library/wire-tests/src/jvmKotlinInteropTest/proto-kotlin',
      '--java_interop',
      '--kotlin_box_oneofs_min_size=8',
      'all_types.proto',
      'deprecated.proto',
      'external_message.proto',
      'foreign.proto',
      'form.proto',
      'map.proto',
      'no_fields.proto',
      'one_of.proto',
      'option_redacted.proto',
      'packed_encoding.proto',
      'percents_in_kdoc.proto',
      'person.proto',
      'resursive_map.proto',
      'redacted_one_of.proto',
      'repeated.proto',
      'same_name_enum.proto',
      'simple_message.proto',
      'unknown_fields.proto',
      'uses_any.proto',
  ]
}

// KOTLIN SERIALIZATION

task generateKotlinSerializationTests(type: JavaExec) {
  group = 'Generate Tests'
  description = 'Generates Kotlin classes for kotlin.serialization'
  classpath = configurations.wire
  main = 'com.squareup.wire.WireCompiler'
  args = [
          '--proto_path=wire-library/wire-tests/src/commonTest/proto/kotlin-serialization',
          '--kotlin_out=wire-library/wire-tests/src/commonTest/proto-kotlin-serialization',
          '--emit_kotlin_serialization_UNSUPPORTED',
          'cdn_resource.proto',
  ]
}

// SWIFT

task generateSwiftTests(type: JavaExec) {
  def swiftOut = 'wire-library/wire-tests-swift/src/main/swift'
  doFirst {
    def outFile = file(swiftOut)
    outFile.deleteDir()
    outFile.mkdir()
  }

  group = 'Generate Tests'
  description = 'Generates Swift classes from the test protos'
  classpath = configurations.wire
  main = 'com.squareup.wire.WireCompiler'
  args = [
      '--proto_path=wire-library/wire-tests/src/commonTest/proto/kotlin',
      "--swift_out=$swiftOut",
      'all_types.proto',
      'custom_options.proto',
      'deprecated.proto',
      'deprecated_enum.proto',
//      'edge_cases.proto',
      'external_message.proto',
      'foreign.proto',
      'form.proto',
//      'keyword.proto',
      'map.proto',
      'no_fields.proto',
      'one_of.proto',
      'option_redacted.proto',
      'optional_enum.proto',
      'packed_encoding.proto',
      'percents_in_kdoc.proto',
      'person.proto',
      'redacted_one_of.proto',
//      'redacted_test.proto',
      'resursive_map.proto',
      'same_name_enum.proto',
//      'simple_message.proto',
      'to_string.proto',
      'unknown_fields.proto',
//      'uses_any.proto',
  ]
}

// GRPC

def GRPC_PROTOS = Files.find(
      Paths.get("${rootProject.rootDir}/wire-grpc-tests/src/test/proto"),
      Integer.MAX_VALUE,
      { path, attrs -> attrs.isRegularFile() }
    )
    .map { path -> Paths.get("${rootProject.rootDir}/wire-grpc-tests/src/test/proto").relativize(path) }
    .filter { path -> !path.startsWith("kotlin") }
    .map { path -> path.toString() }
    .filter { path -> path.endsWith(".proto") }
    .sorted()
    .collect(Collectors.toList())

task generateGrpcTests(type: JavaExec) {
  group = 'Generate gRPC Tests'
  description = 'Generates Kotlin classes from the gRPC test protos'
  classpath = configurations.wire
  main = 'com.squareup.wire.WireCompiler'
  args = [
          '--proto_path=wire-grpc-tests/src/test/proto',
          '--kotlin_out=wire-grpc-tests/src/test/proto-grpc',
          'routeguide/RouteGuideProto.proto'
  ] + GRPC_PROTOS
}

// Moshi Adapter

task generateMoshiAdapterCompactJavaTests(type: JavaExec) {
  group = 'Generate MoshiAdapter Compact Java Tests'
  description = 'Generates compat Java classes from the MoshiAdapter test protos'
  classpath = configurations.wire
  main = 'com.squareup.wire.WireCompiler'
  args = [
          '--proto_path=wire-library/wire-tests/src/commonTest/shared/proto/proto2',
          '--java_out=wire-library/wire-moshi-adapter/src/test/java',
          '--compact',
          'all_types_proto2.proto'
  ]
}

task generateMoshiAdapterJavaTests(type: JavaExec) {
  group = 'Generate MoshiAdapter Java Tests'
  description = 'Generates Java classes from the MoshiAdapter test protos'
  classpath = configurations.wire
  main = 'com.squareup.wire.WireCompiler'
  args = [
          '--proto_path=wire-library/wire-tests/src/commonTest/shared/proto',
          '--proto_path=wire-library/wire-tests/src/commonTest/shared/proto/proto2',
          '--java_out=wire-library/wire-moshi-adapter/src/test/java',
          'collection_types.proto',
          'person_java.proto',
          'dinosaur_java.proto',
          'period_java.proto',
          'keyword_java.proto',
  ]
}

task generateMoshiAdapterInteropKotlinTests(type: JavaExec) {
  group = 'Generate MoshiAdapter Java interop Kotlin Tests'
  description = 'Generates Java interop Kotlin classes from the MoshiAdapter test protos'
  classpath = configurations.wire
  main = 'com.squareup.wire.WireCompiler'
  args = [
          '--proto_path=wire-library/wire-tests/src/commonTest/shared/proto',
          '--proto_path=wire-library/wire-tests/src/commonTest/shared/proto/proto2',
          '--kotlin_out=wire-library/wire-moshi-adapter/src/test/java',
          '--java_interop',
          'person_java_interop_kotlin.proto',
          'dinosaur_java_interop_kotlin.proto',
          'period_java_interop_kotlin.proto',
          'keyword_kotlin.proto',
  ]
}

task generateMoshiAdapterKotlinTests(type: JavaExec) {
  group = 'Generate MoshiAdapter Kotlin Tests'
  description = 'Generates Kotlin classes from the MoshiAdapter test protos'
  classpath = configurations.wire
  main = 'com.squareup.wire.WireCompiler'
  args = [
          '--proto_path=wire-library/wire-tests/src/commonTest/shared/proto',
          '--proto_path=wire-library/wire-tests/src/commonTest/shared/proto/proto2',
          '--kotlin_out=wire-library/wire-moshi-adapter/src/test/java',
          'person_kotlin.proto',
          'dinosaur_kotlin.proto',
          'period_kotlin.proto',
  ]
}

task generateMoshiTests(dependsOn: [generateMoshiAdapterCompactJavaTests,
    generateMoshiAdapterJavaTests, generateMoshiAdapterInteropKotlinTests,
    generateMoshiAdapterKotlinTests]) {
  group = 'Generate Tests'
  description = 'Generates Java and Kotlin classes that use Moshi from the test protos'
}

// Shared Json tests.

task generateSharedJsonCompactJavaTests(type: JavaExec) {
  group = 'Generate Shared JSON Tests'
  description = 'Generates compat Java classes for shared JSON tests'
  classpath = configurations.wire
  main = 'com.squareup.wire.WireCompiler'
  args = [
          '--proto_path=wire-library/wire-tests/src/commonTest/shared/proto/proto2',
          '--proto_path=wire-library/wire-tests/src/commonTest/shared/proto/proto3',
          '--java_out=wire-library/wire-tests/src/jvmJsonJavaTest/proto-java',
          '--compact',
          'all32.proto',
          'all64.proto',
          'all_types_proto2.proto',
          'all_types_proto3_test_proto3_optional.proto',
          'all_structs.proto',
          'all_wrappers.proto',
          'camel_case.proto',
          'map_types.proto',
          'pizza.proto',
  ]
}

task generateSharedJsonKotlinTests(type: JavaExec) {
  group = 'Generate Shared JSON Tests'
  description = 'Generates Kotlin classes for shared JSON tests'
  classpath = configurations.wire
  main = 'com.squareup.wire.WireCompiler'
  args = [
          '--proto_path=wire-library/wire-tests/src/commonTest/shared/proto/proto2',
          '--proto_path=wire-library/wire-tests/src/commonTest/shared/proto/proto3',
          '--kotlin_out=wire-library/wire-tests/src/jvmJsonKotlinTest/proto-kotlin',
          '--java_interop',
          'all32.proto',
          'all64.proto',
          'all_types_proto2.proto',
          'all_types_proto3_test_proto3_optional.proto',
          'all_structs.proto',
          'all_wrappers.proto',
          'camel_case.proto',
          'map_types.proto',
          'pizza.proto',
  ]
}

task generateSharedJson(dependsOn: [generateSharedJsonCompactJavaTests,
    generateSharedJsonKotlinTests]) {
  group = 'Generate Tests'
  description = 'Generates Java and Kotlin classes for shared JSON tests'
}

task generateTests(dependsOn: [generateJavaTests, generateCompactTests, generateNoOptionsTests,
    generateGsonTests, generateIncludesExcludesTests, generateAndroidTests,
    generateAndroidCompactTests, generateKotlinTests, generateKotlinServicesTests,
    generateKotlinAndroidTests, generateKotlinJavaInteropTests, generateSwiftTests,
    generateGrpcTests, generateMoshiTests, generateProto3KotlinTests, generateProto3JavaTests,
    generateSharedJson, generateKotlinSerializationTests]) {
  group = 'Generate Tests'
  description = 'Generates all test classes'
}

task cleanGeneratedTests(type: Delete) {
  group = 'Generate Tests'
  description = 'Delete all generated tests'
  delete 'wire-grpc-tests/src/test/proto-grpc',
         'wire-library/wire-gson-support/src/test/java/com/squareup/wire/protos',
         'wire-library/wire-gson-support/src/test/java/squareup/proto2/keywords/',
         'wire-library/wire-moshi-adapter/src/test/java/com/squareup/wire/protos',
         'wire-library/wire-moshi-adapter/src/test/java/squareup/proto2/keywords/',
         'wire-library/wire-tests/src/commonTest/proto-java',
         'wire-library/wire-tests/src/commonTest/proto-kotlin',
         'wire-library/wire-tests/src/jvmJavaAndroidCompactTest/proto-java',
         'wire-library/wire-tests/src/jvmJavaAndroidTest/proto-java',
         'wire-library/wire-tests/src/jvmJavaCompactTest/proto-java',
         'wire-library/wire-tests/src/jvmJavaNoOptionsTest/proto-java',
         'wire-library/wire-tests/src/jvmJavaPrunedTest/proto-java',
         'wire-library/wire-tests/src/jvmJavaTest/proto-java',
         'wire-library/wire-tests/src/jvmKotlinAndroidTest/proto-kotlin',
         'wire-library/wire-tests/src/jvmKotlinInteropTest/proto-kotlin',
         'wire-library/wire-tests/src/jvmJsonJavaTest/proto-java',
         'wire-library/wire-tests/src/jvmJsonKotlinTest/proto-kotlin'
}