2023年6月19日16:25:23 修复抖音页面修改导致无法获取登录二维码问题

王明元
1 parent 384d35dc
Showing 6 changed files with 164 additions and 3 deletions
db/freya.mv.db
fw-freya.iml
pom.xml
src/main/java/cn/fw/freya/service/crawl/impl/DouYinCrawl.java
src/main/java/cn/fw/freya/utils/dingding/DingTalkService.java
src/main/java/cn/fw/freya/utils/dingding/SendMessageBody.java
@@ -163,5 +163,7 @@
     <orderEntry type="library" name="Maven: cn.hutool:hutool-setting:5.7.16" level="project" />
     <orderEntry type="library" name="Maven: cn.hutool:hutool-log:5.7.16" level="project" />
     <orderEntry type="library" name="Maven: com.alibaba:fastjson:1.2.78" level="project" />
+    <orderEntry type="library" name="Maven: com.aliyun:alibaba-dingtalk-service-sdk:1.0.1" level="project" />
+    <orderEntry type="library" name="Maven: commons-logging:commons-logging:1.1.1" level="project" />
   </component>
 </module>
 \ No newline at end of file
@@ -86,6 +86,18 @@
             <artifactId>fastjson</artifactId>
             <version>1.2.78</version>
         </dependency>
+        <!-- 钉钉消息推送 -->
+        <dependency>
+            <groupId>com.aliyun</groupId>
+            <artifactId>alibaba-dingtalk-service-sdk</artifactId>
+            <version>1.0.1</version>
+            <exclusions>
+                <exclusion>
+                    <groupId>log4j</groupId>
+                    <artifactId>log4j</artifactId>
+                </exclusion>
+            </exclusions>
+        </dependency>
     </dependencies>
  
     <build>
@@ -87,9 +87,9 @@ public class DouYinCrawl implements CrawlStrategy {
             }
             driver.get("https://creator.douyin.com/");// 打开指定的页面
             new WebDriverWait(driver, 10, 300).until(driver1 ->
-                    driver1.findElement(By.xpath("//div[@class='creator-header-nav']//span[@class='login']"))).click();// 获取'登录'按钮元素, 单击
-            new WebDriverWait(driver, 10, 300).until(driver1 ->
-                    driver1.findElement(By.xpath("//span[@class='semi-button-content']"))).click();
+                    driver1.findElement(By.xpath("//div[@class = 'banner-div active']"))).click();// 获取'登录'按钮元素, 单击
+            /*new WebDriverWait(driver, 10, 300).until(driver1 ->
+                    driver1.findElement(By.xpath("//span[@class='semi-button-content']"))).click();*/
             WebElement qrCodeEle = new WebDriverWait(driver, 10, 300).until(driver1 ->
                     driver1.findElement(By.xpath("//div[@class='qrcode-image']/img[1]")));
             return qrCodeEle.getAttribute("src");// 返回对象src属性对应的值
