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.ioc.Inject;
  import org.b3log.latke.repository.*;
  import org.b3log.latke.service.annotation.Service;
  import org.b3log.symphony.model.Article;
  import org.b3log.symphony.model.Comment;
  import org.b3log.symphony.model.Vote;
  import org.b3log.symphony.repository.ArticleRepository;
  import org.b3log.symphony.repository.CommentRepository;
  import org.b3log.symphony.repository.VoteRepository;
  import org.json.JSONObject;
  
  import java.util.ArrayList;
  import java.util.List;
  
  /**
   * Vote query service.
   *
   * @author <a href="http://88250.b3log.org">Liang Ding</a>
   * @version 1.1.0.1, Oct 17, 2016
   * @since 1.3.0
   */
  @Service
  public class VoteQueryService {
  
      /**
       * Logger.
       */
      private static final Logger LOGGER = LogManager.getLogger(VoteQueryService.class);
  
      /**
       * Vote repository.
       */
      @Inject
      private VoteRepository voteRepository;
  
      /**
       * Article repository.
       */
      @Inject
      private ArticleRepository articleRepository;
  
      /**
       * Comment repository.
       */
      @Inject
      private CommentRepository commentRepository;
  
      /**
       * Determines whether the specified user dose vote the specified entity.
       *
       * @param userId the specified user id
       * @param dataId the specified entity id
       * @return voted type, returns {@code -1} if has not voted yet
       */
      public int isVoted(final String userId, final String dataId) {
          try {
              final List<Filter> filters = new ArrayList<>();
              filters.add(new PropertyFilter(Vote.USER_ID, FilterOperator.EQUAL, userId));
              filters.add(new PropertyFilter(Vote.DATA_ID, FilterOperator.EQUAL, dataId));
              final Query query = new Query().setFilter(new CompositeFilter(CompositeFilterOperator.AND, filters));
              final JSONObject vote = voteRepository.getFirst(query);
              if (null == vote) {
                  return -1;
              }
  
              return vote.optInt(Vote.TYPE);
          } catch (final RepositoryException e) {
              LOGGER.log(Level.ERROR, e.getMessage());
              return -1;
          }
      }
  
      /**
       * Determines whether the specified data dose belong to the specified user.
       *
       * @param userId   the specified user id
       * @param dataId   the specified data id
       * @param dataType the specified data type
       * @return {@code true} if it belongs to the user, otherwise returns {@code false}
       */
      public boolean isOwn(final String userId, final String dataId, final int dataType) {
          try {
              if (Vote.DATA_TYPE_C_ARTICLE == dataType) {
                  final JSONObject article = articleRepository.get(dataId);
                  if (null == article) {
                      LOGGER.log(Level.ERROR, "Not found article [id={}]", dataId);
  
                      return false;
                  }
  
                  return article.optString(Article.ARTICLE_AUTHOR_ID).equals(userId);
              } else if (Vote.DATA_TYPE_C_COMMENT == dataType) {
                  final JSONObject comment = commentRepository.get(dataId);
                  if (null == comment) {
                      LOGGER.log(Level.ERROR, "Not found comment [id={}]", dataId);
  
                      return false;
                  }
  
                  return comment.optString(Comment.COMMENT_AUTHOR_ID).equals(userId);
              }
  
              return false;
          } catch (final RepositoryException e) {
              LOGGER.log(Level.ERROR, e.getMessage());
  
              return false;
          }
      }
  }
  