package com.zy.asrs.controller;
|
|
import com.alibaba.fastjson.JSONObject;
|
import com.baomidou.mybatisplus.mapper.EntityWrapper;
|
import com.baomidou.mybatisplus.plugins.Page;
|
import com.core.annotations.ManagerAuth;
|
import com.core.common.Cools;
|
import com.core.common.R;
|
import com.core.controller.AbstractBaseController;
|
import com.zy.asrs.entity.ApkBuildTask;
|
import com.zy.asrs.entity.TvDevice;
|
import com.zy.asrs.service.ApkBuildTaskService;
|
import com.zy.asrs.service.TvDeviceService;
|
import org.springframework.beans.factory.annotation.Value;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.web.bind.annotation.*;
|
import org.springframework.web.multipart.MultipartFile;
|
import java.nio.file.Files;
|
import java.nio.file.Path;
|
import java.nio.file.Paths;
|
import java.util.*;
|
|
/**
|
* 电视机设备Controller
|
*/
|
@RestController
|
public class TvDeviceController extends AbstractBaseController {
|
|
@Autowired
|
private TvDeviceService tvDeviceService;
|
|
@Autowired
|
private ApkBuildTaskService apkBuildTaskService;
|
|
@Value("${apk-build.apk-download-path}")
|
private String apkDownloadPath;
|
|
/**
|
* 查询单个设备
|
*/
|
@RequestMapping(value = "/tvDevice/{id}/auth")
|
@ManagerAuth
|
public R get(@PathVariable("id") Long id) {
|
return R.ok(tvDeviceService.selectById(id));
|
}
|
|
/**
|
* 分页查询设备列表
|
*/
|
@RequestMapping(value = "/tvDevice/list/auth")
|
@ManagerAuth
|
public R list(@RequestParam(defaultValue = "1") Integer curr,
|
@RequestParam(defaultValue = "10") Integer limit,
|
@RequestParam Map<String, Object> param) {
|
excludeTrash(param);
|
EntityWrapper<TvDevice> wrapper = new EntityWrapper<>();
|
|
for (Map.Entry<String, Object> entry : param.entrySet()) {
|
String key = entry.getKey();
|
Object value = entry.getValue();
|
if (value == null || "".equals(value.toString())) {
|
continue;
|
}
|
if ("status".equals(key)) {
|
wrapper.eq(key, value);
|
} else {
|
wrapper.like(key, String.valueOf(value));
|
}
|
}
|
|
wrapper.orderBy("id", false);
|
return R.ok(tvDeviceService.selectPage(new Page<>(curr, limit), wrapper));
|
}
|
|
/**
|
* 获取所有设备（不分页）
|
*/
|
@RequestMapping(value = "/tvDevice/all/auth")
|
@ManagerAuth
|
public R listAll() {
|
EntityWrapper<TvDevice> wrapper = new EntityWrapper<>();
|
wrapper.orderBy("name", true);
|
return R.ok(tvDeviceService.selectList(wrapper));
|
}
|
|
/**
|
* 获取所有在线设备
|
*/
|
@RequestMapping(value = "/tvDevice/online/auth")
|
@ManagerAuth
|
public R listOnline() {
|
return R.ok(tvDeviceService.getOnlineDevices());
|
}
|
|
/**
|
* 新增设备
|
*/
|
@RequestMapping(value = "/tvDevice/add/auth", method = RequestMethod.POST)
|
@ManagerAuth
|
public R add(@RequestBody TvDevice device) {
|
if (Cools.isEmpty(device.getName())) {
|
return R.error("设备名称不能为空");
|
}
|
if (Cools.isEmpty(device.getIp())) {
|
return R.error("设备IP不能为空");
|
}
|
|
// 检查IP是否已存在
|
EntityWrapper<TvDevice> wrapper = new EntityWrapper<>();
|
wrapper.eq("ip", device.getIp());
|
if (tvDeviceService.selectCount(wrapper) > 0) {
|
return R.error("该IP已存在");
|
}
|
|
if (device.getPort() == null) {
|
device.setPort(5555);
|
}
|
device.setStatus((short) 0);
|
device.setCreateTime(new Date());
|
device.setUpdateTime(new Date());
|
tvDeviceService.insert(device);
|
return R.ok(device);
|
}
|
|
/**
|
* 更新设备
|
*/
|
@RequestMapping(value = "/tvDevice/update/auth", method = RequestMethod.POST)
|
@ManagerAuth
|
public R update(@RequestBody TvDevice device) {
|
if (device.getId() == null) {
|
return R.error("设备ID不能为空");
|
}
|
|
// 检查IP是否已被其他设备使用
|
if (!Cools.isEmpty(device.getIp())) {
|
EntityWrapper<TvDevice> wrapper = new EntityWrapper<>();
|
wrapper.eq("ip", device.getIp()).ne("id", device.getId());
|
if (tvDeviceService.selectCount(wrapper) > 0) {
|
return R.error("该IP已被其他设备使用");
|
}
|
}
|
|
device.setUpdateTime(new Date());
|
tvDeviceService.updateById(device);
|
return R.ok();
|
}
|
|
/**
|
* 删除设备
|
*/
|
@RequestMapping(value = "/tvDevice/delete/auth", method = RequestMethod.POST)
|
@ManagerAuth
|
public R delete(Long[] ids) {
|
if (ids == null || ids.length == 0) {
|
return R.error();
|
}
|
tvDeviceService.deleteBatchIds(Arrays.asList(ids));
|
return R.ok();
|
}
|
|
/**
|
* 测试连接
|
*/
|
@RequestMapping(value = "/tvDevice/testConnection/{id}/auth", method = RequestMethod.POST)
|
@ManagerAuth
|
public R testConnection(@PathVariable("id") Long id) {
|
try {
|
String result = tvDeviceService.testConnection(id);
|
Map<String, Object> data = new HashMap<>();
|
data.put("result", result);
|
data.put("device", tvDeviceService.selectById(id));
|
return R.ok(data);
|
} catch (Exception e) {
|
return R.error("连接失败: " + e.getMessage());
|
}
|
}
|
|
/**
|
* 刷新所有设备状态
|
*/
|
@RequestMapping(value = "/tvDevice/refreshAll/auth", method = RequestMethod.POST)
|
@ManagerAuth
|
public R refreshAllStatus() {
|
try {
|
int count = tvDeviceService.refreshAllStatus();
|
Map<String, Object> data = new HashMap<>();
|
data.put("refreshedCount", count);
|
return R.ok(data);
|
} catch (Exception e) {
|
return R.error("刷新失败: " + e.getMessage());
|
}
|
}
|
|
/**
|
* 安装打包任务的APK到设备
|
*/
|
@RequestMapping(value = "/tvDevice/installFromTask/auth", method = RequestMethod.POST)
|
@ManagerAuth
|
public R installFromTask(@RequestBody JSONObject param) {
|
try {
|
Long taskId = param.getLong("taskId");
|
List<Long> deviceIds = param.getJSONArray("deviceIds").toJavaList(Long.class);
|
|
if (taskId == null) {
|
return R.error("请选择打包任务");
|
}
|
if (deviceIds == null || deviceIds.isEmpty()) {
|
return R.error("请选择安装设备");
|
}
|
|
ApkBuildTask task = apkBuildTaskService.selectById(taskId);
|
if (task == null) {
|
return R.error("打包任务不存在");
|
}
|
if (task.getApkPath() == null) {
|
return R.error("APK文件未下载，请先下载");
|
}
|
|
List<String> results = tvDeviceService.batchInstallApk(deviceIds, task.getApkPath());
|
return R.ok(results);
|
} catch (Exception e) {
|
return R.error("安装失败: " + e.getMessage());
|
}
|
}
|
|
/**
|
* 上传APK并安装到设备
|
*/
|
@RequestMapping(value = "/tvDevice/uploadAndInstall/auth", method = RequestMethod.POST)
|
@ManagerAuth
|
public R uploadAndInstall(@RequestParam("file") MultipartFile file,
|
@RequestParam("deviceIds") String deviceIdsStr) {
|
try {
|
if (file.isEmpty()) {
|
return R.error("请选择APK文件");
|
}
|
|
String originalFilename = file.getOriginalFilename();
|
if (originalFilename == null || !originalFilename.toLowerCase().endsWith(".apk")) {
|
return R.error("请上传APK文件");
|
}
|
|
// 解析设备ID
|
List<Long> deviceIds = new ArrayList<>();
|
for (String idStr : deviceIdsStr.split(",")) {
|
if (!idStr.trim().isEmpty()) {
|
deviceIds.add(Long.parseLong(idStr.trim()));
|
}
|
}
|
|
if (deviceIds.isEmpty()) {
|
return R.error("请选择安装设备");
|
}
|
|
// 保存文件
|
Path uploadDir = Paths.get(apkDownloadPath, "uploads");
|
if (!Files.exists(uploadDir)) {
|
Files.createDirectories(uploadDir);
|
}
|
|
String fileName = System.currentTimeMillis() + "_" + originalFilename;
|
Path filePath = uploadDir.resolve(fileName);
|
file.transferTo(filePath.toFile());
|
|
// 安装到设备
|
List<String> results = tvDeviceService.batchInstallApk(deviceIds, filePath.toString());
|
|
Map<String, Object> data = new HashMap<>();
|
data.put("results", results);
|
data.put("apkPath", filePath.toString());
|
return R.ok(data);
|
} catch (Exception e) {
|
return R.error("安装失败: " + e.getMessage());
|
}
|
}
|
|
/**
|
* 启动设备上的应用
|
*/
|
@RequestMapping(value = "/tvDevice/launchApp/{id}/auth", method = RequestMethod.POST)
|
@ManagerAuth
|
public R launchApp(@PathVariable("id") Long id, @RequestBody(required = false) JSONObject param) {
|
try {
|
String packageName = param != null ? param.getString("packageName") : null;
|
String result = tvDeviceService.launchApp(id, packageName);
|
Map<String, Object> data = new HashMap<>();
|
data.put("result", result);
|
data.put("device", tvDeviceService.selectById(id));
|
return R.ok(data);
|
} catch (Exception e) {
|
return R.error("启动失败: " + e.getMessage());
|
}
|
}
|
|
/**
|
* 批量启动设备上的应用
|
*/
|
@RequestMapping(value = "/tvDevice/batchLaunchApp/auth", method = RequestMethod.POST)
|
@ManagerAuth
|
public R batchLaunchApp(@RequestBody JSONObject param) {
|
try {
|
List<Long> deviceIds = param.getJSONArray("deviceIds").toJavaList(Long.class);
|
String packageName = param.getString("packageName");
|
|
if (deviceIds == null || deviceIds.isEmpty()) {
|
return R.error("请选择设备");
|
}
|
|
List<String> results = tvDeviceService.batchLaunchApp(deviceIds, packageName);
|
return R.ok(results);
|
} catch (Exception e) {
|
return R.error("启动失败: " + e.getMessage());
|
}
|
}
|
|
/**
|
* 获取设备屏幕截图
|
*/
|
@RequestMapping(value = "/tvDevice/screenshot/{id}/auth", method = RequestMethod.GET)
|
@ManagerAuth
|
public R captureScreen(@PathVariable("id") Long id) {
|
try {
|
String base64Image = tvDeviceService.captureScreen(id);
|
Map<String, Object> data = new HashMap<>();
|
data.put("image", base64Image);
|
data.put("device", tvDeviceService.selectById(id));
|
return R.ok(data);
|
} catch (Exception e) {
|
return R.error("截图失败: " + e.getMessage());
|
}
|
}
|
|
/**
|
* 重启设备上的应用
|
*/
|
@RequestMapping(value = "/tvDevice/restartApp/{id}/auth", method = RequestMethod.POST)
|
@ManagerAuth
|
public R restartApp(@PathVariable("id") Long id, @RequestBody(required = false) JSONObject param) {
|
try {
|
String packageName = param != null ? param.getString("packageName") : null;
|
String result = tvDeviceService.restartApp(id, packageName);
|
Map<String, Object> data = new HashMap<>();
|
data.put("result", result);
|
data.put("device", tvDeviceService.selectById(id));
|
return R.ok(data);
|
} catch (Exception e) {
|
return R.error("重启失败: " + e.getMessage());
|
}
|
}
|
|
/**
|
* 批量重启设备上的应用
|
*/
|
@RequestMapping(value = "/tvDevice/batchRestartApp/auth", method = RequestMethod.POST)
|
@ManagerAuth
|
public R batchRestartApp(@RequestBody JSONObject param) {
|
try {
|
List<Long> deviceIds = param.getJSONArray("deviceIds").toJavaList(Long.class);
|
String packageName = param.getString("packageName");
|
|
if (deviceIds == null || deviceIds.isEmpty()) {
|
return R.error("请选择设备");
|
}
|
|
List<String> results = tvDeviceService.batchRestartApp(deviceIds, packageName);
|
return R.ok(results);
|
} catch (Exception e) {
|
return R.error("重启失败: " + e.getMessage());
|
}
|
}
|
}
|
