Move addRequiredImports into SharedCodegen

Author: macisamuele

gradle-plugin/plugin/src/main/java/com/yelp/codegen/KotlinGenerator.kt

@@ -203,7 +203,7 @@ open class KotlinGenerator : SharedCodegen() {
     }
 
     @VisibleForTesting
-    internal fun addRequiredImports(codegenModel: CodegenModel) {
+    override fun addRequiredImports(codegenModel: CodegenModel) {
         // If there are any vars, we will mark them with the @Json annotation so we have to make sure to import it
         if (codegenModel.allVars.isNotEmpty() || codegenModel.isEnum) {
             codegenModel.imports.add("com.squareup.moshi.Json")

gradle-plugin/plugin/src/main/java/com/yelp/codegen/SharedCodegen.kt

@@ -559,6 +559,12 @@ abstract class SharedCodegen : DefaultCodegen(), CodegenConfig {
      */
     protected abstract fun nullableTypeWrapper(baseType: String): String
 
+    /**
+     * Hook that allows to add the needed imports for a given [CodegenModel]
+     * This is needed as we might be modifying models in [postProcessAllModels]
+     */
+    internal abstract fun addRequiredImports(codegenModel: CodegenModel)
+
     private fun defaultListType() = typeMapping["list"] ?: ""
 
     private fun defaultMapType() = typeMapping["map"] ?: ""