/*
 * Copyright (C) 2016 AriaLyy(https://github.com/AriaLyy/Aria)
 *
 * 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 com.arialyy.aria.core.inf;

import android.content.Context;
import android.os.Handler;
import android.os.Parcelable;
import android.text.TextUtils;
import com.arialyy.aria.core.common.IUtil;
import com.arialyy.aria.util.ALog;
import com.arialyy.aria.util.CommonUtil;
import java.util.HashMap;
import java.util.Map;

/**
 * Created by AriaL on 2017/6/29.
 */
public abstract class AbsTask<ENTITY extends AbsEntity, TASK_ENTITY extends AbsTaskEntity>
    implements ITask<TASK_ENTITY> {
  public static final String ERROR_INFO_KEY = "ERROR_INFO_KEY";

  /**
   * 是否需要重试，默认为true
   */
  public boolean needRetry = true;
  protected TASK_ENTITY mTaskEntity;
  protected Handler mOutHandler;
  protected Context mContext;
  boolean isHeighestTask = false;
  private boolean isCancel = false, isStop = false;
  protected IUtil mUtil;
  /**
   * 扩展信息
   */
  private Map<String, Object> mExpand = new HashMap<>();
  /**
   * 该任务的调度类型
   */
  @TaskSchedulerType
  private int mSchedulerType = TaskSchedulerType.TYPE_DEFAULT;
  protected IEventListener mListener;
  protected ENTITY mEntity;
  protected String TAG;

  protected AbsTask() {
    TAG = CommonUtil.getClassName(this);
  }

  public Handler getOutHandler() {
    return mOutHandler;
  }

  /**
   * 添加扩展数据 读取扩展数据{@link #getExpand(String)}
   */
  public void putExpand(String key, Object obj) {
    if (TextUtils.isEmpty(key)) {
      ALog.e(TAG, "key 为空");
      return;
    } else if (obj == null) {
      ALog.e(TAG, "扩展数据为空");
      return;
    }
    mExpand.put(key, obj);
  }

  /**
   * 读取扩展数据
   */
  public Object getExpand(String key) {
    return mExpand.get(key);
  }

  /**
   * 设置最大下载/上传速度
   *
   * @param speed 单位为：kb
   */
  public void setMaxSpeed(int speed) {
    if (mUtil != null) {
      mUtil.setMaxSpeed(speed);
    }
  }

  /**
   * 任务是否完成
   *
   * @return {@code true} 已经完成，{@code false} 未完成
   */
  public boolean isComplete() {
    return mTaskEntity.getEntity().isComplete();
  }

  /**
   * 获取当前下载进度
   */
  @Override public long getCurrentProgress() {
    return mTaskEntity.getEntity().getCurrentProgress();
  }

  /**
   * 获取单位转换后的进度
   *
   * @return 如：已经下载3mb的大小，则返回{@code 3mb}
   */
  @Override public String getConvertCurrentProgress() {
    if (mTaskEntity.getEntity().getCurrentProgress() == 0) {
      return "0b";
    }
    return CommonUtil.formatFileSize(mTaskEntity.getEntity().getCurrentProgress());
  }

  /**
   * 转换单位后的文件长度
   *
   * @return 如果文件长度为0，则返回0m，否则返回转换后的长度1b、1kb、1mb、1gb、1tb
   */
  @Override public String getConvertFileSize() {
    if (mTaskEntity.getEntity().getFileSize() == 0) {
      return "0mb";
    }
    return CommonUtil.formatFileSize(mTaskEntity.getEntity().getFileSize());
  }

  /**
   * 获取文件大小
   */
  @Override public long getFileSize() {
    return mTaskEntity.getEntity().getFileSize();
  }

  /**
   * 获取百分比进度
   *
   * @return 返回百分比进度，如果文件长度为0，返回0
   */
  @Override public int getPercent() {
    if (mTaskEntity.getEntity().getFileSize() == 0) {
      return 0;
    }
    return (int) (mTaskEntity.getEntity().getCurrentProgress() * 100 / mTaskEntity.getEntity()
        .getFileSize());
  }

  /**
   * 任务当前状态
   *
   * @return {@link IEntity}
   */
  public int getState() {
    return mTaskEntity.getEntity() == null ? IEntity.STATE_OTHER
        : mTaskEntity.getEntity().getState();
  }

  /**
   * 获取保存的扩展字段
   *
   * @return 如果实体不存在，则返回null，否则返回扩展字段
   */
  public String getExtendField() {
    return mTaskEntity.getEntity() == null ? null : mTaskEntity.getEntity().getStr();
  }

  @Override public void start() {
    if (mUtil.isRunning()) {
      ALog.d(TAG, "任务正在下载");
    } else {
      mUtil.start();
    }
  }

  @Override public void stop() {
    stop(TaskSchedulerType.TYPE_DEFAULT);
  }

  @Override public void stop(@TaskSchedulerType int type) {
    isStop = true;
    mSchedulerType = type;
    if (mUtil.isRunning()) {
      mUtil.stop();
    } else {
      mListener.onStop(mEntity.getCurrentProgress());
    }
  }

  @Override public void cancel() {
    isCancel = true;
    if (!mUtil.isRunning()) {
      mListener.onCancel();
    } else {
      mUtil.cancel();
    }
  }

  /**
   * 是否真正下载
   *
   * @return {@code true} 真正下载
   */
  @Override public boolean isRunning() {
    return mUtil.isRunning();
  }

  /**
   * 任务的调度类型
   */
  public int getSchedulerType() {
    return mSchedulerType;
  }

  /**
   * 任务是否取消了
   *
   * @return {@code true}任务已经取消
   */
  public boolean isCancel() {
    return isCancel;
  }

  /**
   * 任务是否停止了
   *
   * @return {@code true}任务已经停止
   */
  public boolean isStop() {
    return isStop;
  }

  /**
   * @return 返回原始byte速度，需要你在配置文件中配置
   * <pre>
   *   {@code
   *    <xml>
   *      <download>
   *        ...
   *        <convertSpeed value="false"/>
   *      </download>
   *
   *      或在代码中设置
   *      Aria.get(this).getDownloadConfig().setConvertSpeed(false);
   *    </xml>
   *   }
   * </pre>
   * 才能生效
   */
  @Override public long getSpeed() {
    return mTaskEntity.getEntity().getSpeed();
  }

  /**
   * @return 返回转换单位后的速度，需要你在配置文件中配置，转换完成后为：1b/s、1kb/s、1mb/s、1gb/s、1tb/s
   * <pre>
   *   {@code
   *    <xml>
   *      <download>
   *        ...
   *        <convertSpeed value="true"/>
   *      </download>
   *
   *      或在代码中设置
   *      Aria.get(this).getDownloadConfig().setConvertSpeed(true);
   *    </xml>
   *   }
   * </pre>
   * 才能生效
   */
  @Override public String getConvertSpeed() {
    return mTaskEntity.getEntity().getConvertSpeed();
  }

  @Override public TASK_ENTITY getTaskEntity() {
    return mTaskEntity;
  }

  /**
   * 获取任务名，也就是文件名
   */
  public abstract String getTaskName();

  public boolean isHighestPriorityTask() {
    return isHeighestTask;
  }
}