+package cn.fw.freya.utils.dingding;
+
+import com.dingtalk.api.DefaultDingTalkClient;
+import com.dingtalk.api.DingTalkClient;
+import com.dingtalk.api.request.OapiRobotSendRequest;
+import com.dingtalk.api.response.OapiRobotSendResponse;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.codec.binary.Base64;
+
+import javax.crypto.Mac;
+import javax.crypto.spec.SecretKeySpec;
+import java.net.URLEncoder;
+import java.nio.charset.StandardCharsets;
+import java.util.List;
+
+/**
+ * @author wmy3969
+ * @version 1.0
+ * @date 2022/2/8 9:24
+ * @Description 钉钉推送服务类
+ */
+@Slf4j
+public class DingTalkService {
+
+    public static final String TEXT = "text";// 文本消息
+    public static final String WEBHOOK = "https://oapi.dingtalk.com/robot/send?access_token=e4b93e75a58be3bf142ad91b6a4b0f99c404d6eb1115ac6d9d36333a66aeec3c";// webhook
+    public static final String SECRET = "SECb0357db53bc981e5886b5fe6c41560ddaeb3dd0ba1b2528308fdc4908dabd668";// 密钥
+
+    /**
+     * 推送消息
+     *
+     * @param body 发送的消息体
+     * @throws Exception
+     */
+    public static void sendMessage(SendMessageBody body) throws Exception {
+        log.info("开始推送钉钉消息：" + body);
+        Long timestamp = System.currentTimeMillis();
+        String secret = body.getSecret();
+
+        String stringToSign = timestamp + "\n" + secret;
+        Mac mac = Mac.getInstance("HmacSHA256");
+        mac.init(new SecretKeySpec(secret.getBytes(StandardCharsets.UTF_8), "HmacSHA256"));
+        byte[] signData = mac.doFinal(stringToSign.getBytes(StandardCharsets.UTF_8));
+        String sign = URLEncoder.encode(new String(Base64.encodeBase64(signData)), "UTF-8");
+
+        DingTalkClient client = new DefaultDingTalkClient(body.getWebhook() + "&timestamp=" + timestamp + "&sign=" + sign);
+        OapiRobotSendRequest request = new OapiRobotSendRequest();
+
+        OapiRobotSendRequest.At at = new OapiRobotSendRequest.At();
+        if (body.isAtAll() || body.getMobileList() == null || body.getMobileList().isEmpty()) {
+            at.setIsAtAll(true);// 推送所有人
+        } else {
+            at.setAtMobiles(body.getMobileList());
+            at.setIsAtAll(false);// 推送指定用户
+        }
+        request.setAt(at);
+
+        if (TEXT.equals(body.getMsgType())) {// 文本消息
+            request.setMsgtype(TEXT);
+            OapiRobotSendRequest.Text text = new OapiRobotSendRequest.Text();
+            text.setContent(body.getText());
+            request.setText(text);
+        }
+
+        OapiRobotSendResponse response = client.execute(request);
+        log.info("钉钉推送返回结果：" + response);
+    }
+
+    /**
+     * 给群里发送一条钉钉消息 @所有人
+     *
+     * @param message 消息内容
+     * @throws Exception
+     */
+    public static void sendToEverybody(String message) throws Exception {
+        sendMessage(SendMessageBody.builder()
+                .isAtAll(true)
+                .msgType(TEXT)
+                .webhook(WEBHOOK)
+                .secret(SECRET)
+                .text(message)
+                .build());
+    }
+
+    /**
+     * 给群里发送一条钉钉消息 @指定的人
+     *
+     * @param message    消息内容
+     * @param peopleList 指定的人手机号集合
+     * @throws Exception
+     */
+    public static void sendToSpecified(String message, List<String> peopleList) throws Exception {
+        sendMessage(SendMessageBody.builder()
+                .isAtAll(false)
+                .msgType(TEXT)
+                .webhook(WEBHOOK)
+                .secret(SECRET)
+                .text(message)
+                .mobileList(peopleList)
+                .build());
+    }
+}
+package cn.fw.freya.utils.dingding;
+
+import lombok.AllArgsConstructor;
+import lombok.Builder;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+
+import java.util.List;
+
+/**
+ * @author wmy3969
+ * @version 1.0
+ * @date 2022/2/8 9:28
+ * @Description 发送钉钉消息
+ */
+@Data
+@NoArgsConstructor
+@AllArgsConstructor
+@Builder(toBuilder = true)
+public class SendMessageBody {
+    /**
+     * 消息类型
+     */
+    private String msgType;
+    /**
+     * webhook
+     */
+    private String webhook;
+    /**
+     * 密钥
+     */
+    private String secret;
+    /**
+     * 消息文本
+     */
+    private String text;
+    /**
+     * 群内指定成员手机号
+     */
+    private List<String> mobileList;
+    /**
+     * 是否推送所有人
+     */
+    private boolean isAtAll;
+}