1- use anyhow:: { anyhow, Context , Result } ;
2- use reqwest:: { Client , StatusCode } ;
31use std:: time:: Duration ;
42
3+ use anyhow:: anyhow;
4+ use anyhow:: Context ;
5+ use anyhow:: Result ;
6+ use reqwest:: Client ;
7+ use reqwest:: StatusCode ;
8+
59const USER_AGENT : & str = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/102.0.5005.63 Safari/537.36" ;
610const TIMEOUT : Duration = Duration :: from_secs ( 5 ) ;
711
812fn build_client ( proxy_url : & str ) -> Result < Client > {
913 Client :: builder ( )
10- . proxy ( reqwest:: Proxy :: all ( proxy_url)
11- . context ( "Failed to create proxy configuration" ) ?)
14+ . proxy ( reqwest:: Proxy :: all ( proxy_url) . context ( "Failed to create proxy configuration" ) ?)
1215 . timeout ( TIMEOUT )
1316 . build ( )
1417 . context ( "Failed to build HTTP client" )
@@ -36,8 +39,12 @@ pub async fn claude_is_ok(proxy_url: &str) -> Result<()> {
3639pub async fn openai_is_ok ( proxy_url : & str ) -> Result < ( ) > {
3740 let url = "https://auth.openai.com/favicon.ico" ;
3841 let client = build_client ( proxy_url) ?;
39- let resp = client. get ( url) . header ( "User-Agent" , USER_AGENT )
40- . send ( ) . await . with_context ( || "Failed to send request to OpenAI" ) ?;
42+ let resp = client
43+ . get ( url)
44+ . header ( "User-Agent" , USER_AGENT )
45+ . send ( )
46+ . await
47+ . with_context ( || "Failed to send request to OpenAI" ) ?;
4148 let status = resp. status ( ) ;
4249 if status == StatusCode :: OK {
4350 return Ok ( ( ) ) ;
@@ -49,8 +56,12 @@ pub async fn openai_is_ok(proxy_url: &str) -> Result<()> {
4956pub async fn youtube_music_is_ok ( proxy_url : & str ) -> Result < bool > {
5057 let url = "https://music.youtube.com/generate_204" ;
5158 let client = build_client ( proxy_url) ?;
52- let resp = client. head ( url) . header ( "User-Agent" , USER_AGENT )
53- . send ( ) . await . with_context ( || "Failed to send request to OpenAI" ) ?;
59+ let resp = client
60+ . head ( url)
61+ . header ( "User-Agent" , USER_AGENT )
62+ . send ( )
63+ . await
64+ . with_context ( || "Failed to send request to OpenAI" ) ?;
5465 let status = resp. status ( ) ;
5566 if status == StatusCode :: NO_CONTENT {
5667 return Ok ( true ) ;
@@ -69,4 +80,4 @@ mod test {
6980 let result = super :: openai_is_ok ( "http://localhost:7890" ) . await ;
7081 println ! ( "{:#?}" , result) ;
7182 }
72- }
83+ }
0 commit comments