From 4b13aa48a38df2bc86f8b012b0815d5cac9eb7fb Mon Sep 17 00:00:00 2001
From: syui <syui@syui.ai>
Date: Wed, 7 Feb 2024 11:58:39 +0900
Subject: [PATCH] fix host

---
 README.md    | 18 +++++++++++++++++-
 src/main.rs  | 37 ++++++++++++++++++++++++++-----------
 src/token.rs |  5 ++---
 3 files changed, 45 insertions(+), 15 deletions(-)

diff --git a/README.md b/README.md
index a58b53e..336b472 100644
--- a/README.md
+++ b/README.md
@@ -14,15 +14,31 @@ $ ./target/debug/ai ai -t
 ### login
 
 ```sh
-# ai t $handle -p $password
+# ai token $handle -p $password
 $ ai t yui.syui.ai -p password
 
 $ cat ~/.config/ai/token.toml
 ```
 
+```sh
+# ai token $handle -p $password -s $server
+$ ai t ai.syu.is -p password -s syu.is
+```
+
 ### refresh
 
 ```
 $ ai r
 ```
 
+```
+# server: syu.is
+$ ai r -s syu.is
+```
+
+### notify
+
+```
+$ ai n
+```
+
diff --git a/src/main.rs b/src/main.rs
index 5a8701c..b30798f 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -36,12 +36,22 @@ fn main() {
                 .description("password flag")
                 .alias("p"),
                 )
+            .flag(
+                Flag::new("server", FlagType::String)
+                .description("server flag")
+                .alias("s"),
             )
+        )
         .command(
             Command::new("refresh")
             .alias("r")
-            .action(c_refresh),
+            .action(c_refresh)
+            .flag(
+                Flag::new("server", FlagType::String)
+                .description("server flag")
+                .alias("s"),
             )
+        )
         .command(
             Command::new("notify")
             .alias("n")
@@ -59,9 +69,13 @@ fn token(c: &Context) {
     let m = c.args[0].to_string();
     let h = async {
         if let Ok(p) = c.string_flag("password") {
-            let res = token::post_request(m.to_string(), p.to_string()).await;
-            println!("{}", res);
-            w_cfg("bsky.social", &res)
+            if let Ok(s) = c.string_flag("server") {
+                let res = token::post_request(m.to_string(), p.to_string(), s.to_string()).await;
+                w_cfg(&s, &res)
+            } else {
+                let res = token::post_request(m.to_string(), p.to_string(), "bsky.social".to_string()).await;
+                w_cfg(&"bsky.social", &res)
+            }
         }
     };
     let res = tokio::runtime::Runtime::new().unwrap().block_on(h);
@@ -72,22 +86,23 @@ fn c_token(c: &Context) {
     token(c);
 }
 
-fn refresh() {
+fn refresh(c: &Context) {
     let h = async {
         let res = refresh::post_request().await;
-        println!("{}", res);
-        w_cfg("bsky.social", &res)
+        if let Ok(s) = c.string_flag("server") {
+            w_cfg(&s, &res)
+        } else {
+            w_cfg("bsky.social", &res)
+        }
     };
     let res = tokio::runtime::Runtime::new().unwrap().block_on(h);
     return res
 }
 
-fn c_refresh(_c: &Context) {
-    refresh();
+fn c_refresh(c: &Context) {
+    refresh(c);
 }
 
-
-
 fn notify() {
     let h = async {
         let j = notify::get_request(100).await;
diff --git a/src/token.rs b/src/token.rs
index 2b35127..4f0f715 100644
--- a/src/token.rs
+++ b/src/token.rs
@@ -1,10 +1,9 @@
 extern crate reqwest;
 use std::collections::HashMap;
-use crate::url;
 
-pub async fn post_request(handle: String, pass: String) -> String {
+pub async fn post_request(handle: String, pass: String, host: String) -> String {
 
-    let url = url(&"session_create");
+    let url = "https://".to_owned() + &host.to_string() + &"/xrpc/com.atproto.server.createSession".to_string();
 
     let mut map = HashMap::new();
     map.insert("identifier", &handle);