content: /*
   * Symphony - A modern community (forum/BBS/SNS/blog) platform written in Java.
   * Copyright (C) 2012-present, b3log.org
   *
   * This program is free software: you can redistribute it and/or modify
   * it under the terms of the GNU Affero General Public License as published by
   * the Free Software Foundation, either version 3 of the License, or
   * (at your option) any later version.
   *
   * This program is distributed in the hope that it will be useful,
   * but WITHOUT ANY WARRANTY; without even the implied warranty of
   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   * GNU Affero General Public License for more details.
   *
   * You should have received a copy of the GNU Affero General Public License
   * along with this program.  If not, see <https://www.gnu.org/licenses/>.
   */
  package org.b3log.symphony.service;
  
  import org.apache.logging.log4j.Level;
  import org.apache.logging.log4j.LogManager;
  import org.apache.logging.log4j.Logger;
  import org.b3log.latke.Keys;
  import org.b3log.latke.ioc.Inject;
  import org.b3log.latke.repository.RepositoryException;
  import org.b3log.latke.repository.annotation.Transactional;
  import org.b3log.latke.service.annotation.Service;
  import org.b3log.symphony.model.Referral;
  import org.b3log.symphony.repository.ReferralRepository;
  import org.json.JSONObject;
  
  /**
   * Referral management service.
   *
   * @author <a href="http://88250.b3log.org">Liang Ding</a>
   * @version 1.0.0.0, Apr 28, 2016
   * @since 1.4.0
   */
  @Service
  public class ReferralMgmtService {
  
      /**
       * Logger.
       */
      private static final Logger LOGGER = LogManager.getLogger(ReferralMgmtService.class);
  
      /**
       * Referral repository.
       */
      @Inject
      private ReferralRepository referralRepository;
  
      /**
       * Adds or updates a referral.
       *
       * @param referral the specified referral
       */
      @Transactional
      public void updateReferral(final JSONObject referral) {
          final String dataId = referral.optString(Referral.REFERRAL_DATA_ID);
          final String ip = referral.optString(Referral.REFERRAL_IP);
  
          try {
              JSONObject record = referralRepository.getByDataIdAndIP(dataId, ip);
              if (null == record) {
                  record = new JSONObject();
                  record.put(Referral.REFERRAL_AUTHOR_HAS_POINT, false);
                  record.put(Referral.REFERRAL_CLICK, 1);
                  record.put(Referral.REFERRAL_DATA_ID, dataId);
                  record.put(Referral.REFERRAL_IP, ip);
                  record.put(Referral.REFERRAL_TYPE, referral.optInt(Referral.REFERRAL_TYPE));
                  record.put(Referral.REFERRAL_USER, referral.optString(Referral.REFERRAL_USER));
                  record.put(Referral.REFERRAL_USER_HAS_POINT, false);
  
                  referralRepository.add(record);
              } else {
                  final String currentReferralUser = referral.optString(Referral.REFERRAL_USER);
                  final String firstReferralUser = record.optString(Referral.REFERRAL_USER);
                  if (!currentReferralUser.equals(firstReferralUser)) {
                      return;
                  }
  
                  record.put(Referral.REFERRAL_CLICK, record.optInt(Referral.REFERRAL_CLICK) + 1);
  
                  referralRepository.update(record.optString(Keys.OBJECT_ID), record);
              }
          } catch (final RepositoryException e) {
              LOGGER.log(Level.ERROR, "Updates a referral failed", e);
          }
      }
  }
  