content: /*
   * Copyright (c) 2024, Alibaba Cloud;
   * Licensed under the Apache License, Version 2.0 (the "License");
   * you may not use this file except in compliance with the License.
   * You may obtain a copy of the License at
   *
   *    http://www.apache.org/licenses/LICENSE-2.0
   *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  
  package com.aliyun.dataworks.migrationx.domain.dataworks.dolphinscheduler.v1.v139.datasource;
  
  import org.apache.commons.collections4.MapUtils;
  import org.apache.commons.lang3.StringUtils;
  import org.slf4j.Logger;
  import org.slf4j.LoggerFactory;
  
  import java.util.LinkedHashMap;
  import java.util.Map;
  
  import com.aliyun.dataworks.migrationx.domain.dataworks.dolphinscheduler.v1.v139.enums.DbType;
  
  /**
   * data source of mySQL
   */
  public class MySQLDataSource extends BaseDataSource {
  
      private static final Logger logger = LoggerFactory.getLogger(MySQLDataSource.class);
  
      private static final String ALLOW_LOAD_LOCAL_IN_FILE_NAME = "allowLoadLocalInfile";
  
      private static final String AUTO_DESERIALIZE = "autoDeserialize";
  
      private static final String ALLOW_LOCAL_IN_FILE_NAME = "allowLocalInfile";
  
      private static final String ALLOW_URL_IN_LOCAL_IN_FILE_NAME = "allowUrlInLocalInfile";
  
      private static final String APPEND_PARAMS
              = "allowLoadLocalInfile=false&autoDeserialize=false&allowLocalInfile=false&allowUrlInLocalInfile=false";
  
      private static boolean checkKeyIsLegitimate(String key) {
          return !key.contains(ALLOW_LOAD_LOCAL_IN_FILE_NAME) && !key.contains(AUTO_DESERIALIZE) && !key.contains(
                  ALLOW_LOCAL_IN_FILE_NAME) && !key.contains(ALLOW_URL_IN_LOCAL_IN_FILE_NAME);
      }
  
      /**
       * gets the JDBC url for the data source connection
       *
       * @return jdbc url
       */
      @Override
      public String driverClassSelector() {
          return Constants.COM_MYSQL_JDBC_DRIVER;
      }
  
      /**
       * @return db type
       */
      @Override
      public DbType dbTypeSelector() {
          return DbType.MYSQL;
      }
  
      public static Map<String, String> buildOtherParams(Map<String, String> paramMap) {
  
          if (MapUtils.isEmpty(paramMap)) {
              return null;
          }
          LinkedHashMap<String, String> newParamMap = new LinkedHashMap<>();
          paramMap.forEach((k, v) -> {
              if (!checkKeyIsLegitimate(k)) {
                  return;
              }
              newParamMap.put(k, v);
          });
          return newParamMap;
      }
  
      @Override
      public String getUser() {
          if (user == null) {
              return null;
          }
  
          if (user.contains(AUTO_DESERIALIZE)) {
              logger.warn("sensitive param : {} in username field is filtered", AUTO_DESERIALIZE);
              user = user.replace(AUTO_DESERIALIZE, "");
          }
          logger.debug("username : {}", user);
          return user;
      }
  
      @Override
      protected String filterOther(String otherParams) {
          if (StringUtils.isBlank(otherParams)) {
              return APPEND_PARAMS;
          }
          char symbol = '&';
          return otherParams + symbol + APPEND_PARAMS;
      }
  
      @Override
      public String getPassword() {
          // password need decode
          password = super.getPassword();
          if (password.contains(AUTO_DESERIALIZE)) {
              logger.warn("sensitive param : {} in password field is filtered", AUTO_DESERIALIZE);
              password = password.replace(AUTO_DESERIALIZE, "");
          }
          return password;
      }
  }
  