diff --git a/Templates/Java/BaseJavaModel.template.tt b/Templates/Java/BaseJavaModel.template.tt index a974c5d63..eb4cd518a 100644 --- a/Templates/Java/BaseJavaModel.template.tt +++ b/Templates/Java/BaseJavaModel.template.tt @@ -567,6 +567,45 @@ return sb.ToString(); } + public string CreatePackageDefIBaseMethodRequestBuilder(CustomT4Host host) + { + var sb = new StringBuilder(); + sb.Append(CreatePackageDefinition(host)); + var importFormat = @"import {0}.{1}.{2};"; + sb.AppendFormat(importFormat, + host.CurrentModel.NamespaceName(), + "requests.extensions", + ITypeRequest(host.CurrentType)); + sb.Append("\n"); + return sb.ToString(); + } + + public string CreatePackageDefForBaseEntityCollectionPage(CustomT4Host host) + { + var sb = new StringBuilder(); + sb.Append(CreatePackageDefinition(host)); + + var importFormat = @"import {0}.{1}.{2};"; + string modelClassName; + if (host.CurrentType is OdcmMethod) + modelClassName = OdcmMethodReturnType(host.CurrentType as OdcmMethod); + else + modelClassName = TypeName(host.CurrentType); + + + sb.AppendFormat(importFormat, + host.CurrentModel.NamespaceName(), + "models.extensions", + modelClassName); + sb.Append("\n"); + sb.AppendFormat(importFormat, + host.CurrentModel.NamespaceName(), + "requests.extensions", + ITypeCollectionRequestBuilder(host.CurrentType)); + + return sb.ToString(); + } + //Fixing package and import statement for model classes public string CreatePackageDefForEntity(CustomT4Host host) { diff --git a/Templates/Java/requests_generated/BaseEntityCollectionPage.java.tt b/Templates/Java/requests_generated/BaseEntityCollectionPage.java.tt index befb77f54..b81d0d03d 100644 --- a/Templates/Java/requests_generated/BaseEntityCollectionPage.java.tt +++ b/Templates/Java/requests_generated/BaseEntityCollectionPage.java.tt @@ -4,10 +4,8 @@ <#@ output extension="\\" #> <#=writer.WriteHeader()#> <#=CreatAutogeneratedWarning()#> -<#=CreatePackageDef(host)#> - -import com.google.gson.JsonObject; -import com.google.gson.annotations.*; +<#=CreatePackageDefForBaseEntityCollectionPage(host)#> +import com.microsoft.graph.http.BaseCollectionPage; <#=CreateClassDef(BaseTypeCollectionPage(c), "BaseCollectionPage" + CollectionPageGeneric(c), IBaseTypeCollectionPage(c))#> <# diff --git a/Templates/Java/requests_generated/IBaseMethodRequestBuilder.java.tt b/Templates/Java/requests_generated/IBaseMethodRequestBuilder.java.tt index 2bbeab20e..9cc6aeb52 100644 --- a/Templates/Java/requests_generated/IBaseMethodRequestBuilder.java.tt +++ b/Templates/Java/requests_generated/IBaseMethodRequestBuilder.java.tt @@ -4,7 +4,9 @@ <#@ output extension="\\" #> <#host.TemplateName = IBaseTypeRequestBuilder(c);#> <#=writer.WriteHeader()#> -<#=CreatePackageDef(host)#> +<#=CreatePackageDefIBaseMethodRequestBuilder(host)#> +import com.microsoft.graph.http.IRequestBuilder; +import com.microsoft.graph.options.Option; <#=CreateInterfaceDef(IBaseTypeRequestBuilder(c), "IRequestBuilder")#>