From 0a70924df7b85d482b1d95f86b72829671233e1b Mon Sep 17 00:00:00 2001
From: Borber <borber.drum@gmail.com>
Date: Wed, 11 Jan 2023 23:48:47 +0800
Subject: [PATCH] =?UTF-8?q?[mod]=20=E6=94=AF=E6=8C=81=E5=BF=AB=E6=89=8B?=
 =?UTF-8?q?=E7=9B=B4=E6=92=AD=E6=BA=90=E8=8E=B7=E5=8F=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 CHANGELOG.md         |  3 +++
 README.md            |  1 +
 src/live/kuaishou.rs | 56 ++++++++++++++++++++++++++++++++++++++++++++
 src/live/mod.rs      |  1 +
 src/main.rs          |  8 ++++++-
 5 files changed, 68 insertions(+), 1 deletion(-)
 create mode 100644 src/live/kuaishou.rs

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 5f9b6a8..03bb989 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -7,6 +7,9 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+### Added
+
+- 支持快手直播源获取
 
 ## [0.1.5] - 2023-01-11
 
diff --git a/README.md b/README.md
index 2645b13..d4d7d73 100644
--- a/README.md
+++ b/README.md
@@ -51,6 +51,7 @@
 - [ ] 根据画质排序
 - [ ] gui
 - [ ] mpv播放
+- [ ] 是否一次性链接标识字段
 
 # 说明
 
diff --git a/src/live/kuaishou.rs b/src/live/kuaishou.rs
new file mode 100644
index 0000000..134a9cc
--- /dev/null
+++ b/src/live/kuaishou.rs
@@ -0,0 +1,56 @@
+use anyhow::{Ok, Result};
+use regex::Regex;
+
+use crate::{
+    common::{CLIENT, USER_AGENT},
+    modle::{Node, ShowType},
+};
+
+const URL: &str = "https://live.kuaishou.com/u/";
+
+/// 快手直播
+///
+/// https://live.kuaishou.com/
+pub async fn get(rid: &str) -> Result<ShowType> {
+    let text = CLIENT
+        .get(format!("{}{}", URL, rid))
+        .header("User-Agent", USER_AGENT)
+        .header("Cookie", "did=web_d563dca728d28b00336877723e0359ed;")
+        .send()
+        .await?
+        .text()
+        .await?;
+    let re = Regex::new(r#"<script>window.__INITIAL_STATE__=([\s\S]*?);\(function"#).unwrap();
+    let stream = match re.captures(&text) {
+        Some(caps) => caps.get(1).unwrap().as_str(),
+        None => {
+            return Ok(ShowType::Error("直播间不存在".to_string()));
+        }
+    };
+    let json: serde_json::Value = serde_json::from_str(stream).unwrap();
+    // TODO 更改其他逻辑 多用Null
+    match &json["liveroom"]["liveStream"]["playUrls"][0]["adaptationSet"]["representation"] {
+        serde_json::Value::Null => {
+            return Ok(ShowType::Off);
+        }
+        reps => {
+            let list = reps.as_array().unwrap();
+            let url = list[list.len() - 1]["url"].as_str().unwrap();
+            Ok(ShowType::On(vec![Node {
+                rate: "蓝光8M".to_string(),
+                url: url.to_string(),
+            }]))
+        }
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use crate::live::kuaishou::get;
+    use crate::util::match_show_type;
+
+    #[tokio::test]
+    async fn test_kuaishou() {
+        match_show_type(get("3xgexgpig9gwwi2").await.unwrap());
+    }
+}
diff --git a/src/live/mod.rs b/src/live/mod.rs
index 0f7a500..a871f83 100644
--- a/src/live/mod.rs
+++ b/src/live/mod.rs
@@ -3,5 +3,6 @@ pub mod douyin;
 pub mod douyu;
 pub mod huajiao;
 pub mod huya;
+pub mod kuaishou;
 pub mod mht;
 pub mod yqs;
diff --git a/src/main.rs b/src/main.rs
index b2660cc..7f5afaa 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -15,7 +15,7 @@ ________ _______ _______ _______
 |__     |    ___|       |       |
 |_______|_______|___|___|__|_|__|
 
-获取直播源地址, 目前支持 B站, 斗鱼, 抖音, 虎牙, 花椒, 艺气山, 棉花糖", long_about = None)]
+获取直播源地址, 目前支持 B站, 斗鱼, 抖音, 虎牙, 快手, 花椒, 艺气山, 棉花糖", long_about = None)]
 struct Cli {
     #[command(subcommand)]
     command: Commands,
@@ -43,6 +43,11 @@ enum Commands {
         /// 房间号
         rid: String,
     },
+    /// 快手
+    Kuaishou {
+        /// 房间号
+        rid: String,
+    },
     /// 花椒
     Huajiao {
         /// 房间号
@@ -68,6 +73,7 @@ async fn main() -> Result<()> {
         Commands::Douyu { rid } => util::match_show_type(live::douyu::get(&rid).await?),
         Commands::Douyin { rid } => util::match_show_type(live::douyin::get(&rid).await?),
         Commands::Huya { rid } => util::match_show_type(live::huya::get(&rid).await?),
+        Commands::Kuaishou { rid } => util::match_show_type(live::kuaishou::get(&rid).await?),
         Commands::Huajiao { rid } => util::match_show_type(live::huajiao::get(&rid).await?),
         Commands::Yqs { rid } => util::match_show_type(live::yqs::get(&rid).await?),
         Commands::Mht { rid } => util::match_show_type(live::mht::get(&rid).await?),