package com.ld.vps.controller;

import cn.hutool.core.io.FileUtil;
import cn.hutool.json.JSONObject;
import cn.hutool.json.JSONUtil;
import com.ld.vps.bean.BaseResp;
import com.ld.vps.service.BitsPressService;
import com.ld.vps.service.UploadTaskService;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.repository.query.Param;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.ResponseBody;

@Controller
public class MainController {
  Logger logger = LoggerFactory.getLogger(MainController.class);
  @Autowired
  UploadTaskService taskService;
  @Autowired
  BitsPressService bitsPressService;

  @GetMapping("/getUserList")
  @ResponseBody
  String getUserList(@RequestParam("count") int count) {
    return bitsPressService.getTopQueueItem(count);
  }

  @PostMapping("/uploadFile")
  @ResponseBody
  BaseResp<Long> uploadFile(
      @RequestParam("file") String file,
      @RequestParam(value = "userId", required = false) String userId) {
    if (!FileUtil.isFile(file)) {
      BaseResp<Long> resp = new BaseResp<>(0L);
      resp.setMsg("文件不存在：" + file);
      resp.setSuccess(false);
      return resp;
    }
    Long id = taskService.createTask(file, userId == null ? "" : userId);
    return new BaseResp<>(id);
  }

  @GetMapping("/refreshToken")
  @ResponseBody
  BaseResp<String> refreshToken(@Param("token") String token) {
    bitsPressService.refreshToken(token);
    return new BaseResp<>(bitsPressService.getToken());
  }
}