package com.qianwen.smartman.common.config;

import com.fasterxml.classmate.TypeResolver;
import com.github.xiaoymin.knife4j.spring.extension.OpenApiExtensionResolver;
import com.google.common.collect.Lists;
import java.lang.reflect.Type;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import com.qianwen.smartman.common.constant.MenuConstant;
import com.qianwen.smartman.common.launch.SwaggerPlugin;
import com.qianwen.core.secure.BladeUser;
import com.qianwen.core.swagger.EnableSwagger;
import com.qianwen.core.swagger.SwaggerProperties;
import com.qianwen.core.swagger.SwaggerUtil;
//import com.qianwen.smartman.modules.dmpLog.constant.DmpLogSyncConstant;

import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.plugin.core.config.EnablePluginRegistries;
import springfox.documentation.builders.ApiInfoBuilder;
import springfox.documentation.builders.PathSelectors;
import springfox.documentation.schema.AlternateTypeRule;
import springfox.documentation.schema.AlternateTypeRules;
import springfox.documentation.service.ApiInfo;
import springfox.documentation.service.AuthorizationScope;
import springfox.documentation.service.Contact;
import springfox.documentation.service.SecurityReference;
import springfox.documentation.service.SecurityScheme;
import springfox.documentation.spi.DocumentationType;
import springfox.documentation.spi.service.contexts.SecurityContext;
import springfox.documentation.spring.web.plugins.Docket;

@EnableSwagger
@Configuration
@EnablePluginRegistries({SwaggerPlugin.class})


public class SwaggerConfiguration {
    private final SwaggerProperties swaggerProperties;
    private final OpenApiExtensionResolver openApiExtensionResolver;

    public SwaggerConfiguration(final SwaggerProperties swaggerProperties, final OpenApiExtensionResolver openApiExtensionResolver) {
        this.swaggerProperties = swaggerProperties;
        this.openApiExtensionResolver = openApiExtensionResolver;
    }

    @Bean
    public Docket authDocket() {
        return docket("authentication", Collections.singletonList("com.qianwen.smartman.modules.auth"));
    }

    @Bean
    public Docket sysDocket() {
        return docket("system", Arrays.asList("com.qianwen.smartman.modules.system", "com.qianwen.smartman.modules.resource"));
    }

    @Bean
    public Docket notifyDocket() {
        return docket("notify", Arrays.asList("com.qianwen.smartman.modules.notify"));
    }

    @Bean
    public Docket cpsDocket() {
        return docket("cps", Arrays.asList("com.qianwen.smartman.modules.cps"));
    }

    @Bean
    public Docket fmsDocket() {
        return docket("fms", Arrays.asList("com.qianwen.smartman.modules.fms"));
    }

    @Bean
    public Docket mdcDocket() {
        return docket("mdc", Lists.newArrayList(new String[]{"com.qianwen.smartman.modules.mdc"}));
    }

    @Bean
    public Docket tpmDocket() {
        return docket("tpm", Lists.newArrayList(new String[]{"com.qianwen.smartman.modules.tpm"}));
    }

    @Bean
    public Docket dncDocket() {
        return docket("dnc", Lists.newArrayList(new String[]{"com.qianwen.smartman.modules.dnc"}));
    }

    @Bean
    public Docket visualDocket() {
        return docket("visual", Lists.newArrayList(new String[]{"com.qianwen.smartman.modules.visual"}));
    }

    @Bean
    public Docket deskDocket() {
        return docket(MenuConstant.DESK, Collections.singletonList("com.qianwen.smartman.modules.desk"));
    }

    @Bean
    public Docket tpDocket() {
        return docket("trace", Collections.singletonList("com.qianwen.smartman.modules.trace"));
    }

    @Bean
    public Docket reportDocket() {
        return docket("report", Collections.singletonList("com.qianwen.smartman.modules.report"));
    }

    @Bean
    public Docket syncDocket() {
        return docket("sync", Collections.singletonList("com.qianwen.smartman.modules.sync"));
    }
    /*
    @Bean
    public Docket dmpLogDocket() {
        return docket(DmpLogSyncConstant.DMP_LOG_TOPIC, Collections.singletonList("com.qianwen.smartman.modules.dmpLog"));
    }
    */

    @Bean
    public Docket toolDocket() {
        return docket("tool", Collections.singletonList("com.qianwen.smartman.modules.tool"));
    }

    @Bean
    public Docket perfDocket() {
        return docket("perf", Collections.singletonList("com.qianwen.smartman.modules.perf"));
    }

    @Bean
    public Docket productionDocket() {
        return docket("coproduction", Collections.singletonList("com.qianwen.smartman.modules.coproduction"));
    }

    @Bean
    public Docket andonDocket() {
        return docket("andon", Collections.singletonList("com.qianwen.smartman.modules.andon"));
    }

    @Bean
    public Docket workinghourDocket() {
        return docket("workinghour", Collections.singletonList("com.qianwen.smartman.modules.workinghour"));
    }
    
    private Docket docket(String groupName, List<String> basePackages) {
        TypeResolver resolver = new TypeResolver();
        AlternateTypeRule listLongToString = AlternateTypeRules.newRule(resolver.resolve(List.class, new Type[]{Long.class}), resolver.resolve(List.class, new Type[]{String.class}));
        return new Docket(DocumentationType.SWAGGER_2).groupName(groupName).apiInfo(apiInfo()).ignoredParameterTypes(new Class[]{BladeUser.class}).alternateTypeRules(new AlternateTypeRule[]{listLongToString}).directModelSubstitute(Long.class, String.class).select().apis(SwaggerUtil.basePackages(basePackages)).paths(PathSelectors.any()).build().securityContexts(securityContexts()).securitySchemes(securitySchemas()).extensions(this.openApiExtensionResolver.buildExtensions(groupName));
    }

    private List<SecurityContext> securityContexts() {
        return Collections.singletonList(SecurityContext.builder().securityReferences(defaultAuth()).forPaths(PathSelectors.regex("^.*$")).build());
    }

    List<SecurityReference> defaultAuth() {
        AuthorizationScope authorizationScope = new AuthorizationScope("global", "accessEverywhere");
        AuthorizationScope[] authorizationScopes = {authorizationScope};
        return Lists.newArrayList(new SecurityReference[]{new SecurityReference(SwaggerUtil.clientInfo().getName(), authorizationScopes), new SecurityReference(SwaggerUtil.bladeAuth().getName(), authorizationScopes), new SecurityReference(SwaggerUtil.bladeTenant().getName(), authorizationScopes)});
    }

    private List<SecurityScheme> securitySchemas() {
        return Lists.newArrayList(new SecurityScheme[]{SwaggerUtil.clientInfo(), SwaggerUtil.bladeAuth(), SwaggerUtil.bladeTenant()});
    }

    private ApiInfo apiInfo() {
        return new ApiInfoBuilder().title(this.swaggerProperties.getTitle()).description(this.swaggerProperties.getDescription()).license(this.swaggerProperties.getLicense()).licenseUrl(this.swaggerProperties.getLicenseUrl()).termsOfServiceUrl(this.swaggerProperties.getTermsOfServiceUrl()).contact(new Contact(this.swaggerProperties.getContact().getName(), this.swaggerProperties.getContact().getUrl(), this.swaggerProperties.getContact().getEmail())).version(this.swaggerProperties.getVersion()).build();
    }
}