/*
    * Copyright 2012-2021 CodeLibs Project and the Others.
    *
    * 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 org.codelibs.fess.app.job;
  
  import javax.annotation.Resource;
  
  import org.apache.logging.log4j.LogManager;
  import org.apache.logging.log4j.Logger;
  import org.codelibs.core.lang.StringUtil;
  import org.codelibs.core.timer.TimeoutManager;
  import org.codelibs.fess.Constants;
  import org.codelibs.fess.app.logic.AccessContextLogic;
  import org.codelibs.fess.app.service.ScheduledJobService;
  import org.codelibs.fess.es.config.exbhv.JobLogBhv;
  import org.codelibs.fess.helper.JobHelper;
  import org.codelibs.fess.helper.SystemHelper;
  import org.codelibs.fess.mylasta.direction.FessConfig;
  import org.codelibs.fess.util.ComponentUtil;
  import org.dbflute.optional.OptionalThing;
  import org.lastaflute.core.time.TimeManager;
  import org.lastaflute.job.LaCron;
  import org.lastaflute.job.LaJob;
  import org.lastaflute.job.LaJobRunner;
  import org.lastaflute.job.LaJobScheduler;
  
  public class AllJobScheduler implements LaJobScheduler {
  
      private static final Logger logger = LogManager.getLogger(AllJobScheduler.class);
  
      protected static final String APP_TYPE = "JOB";
  
      @Resource
      private TimeManager timeManager;
  
      @Resource
      private FessConfig fessConfig;
  
      @Resource
      private AccessContextLogic accessContextLogic;
  
      @Resource
      private ScheduledJobService scheduledJobService;
  
      @Resource
      private SystemHelper systemHelper;
  
      @Resource
      private JobHelper jobHelper;
  
      protected Class<? extends LaJob> jobClass = ScriptExecutorJob.class;
  
      protected long schedulerTime;
  
      @Override
      public void schedule(final LaCron cron) {
          schedulerTime = System.currentTimeMillis();
          scheduledJobService.start(cron);
  
          final String myName = fessConfig.getSchedulerTargetName();
          if (StringUtil.isNotBlank(myName)) {
              ComponentUtil.getComponent(JobLogBhv.class).queryDelete(cb -> {
                  cb.query().setJobStatus_Equal(Constants.RUNNING);
                  cb.query().setTarget_Equal(myName);
              });
          }
  
          TimeoutManager.getInstance().addTimeoutTarget(() -> {
              if (logger.isDebugEnabled()) {
                  logger.debug("Updating scheduled jobs. time:{}", schedulerTime);
              }
              final long now = System.currentTimeMillis();
              scheduledJobService.getScheduledJobListAfter(schedulerTime).forEach(scheduledJob -> {
                  if (logger.isDebugEnabled()) {
                      logger.debug("Updating job schedule:{}", scheduledJob.getName());
                  }
                  try {
                      jobHelper.register(scheduledJob);
                  } catch (final Exception e) {
                      logger.warn("Failed to update schdule {}", scheduledJob, e);
                  }
              });
              schedulerTime = now;
          }, fessConfig.getSchedulerMonitorIntervalAsInteger(), true);
      }
  
      @Override
     public LaJobRunner createRunner() {
         return new LaJobRunner().useAccessContext(
                 resource -> accessContextLogic.create(resource, OptionalThing::empty, OptionalThing::empty, () -> APP_TYPE));
     }
 
     public void setJobClass(final Class<? extends LaJob> jobClass) {
         this.jobClass = jobClass;
     }
 }