package com.ycl.controller.dingding; import cn.hutool.core.util.ObjectUtil; import com.alibaba.fastjson.JSONObject; import com.alibaba.xxpt.gateway.shared.client.http.ExecutableClient; import com.alibaba.xxpt.gateway.shared.client.http.GetClient; import com.alibaba.xxpt.gateway.shared.client.http.PostClient; import com.aliyun.oss.OSS; import com.aliyun.oss.OSSClientBuilder; import com.aliyun.oss.model.ObjectMetadata; import com.ycl.api.CommonResult; import com.ycl.bo.AdminUserDetails; import com.ycl.common.dingding.DingCommon; import com.ycl.config.DingConfig; import com.ycl.controller.BaseController; import com.ycl.entity.dingding.DingUserInfo; import com.ycl.service.auth.AuthService; import com.ycl.service.ding.BookRemarkService; import com.ycl.service.ding.DingService; import com.ycl.utils.ConstantPropertiesUtils; import com.ycl.vo.AddressBookVO; import com.ycl.vo.NewAddressBookVO; import io.swagger.annotations.Api; import io.swagger.annotations.ApiOperation; import lombok.extern.slf4j.Slf4j; import org.apache.http.HttpEntity; import org.apache.http.client.methods.CloseableHttpResponse; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Value; import org.springframework.security.core.context.SecurityContextHolder; import org.springframework.web.bind.annotation.*; import org.springframework.web.multipart.MultipartFile; import javax.annotation.Resource; import java.io.*; import java.util.Date; import java.util.HashMap; import java.util.List; import java.util.Map; import static com.ycl.common.constant.DingConst.*; @RestController @RequestMapping("/ding") @Api(tags = "钉钉") @Slf4j public class DingController extends BaseController { @Autowired private DingCommon dingCommon; @Value("${jwt.tokenHead}") private String tokenHead; /** * DING 免登 * 根据临时授权码和access_token获取用户信息 * * @author cjx */ @Autowired private AuthService authService; @Autowired private DingService dingService; @Autowired private BookRemarkService bookRemarkService; @Resource private ExecutableClient executableClient; @Resource private DingConfig dingConfig; @ApiOperation(value = "根据钉钉授权码获取token") @GetMapping("/dingLogin") public CommonResult dingLogin(@RequestParam String code) { HashMap map = new HashMap<>(); //获取用户信息 DingUserInfo dingUser = dingCommon.getDingUserInfo(code); //按ding登录 String token = authService.dingLogin(dingUser); map.put("token", token); map.put("tokenHead", tokenHead); map.put("accountId", dingUser.getAccountId()); return CommonResult.success(map); } @ApiOperation(value = "根据钉钉授权码获取jsapiToken") @GetMapping("/dingTicker") public CommonResult dingTicker() { String accessToken = getToken(); String ticket = getTiker(accessToken); HashMap map = new HashMap<>(); map.put("accessToken", accessToken); map.put("ticket", ticket); return CommonResult.success(map); } private String getToken() { //调用API GetClient getTokenClient = executableClient.newGetClient(GET_TOKEN); //设置参数 getTokenClient.addParameter("appkey", dingConfig.getAppKey()); getTokenClient.addParameter("appsecret", dingConfig.getAppSecret()); String apiResult = getTokenClient.get(); return parsingResult(apiResult); } private String getTiker(String accToken) { //调用API PostClient postClient = executableClient.newPostClient(GET_TIKER); //设置参数 if (ObjectUtil.isNotNull(accToken)) { postClient.addParameter("accessToken", accToken); postClient.addParameter("appkey", dingConfig.getAppKey()); postClient.addParameter("appsecret", dingConfig.getAppSecret()); String apiResult = postClient.post(); return parsingResult(apiResult); } return null; } private String parsingResult(String apiResult) { if (ObjectUtil.isNotNull(apiResult)) { JSONObject resJson = JSONObject.parseObject(apiResult); if (resJson.getBoolean("success")) { JSONObject content = resJson.getJSONObject("content"); if (content.getBoolean("success")) { JSONObject dataObj = content.getJSONObject("data"); String accessToken = dataObj.getString("accessToken"); return accessToken; } } else { log.error(resJson.toJSONString()); } } return null; } @ApiOperation(value = "通讯录") @GetMapping("/addressBook") public CommonResult> getAddressBook() { AdminUserDetails user = (AdminUserDetails) SecurityContextHolder.getContext().getAuthentication().getPrincipal(); return CommonResult.success(dingService.getAddressBook(user.getUserId())); } @ApiOperation(value = "通讯录递归") @GetMapping("/addressBookById") public CommonResult> getAddressBook(@RequestParam("id") Long id) { AdminUserDetails user = (AdminUserDetails) SecurityContextHolder.getContext().getAuthentication().getPrincipal(); return CommonResult.success(dingService.getAddressBookByParentId(id, user.getUserId())); } @ApiOperation(value = "通讯录递归") @GetMapping("/getDddressBook") public CommonResult getNewAddressBook(@RequestParam("id") Long id) { AdminUserDetails user = (AdminUserDetails) SecurityContextHolder.getContext().getAuthentication().getPrincipal(); return CommonResult.success(dingService.getAddressBookById(id, user.getUserId())); } @ApiOperation(value = "修改通讯录备注") @PutMapping("/remark") public CommonResult updateRemark(@RequestParam("userId") Long userId, @RequestParam("remark") String remark) { AdminUserDetails user = (AdminUserDetails) SecurityContextHolder.getContext().getAuthentication().getPrincipal(); return CommonResult.success(bookRemarkService.updateRemark(user, userId, remark)); } private String getVideo(String MediaId,MultipartFile file) { CloseableHttpResponse response = null; //调用API GetClient getClient = executableClient.newGetClient(GET_VIDEO); String token =getToken(); //设置参数 try { if (ObjectUtil.isNotNull(MediaId)) { getClient.addParameter("accessToken", token); getClient.addParameter("mediaId", MediaId); // String result = getClient.get(); // return parsingResult(result); response = getClient.getB(); HttpEntity entity = response.getEntity(); InputStream is = entity.getContent(); File voice = new File("/Users/video/log1.xlsx"); OutputStream fos = new FileOutputStream(voice); int cache = 10 * 1024; byte[] buffer = new byte[cache]; int ch = 0; while ((ch = is.read(buffer)) != -1) { fos.write(buffer, 0, ch); } is.close(); fos.flush(); fos.close(); } if (file == null) { return "上传视频为空"; } String endpoint = ConstantPropertiesUtils.END_POINT; String accessKeyId = ConstantPropertiesUtils.ACCESS_KEY_ID; String accessKeySecret = ConstantPropertiesUtils.ACCESS_KEY_SECRET; String bucketName = ConstantPropertiesUtils.BUCKET_NAME; // 创建OSSClient实例 OSS ossClient = new OSSClientBuilder().build(endpoint, accessKeyId, accessKeySecret); // 获取文件的名称 String fileName = "sczf/" + file.getOriginalFilename(); ObjectMetadata objectMetadata = new ObjectMetadata(); objectMetadata.setContentType(getcontentType(fileName.substring(fileName.lastIndexOf(".")))); // 调用oss的方法实现长传 // 第一个参数 bucketName // 第二个参数 上传到oss的文件路径和文件名称 ossClient.putObject(bucketName, fileName, new ByteArrayInputStream(file.getBytes()), objectMetadata); // 关闭OSSClient。 ossClient.shutdown(); // 把上传的文件路径返回 (手动拼接) // 这里设置图片有效时间 我设置了30年 Date expiration = new Date(System.currentTimeMillis() + (long) 946080000 * 1000); String url = ossClient.generatePresignedUrl(bucketName, fileName, expiration).toString(); return url; }catch (Exception e) { e.printStackTrace(); }finally { if (response != null) { try { //特别提醒:需要调用response的close方法关闭网络连接!!! response.close(); } catch (Exception e) { e.printStackTrace(); } } } return null; } public static String getcontentType(String FilenameExtension) { if (FilenameExtension.equalsIgnoreCase(".mp4")||FilenameExtension.equalsIgnoreCase(".mov")) { return "video/mp4"; } if (FilenameExtension.equalsIgnoreCase(".bmp")) { return "image/bmp"; } if (FilenameExtension.equalsIgnoreCase(".gif")) { return "image/gif"; } if (FilenameExtension.equalsIgnoreCase(".jpeg") || FilenameExtension.equalsIgnoreCase(".jpg") || FilenameExtension.equalsIgnoreCase(".png")) { return "image/jpg"; } if (FilenameExtension.equalsIgnoreCase(".html")) { return "text/html"; } if (FilenameExtension.equalsIgnoreCase(".txt")) { return "text/plain"; } if (FilenameExtension.equalsIgnoreCase(".vsd")) { return "application/vnd.visio"; } if (FilenameExtension.equalsIgnoreCase(".pptx") || FilenameExtension.equalsIgnoreCase(".ppt")) { return "application/vnd.ms-powerpoint"; } if (FilenameExtension.equalsIgnoreCase(".docx") || FilenameExtension.equalsIgnoreCase(".doc")) { return "application/msword"; } if (FilenameExtension.equalsIgnoreCase(".xml")) { return "text/xml"; } return "image/jpg"; } }