/*
    * 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.apache.chemistry.opencmis.client.bindings.spi.browser;
  
  import java.io.Serializable;
  import java.util.HashMap;
  import java.util.Map;
  import java.util.concurrent.locks.ReentrantReadWriteLock;
  
  import org.apache.chemistry.opencmis.commons.impl.Constants;
  import org.apache.chemistry.opencmis.commons.impl.UrlBuilder;
  
  /**
   * URL cache for repository and root URLs.
   */
  public class RepositoryUrlCache implements Serializable {
  
      public static final String OBJECT_ID = "objectId";
  
      private static final long serialVersionUID = 1L;
  
      private final Map<String, String> repositoryUrls;
      private final Map<String, String> rootUrls;
  
      private final ReentrantReadWriteLock lock = new ReentrantReadWriteLock();
  
      public RepositoryUrlCache() {
          repositoryUrls = new HashMap<String, String>();
          rootUrls = new HashMap<String, String>();
      }
  
      /**
       * Adds the URLs of a repository to the cache.
       */
      public void addRepository(String repositoryId, String repositoryUrl, String rootUrl) {
          if (repositoryId == null || repositoryUrl == null || rootUrl == null) {
              throw new IllegalArgumentException("Repository Id or Repository URL or Root URL is not set!");
          }
  
          lock.writeLock().lock();
          try {
              repositoryUrls.put(repositoryId, repositoryUrl);
              rootUrls.put(repositoryId, rootUrl);
          } finally {
              lock.writeLock().unlock();
          }
      }
  
      /**
       * Removes the URLs of a repository from the cache.
       */
      public void removeRepository(String repositoryId) {
          lock.writeLock().lock();
          try {
              repositoryUrls.remove(repositoryId);
              rootUrls.remove(repositoryId);
          } finally {
              lock.writeLock().unlock();
          }
      }
  
      /**
       * Returns the base repository URL of a repository.
       */
      public String getRepositoryBaseUrl(String repositoryId) {
          lock.readLock().lock();
          try {
              return repositoryUrls.get(repositoryId);
          } finally {
              lock.readLock().unlock();
          }
      }
  
      /**
       * Returns the root URL of a repository.
       */
      public String getRootUrl(String repositoryId) {
          lock.readLock().lock();
          try {
              return rootUrls.get(repositoryId);
          } finally {
              lock.readLock().unlock();
          }
     }
 
     /**
      * Returns the repository URL.
      */
     public UrlBuilder getRepositoryUrl(String repositoryId) {
         String base = getRepositoryBaseUrl(repositoryId);
         if (base == null) {
             return null;
         }
 
         return new UrlBuilder(base);
     }
 
     /**
      * Returns the repository URL with the given selector.
      */
     public UrlBuilder getRepositoryUrl(String repositoryId, String selector) {
         UrlBuilder result = getRepositoryUrl(repositoryId);
         if (result == null) {
             return null;
         }
 
         result.addParameter(Constants.PARAM_SELECTOR, selector);
 
         return result;
     }
 
     /**
      * Returns an object URL with the given selector.
      */
     public UrlBuilder getObjectUrl(String repositoryId, String objectId) {
         String root = getRootUrl(repositoryId);
         if (root == null) {
             return null;
         }
 
         UrlBuilder result = new UrlBuilder(root);
         result.addParameter(OBJECT_ID, objectId);
 
         return result;
     }
 
     /**
      * Returns an object URL with the given selector.
      */
     public UrlBuilder getObjectUrl(String repositoryId, String objectId, String selector) {
         UrlBuilder result = getObjectUrl(repositoryId, objectId);
         if (result == null) {
             return null;
         }
 
         result.addParameter(Constants.PARAM_SELECTOR, selector);
 
         return result;
     }
 
     /**
      * Returns an object URL with the given selector.
      */
     public UrlBuilder getPathUrl(String repositoryId, String path) {
         String root = getRootUrl(repositoryId);
         if (root == null) {
             return null;
         }
 
         UrlBuilder result = new UrlBuilder(root);
         result.addPath(path);
 
         return result;
     }
 
     /**
      * Returns an object URL with the given selector.
      */
     public UrlBuilder getPathUrl(String repositoryId, String path, String selector) {
         UrlBuilder result = getPathUrl(repositoryId, path);
         if (result == null) {
             return null;
         }
 
         result.addParameter(Constants.PARAM_SELECTOR, selector);
 
         return result;
     }
 }