ai/bot
ai/bot
1
0
Fork 0
Code Actions Wiki

add bot

Browse Source
This commit is contained in:
syui 2024-02-09 12:35:17 +09:00
parent 39a18b2f9d
commit a6b9d1c5df
Signed by: syui
GPG Key ID: 5417CFEBAD92DF56
4 changed files with 157 additions and 43 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

100
src/data.rs
View File

@ -2,6 +2,9 @@ use config::{Config, ConfigError, File};
use serde_derive::{Deserialize, Serialize};
use std::fs;
use std::io::Write;
use std::io::Read;
use std::fs::OpenOptions;
use std::path::Path;
pub fn data_file(s: &str) -> String {
let file = "/.config/ai/token";
@ -14,6 +17,21 @@ pub fn data_file(s: &str) -> String {
}
}
pub fn log_file(s: &str) -> String {
let file = "/.config/ai/txt/";
let mut f = shellexpand::tilde("~").to_string();
f.push_str(&file);
let path = Path::new(&f);
if path.is_dir() == false {
let _ = fs::create_dir_all(f.clone());
}
match &*s {
"n1" => f + &"notify_cid.txt",
"n2" => f + &"notify_cid_run.txt",
_ => f + &s,
}
}
impl Token {
pub fn new() -> Result<Self, ConfigError> {
let d = data_file("json");
@ -167,24 +185,6 @@ pub fn data_toml(s: &str) -> String {
}
}
pub fn w_cfg(h: &str, res: &str) {
let f = data_file(&"json");
let ff = data_file(&"toml");
let mut f = fs::File::create(f.clone()).unwrap();
let mut ff = fs::File::create(ff.clone()).unwrap();
f.write_all(&res.as_bytes()).unwrap();
let json: Token = serde_json::from_str(&res).unwrap();
let datas = Data {
host: h.to_string(),
did: json.did.to_string(),
handle: json.handle.to_string(),
access: json.accessJwt.to_string(),
refresh: json.refreshJwt.to_string(),
};
let toml = toml::to_string(&datas).unwrap();
ff.write_all(&toml.as_bytes()).unwrap();
}
#[derive(Serialize, Deserialize)]
pub struct Notify {
pub notifications: Vec<Notifications>
@ -433,3 +433,67 @@ pub struct Profile {
pub viewer: Viewer,
pub labels: Labels,
}
pub fn c_char(i: String) -> String {
let l = 250;
let mut s = String::new();
for ii in i.chars().enumerate() {
match ii.0 {
n if n > l.try_into().unwrap() => {break}
_ => {s.push(ii.1)}
}
}
return s
}
pub fn w_cfg(h: &str, res: &str) {
let f = data_file(&"json");
let ff = data_file(&"toml");
let mut f = fs::File::create(f.clone()).unwrap();
let mut ff = fs::File::create(ff.clone()).unwrap();
f.write_all(&res.as_bytes()).unwrap();
let json: Token = serde_json::from_str(&res).unwrap();
let datas = Data {
host: h.to_string(),
did: json.did.to_string(),
handle: json.handle.to_string(),
access: json.accessJwt.to_string(),
refresh: json.refreshJwt.to_string(),
};
let toml = toml::to_string(&datas).unwrap();
ff.write_all(&toml.as_bytes()).unwrap();
}
pub fn w_cid(cid :String, file: String, t: bool) -> bool {
let f = file;
let mut file = match OpenOptions::new()
.create(true)
.write(true)
.read(true)
.append(true)
.open(f.clone())
{
Err(why) => panic!("Couldn't open {}: {}", f, why),
Ok(file) => file,
};
let mut contents = String::new();
match file.read_to_string(&mut contents) {
Err(why) => panic!("Couldn't read {}: {}", f, why),
Ok(_) => (),
}
if contents.contains(&cid) == false {
if t {
let cid = cid + "\n";
match file.write_all(cid.as_bytes()) {
Err(why) => panic!("Couldn't write \"{}\" to {}: {}", contents, f, why),
Ok(_) => println!("finished"),
}
}
let check = false;
return check
} else {
let check = true;
return check
}
}

78
src/main.rs
View File

@ -4,8 +4,11 @@ use std::env;
use crate::ascii::c_ascii;
use crate::data::data_toml;
use crate::data::log_file;
use crate::data::url;
use crate::data::w_cfg;
use crate::data::w_cid;
use crate::data::c_char;
use data::Notify as Notify;
pub mod data;
@ -64,7 +67,12 @@ fn main() {
.command(
Command::new("bot")
.alias("b")
.action(c_bot),
.action(c_bot)
.flag(
Flag::new("server", FlagType::String)
.description("server flag")
.alias("s"),
)
)
;
app.run(args);
@ -125,22 +133,23 @@ fn c_notify(_c: &Context) {
notify();
}
pub fn char_c(i: String) -> String {
let l = 250;
let mut s = String::new();
for ii in i.chars().enumerate() {
match ii.0 {
n if n > l.try_into().unwrap() => {break}
_ => {s.push(ii.1)}
}
}
return s
}
fn bot(c: &Context) {
fn bot(_c: &Context) {
let h = async {
let notify = notify::get_request(100).await;
let server = "bsky.social";
let mut notify = notify::get_request(100).await;
if notify == "err" {
if let Ok(s) = c.string_flag("server") {
let res = refresh::post_request().await;
w_cfg(&s, &res);
} else {
let res = refresh::post_request().await;
w_cfg(&server, &res);
}
notify = notify::get_request(100).await;
}
let notify: Notify = serde_json::from_str(&notify).unwrap();
let n = notify.notifications;
let length = &n.len();
let reason = &n[0].reason;
@ -149,12 +158,53 @@ fn bot(_c: &Context) {
let read = n[0].isRead;
let cid = &n[0].cid;
let uri = &n[0].uri;
let time = &n[0].indexedAt;
let mut cid_root = cid;
let mut uri_root = uri;
let check_cid = w_cid(cid.to_string(), log_file(&"n1"), false);
let check_cid_run = w_cid(cid.to_string(), log_file(&"n2"), false);
// thread
if ! n[0].record.reply.is_none() {
cid_root = &n[0].record.reply.as_ref().unwrap().root.cid;
uri_root = &n[0].record.reply.as_ref().unwrap().root.uri;
}
println!("{}", length);
println!("{}", reason);
println!("{}", handle);
println!("{}", did);
println!("{}", read);
println!("{} {}", cid, uri);
let mut text = "";
if ! n[0].record.text.is_none() {
text = &n[0].record.text.as_ref().unwrap();
}
let vec: Vec<&str> = text.split_whitespace().collect();
let rep_com = &vec[0..].join(" ");
if check_cid == false && { reason == "mention" || reason == "reply" } || check_cid_run == false && { reason == "mention" || reason == "reply" } {
w_cid(cid.to_string(), log_file(&"n2"), true);
if rep_com.contains("did") == true || rep_com.contains("/did") == true {
let link = "https://plc.directory/".to_owned() + &did + &"/log";
let s = 0;
let e = link.chars().count();
println!("{}", link);
println!("{}", e);
let d = "\n".to_owned() + &did.to_string();
println!("{}", d);
let text_limit = c_char(d);
println!("{}", text_limit);
if text_limit.len() > 3 {
let str_rep = reply_link::post_request(text_limit.to_string(), link.to_string(), s, e.try_into().unwrap(), cid.to_string(), uri.to_string(), cid_root.to_string(), uri_root.to_string()).await;
println!("{}", str_rep);
let str_notify = notify_read::post_request(time.to_string()).await;
println!("{}", str_notify);
w_cid(cid.to_string(), log_file(&"n1"), true);
}
}
}
};
let res = tokio::runtime::Runtime::new().unwrap().block_on(h);
return res

10
src/reply.rs
View File

@ -4,7 +4,7 @@ use crate::url;
use serde_json::json;
use iso8601_timestamp::Timestamp;
pub async fn post_request(text: String, cid: String, uri: String, cid_p: String, uri_p: String) -> String {
pub async fn post_request(text: String, cid: String, uri: String, cid_root: String, uri_root: String) -> String {
let token = data_toml(&"access");
let did = data_toml(&"did");
@ -26,12 +26,12 @@ pub async fn post_request(text: String, cid: String, uri: String, cid_p: String,
"createdAt": d.to_string(),
"reply": {
"root": {
"cid": cid.to_string(),
"uri": uri.to_string()
"cid": cid_root.to_string(),
"uri": uri_root.to_string()
},
"parent": {
"cid": cid_p.to_string(),
"uri": uri_p.to_string()
"cid": cid.to_string(),
"uri": uri.to_string()
}
}
},

10
src/reply_link.rs
View File

@ -4,7 +4,7 @@ use crate::url;
use serde_json::json;
use iso8601_timestamp::Timestamp;
pub async fn post_request(text: String, link: String, s: i32, e: i32, cid: String, uri: String, cid_b: String, uri_b: String) -> String {
pub async fn post_request(text: String, link: String, s: i32, e: i32, cid: String, uri: String, cid_root: String, uri_root: String) -> String {
let token = data_toml(&"access");
let did = data_toml(&"did");
@ -25,12 +25,12 @@ pub async fn post_request(text: String, link: String, s: i32, e: i32, cid: Strin
"createdAt": d.to_string(),
"reply": {
"root": {
"cid": cid.to_string(),
"uri": uri.to_string()
"cid": cid_root.to_string(),
"uri": uri_root.to_string()
},
"parent": {
"cid": cid_b.to_string(),
"uri": uri_b.to_string()
"cid": cid.to_string(),
"uri": uri.to_string()
}
},
"facets": [