/*
    * Licensed to the Apache Software Foundation (ASF) under one
    * or more contributor license agreements.  See the NOTICE file
    * distributed with this work for additional information
    * regarding copyright ownership.  The ASF licenses this file
    * to you 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 org.eclipse.aether.named.redisson;
  
  import java.io.IOException;
  import java.io.InputStream;
  import java.nio.file.Files;
  import java.nio.file.Path;
  import java.nio.file.Paths;
  
  import org.eclipse.aether.named.support.NamedLockFactorySupport;
  import org.redisson.Redisson;
  import org.redisson.api.RedissonClient;
  import org.redisson.config.Config;
  
  /**
   * Support class for factories using {@link RedissonClient}.
   */
  public abstract class RedissonNamedLockFactorySupport extends NamedLockFactorySupport {
      protected static final String NAME_PREFIX = "maven:resolver:";
  
      private static final String DEFAULT_CONFIG_FILE_NAME = "maven-resolver-redisson.yaml";
  
      private static final String DEFAULT_CLIENT_NAME = "maven-resolver";
  
      /**
       * Path to a Redisson configuration file in YAML format. Read official documentation for details.
       *
       * @since 1.7.0
       * @configurationSource {@link System#getProperty(String, String)}
       * @configurationType {@link java.lang.String}
       */
      public static final String SYSTEM_PROP_CONFIG_FILE = "aether.syncContext.named.redisson.configFile";
  
      /**
       * Address of the Redis instance. Optional.
       *
       * @since 2.0.0
       * @configurationSource {@link System#getProperty(String, String)}
       * @configurationType {@link java.lang.String}
       * @configurationDefaultValue {@link #DEFAULT_REDIS_ADDRESS}
       */
      public static final String SYSTEM_PROP_REDIS_ADDRESS = "aether.syncContext.named.redisson.address";
  
      public static final String DEFAULT_REDIS_ADDRESS = "redis://localhost:6379";
  
      protected final RedissonClient redissonClient;
  
      public RedissonNamedLockFactorySupport() {
          this.redissonClient = createRedissonClient();
      }
  
      @Override
      protected void doShutdown() {
          logger.trace("Shutting down Redisson client with id '{}'", redissonClient.getId());
          redissonClient.shutdown();
      }
  
      private RedissonClient createRedissonClient() {
          Path configFilePath = null;
  
          String configFile = System.getProperty(SYSTEM_PROP_CONFIG_FILE);
          if (configFile != null && !configFile.isEmpty()) {
              configFilePath = Paths.get(configFile);
              if (Files.notExists(configFilePath)) {
                  throw new IllegalArgumentException(
                          "The specified Redisson config file does not exist: " + configFilePath);
              }
          }
  
          if (configFilePath == null) {
              String mavenConf = System.getProperty("maven.conf");
              if (mavenConf != null && !mavenConf.isEmpty()) {
                  configFilePath = Paths.get(mavenConf, DEFAULT_CONFIG_FILE_NAME);
                  if (Files.notExists(configFilePath)) {
                      configFilePath = null;
                  }
              }
          }
  
          Config config;
  
          if (configFilePath != null) {
             logger.trace("Reading Redisson config file from '{}'", configFilePath);
             try (InputStream is = Files.newInputStream(configFilePath)) {
                 config = Config.fromYAML(is);
             } catch (IOException e) {
                 throw new IllegalStateException("Failed to read Redisson config file: " + configFilePath, e);
             }
         } else {
             config = new Config();
             String defaultRedisAddress = System.getProperty(SYSTEM_PROP_REDIS_ADDRESS, DEFAULT_REDIS_ADDRESS);
             config.useSingleServer().setAddress(defaultRedisAddress).setClientName(DEFAULT_CLIENT_NAME);
         }
 
         RedissonClient redissonClient = Redisson.create(config);
         logger.trace("Created Redisson client with id '{}'", redissonClient.getId());
 
         return redissonClient;
     }
 }