content: /*
   * Solo - A small and beautiful blogging system written in Java.
   * Copyright (c) 2010-present, b3log.org
   *
   * Solo is licensed under Mulan PSL v2.
   * You can use this software according to the terms and conditions of the Mulan PSL v2.
   * You may obtain a copy of Mulan PSL v2 at:
   *         http://license.coscl.org.cn/MulanPSL2
   * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
   * See the Mulan PSL v2 for more details.
   */
  package org.b3log.solo.repository;
  
  import org.apache.commons.lang3.time.DateUtils;
  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.*;
  import org.b3log.latke.repository.annotation.Repository;
  import org.b3log.solo.model.ArchiveDate;
  import org.json.JSONObject;
  
  import java.text.ParseException;
  import java.util.List;
  
  /**
   * Archive date repository.
   *
   * @author <a href="http://88250.b3log.org">Liang Ding</a>
   * @version 1.0.0.5, Sep 11, 2019
   * @since 0.3.1
   */
  @Repository
  public class ArchiveDateRepository extends AbstractRepository {
  
      /**
       * Logger.
       */
      private static final Logger LOGGER = LogManager.getLogger(ArchiveDateRepository.class);
  
      /**
       * Archive date-Article repository.
       */
      @Inject
      private ArchiveDateArticleRepository archiveDateArticleRepository;
  
      /**
       * Public constructor.
       */
      public ArchiveDateRepository() {
          super(ArchiveDate.ARCHIVE_DATE.toLowerCase());
      }
  
      /**
       * Gets an archive date by the specified archive date string.
       *
       * @param archiveDate the specified archive date stirng (yyyy/MM)
       * @return an archive date, {@code null} if not found
       * @throws RepositoryException repository exception
       */
      public JSONObject getByArchiveDate(final String archiveDate) throws RepositoryException {
          long time;
          try {
              time = DateUtils.parseDate(archiveDate, new String[]{"yyyy/MM"}).getTime();
          } catch (final ParseException e) {
              return null;
          }
  
          LOGGER.log(Level.TRACE, "Archive date [{}] parsed to time [{}]", archiveDate, time);
  
          Query query = new Query().setFilter(new PropertyFilter(ArchiveDate.ARCHIVE_TIME, FilterOperator.EQUAL, time)).setPageCount(1);
          List<JSONObject> result = getList(query);
          if (result.isEmpty()) {
              // Try to fix wired timezone issue: https://github.com/b3log/solo/issues/12435
              try {
                  time = DateUtils.parseDate(archiveDate, new String[]{"yyyy/MM"}).getTime();
                  time += 60 * 1000 * 60 * 8;
              } catch (final ParseException e) {
                  return null;
              }
  
              LOGGER.log(Level.TRACE, "Fix archive date [{}] parsed to time [{}]", archiveDate, time);
              query = new Query().setFilter(new PropertyFilter(ArchiveDate.ARCHIVE_TIME, FilterOperator.EQUAL, time)).setPageCount(1);
              result = getList(query);
              if (result.isEmpty()) {
                  return null;
              }
          }
          return result.get(0);
      }
  
      /**
       * Get archive dates.
       *
       * @return a list of archive date, returns an empty list if not found
       * @throws RepositoryException repository exception
       */
      public List<JSONObject> getArchiveDates() throws RepositoryException {
          final Query query = new Query().addSort(ArchiveDate.ARCHIVE_TIME, SortDirection.DESCENDING).setPageCount(1);
          final List<JSONObject> ret = getList(query);
          for (final JSONObject archiveDate : ret) {
              final String archiveDateId = archiveDate.optString(Keys.OBJECT_ID);
              final int publishedArticleCount = archiveDateArticleRepository.getPublishedArticleCount(archiveDateId);
              archiveDate.put(ArchiveDate.ARCHIVE_DATE_T_PUBLISHED_ARTICLE_COUNT, publishedArticleCount);
          }
  
          return ret;
      }
  }
  