inetsoft.spark.quickbooks.QuickbooksConfig Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of spark-quickbooks-runtime Show documentation

The runtime implementation of the Spark QuickBooks connector. This library is isolated from the rest of the connector because the QuickBooks Online libraries have dependencies that conflict with Spark and need to be on an isolated classpath.

The newest version!

/*
 * Copyright 2019 InetSoft Technology
 *
 * 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 inetsoft.spark.quickbooks;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.*;
import java.lang.invoke.MethodHandles;

/**
 * Manage the storage of the QuickBooks OAuth tokens
 */
public class QuickbooksConfig {
   private QuickbooksConfig() {
      this(null, null, -1);
   }

   private QuickbooksConfig(String accessToken,
                            String refreshToken,
                            long expiration)
   {
      this.accessToken = accessToken;
      this.refreshToken = refreshToken;
      this.expiration = expiration;
   }

   public static QuickbooksConfig readConfig(String clientId, String companyId) {
      final File file = getConfigFile(clientId, companyId);

      try(FileReader reader = new FileReader(file)) {
         final BufferedReader bufferedReader = new BufferedReader(reader);
         final String accessToken = bufferedReader.readLine();
         final String refreshToken = bufferedReader.readLine();
         long expiration = Long.parseLong(bufferedReader.readLine());
         return new QuickbooksConfig(accessToken, refreshToken, expiration);
      }
      catch(NumberFormatException | IOException e) {
         LOG.info("Could not read configuration file. Generating new tokens");

         if(file.exists() && !file.delete()) {
            LOG.warn("Configuration file could not be removed");
         }
      }

      return new QuickbooksConfig();
   }

   private static File getConfigFile(String clientId, String companyId) {
      LOG.debug("Reading OAuth config file from {}", qbLibDir);
      return new File(qbLibDir, clientId + companyId);
   }

   public QuickbooksConfig updateCredentials(long expiresIn, String accessToken, String refreshToken, String clientId,
                                             String companyId)
   {
      final long expiration = expiresIn * 1000 + System.currentTimeMillis();
      final QuickbooksConfig newConfig = new QuickbooksConfig(accessToken, refreshToken,
                                                              expiration);
      return newConfig.saveConfig(clientId, companyId);
   }

   public String getAccessToken() {
      return accessToken;
   }

   public String getRefreshToken() {
      return refreshToken;
   }

   public long getExpiration() {
      return expiration;
   }

   private QuickbooksConfig saveConfig(String clientId, String companyId) {
      final File configFile = getConfigFile(clientId, companyId);

      try(PrintWriter writer = new PrintWriter(configFile)) {
         writer.println(accessToken);
         writer.println(refreshToken);
         writer.println(expiration);
      }
      catch(IOException e) {
         final String msg = "Could not save OAuth configuration. Generate a new authorization code";
         throw new RuntimeException(msg, e);
      }

      return this;
   }

   private static final Logger LOG = LoggerFactory.getLogger(MethodHandles.lookup().lookupClass());
   private static final String qbLibDir = QuickbooksUtil.getQbLibDir();
   private final String accessToken;
   private final String refreshToken;
   private final long expiration;
}