syui
/
atr
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

test shell

Browse Source
main
syui 3 months ago
parent c7dda61adf
commit d44ef061da
Signed by: syui
GPG Key ID: 67AC97A939D3EA19
5 changed files with 277 additions and 0 deletions
Show Stats Download Patch File Download Diff File

2
.gitignore vendored
Unescape Escape View File

@ -10,3 +10,5 @@ target
*/node_mobules
*/*.lock
json
*.png
*.jpg

79
scpt/arch.zsh
Unescape Escape View File

@ -0,0 +1,79 @@
#!/bin/zsh
home=/home/syui
name=arch
a=$home/$name
function arch_in(){
mkdir -p $a
sudo pacstrap -c $a base
#sudo echo pts/0 >> $a/etc/securetty
#sudo echo pts/1 >> $a/etc/securetty
sudo rm -rf /var/lib/machines/$name
sudo rm -rf /var/lib/machines/${name}back
sudo mv $a /var/lib/machines/
sudo machinectl clone arch archback
}
function arch_rm(){
sudo machinectl remove $name
}
function arch_up(){
sudo machinectl poweroff $name > /dev/null 2>&1
sleep 5
sudo machinectl terminate $name > /dev/null 2>&1
sleep 5
sudo machinectl start ${name}back
sleep 5
ssh ${name}back pacman -Syu --noconfirm
sleep 5
sudo machinectl poweroff ${name}back
}
function arch_st(){
sudo machinectl start $name
}
function arch_of(){
sudo machinectl poweroff $name
}
function arch_ex(){
sudo machinectl shell $name
$1
poweroff
}
function arch_re(){
sudo machinectl poweroff $name > /dev/null 2>&1
sleep 5
sudo machinectl terminate $name > /dev/null 2>&1
sleep 5
sudo machinectl remove $name
sleep 5
sudo machinectl clone ${name}back $name
sleep 5
sudo machinectl start $name
}
case "$1" in
"update"|"-u")
arch_up
arch_re
echo "machinectl update done"
;;
"reset"|"-r")
arch_re
echo "machinectl reset done"
;;
*)
sudo machinectl start $name
t=`ssh $name "$*"`
if [ -z "$t" ];then
ssh -tt $name "$*"
else
echo "$t"
fi
;;
esac

40
scpt/diffusion.zsh
Unescape Escape View File

@ -0,0 +1,40 @@
#!/bin/zsh
dir=${0:a:h}
f=$dir/diffusion_prompt.txt
seed=$RANDOM
sd_prompt_a="masterpiece, high quality, very_high_resolution, large_filesize, full color"
## kawaii girl random prompt
# ra=$(($RANDOM % 2 + 1))
# rp="beautiful kawaii "`echo "little girl,girl"|cut -d , -f $ra|tr -d ,`
# rb=$(($RANDOM % 3 + 1))
# rbp="with "`echo "gold,silver,black"|cut -d , -f $rb|tr -d ,`" hair"
# rc=$(($RANDOM % 4 + 1))
# rcp=`echo "wavy,long,straight,"|cut -d , -f $rc|tr -d ,`
# rd=$(($RANDOM % 3 + 1))
# rdp="in "`echo "fluttery white onepice,simple white onepice,normal white school uniform"|cut -d , -f $rd|tr -d ,`
# echo $sd_prompt_a, $rp $rdp $rcp $rbp
if [ -n "$1" ];then
echo $sd_prompt_a $* >! $f
cat $f
q=`cat $f`
else
echo no query
exit
fi
rm -rf $dir/png
mkdir -p $dir/png
ssh win Remove-Item -Recurse -Force stable-diffusion/outputs
ssh win Remove-Item -Recurse -Force msbot
ssh win mkdir stable-diffusion/outputs
ssh win mkdir msbot
ssh win "conda activate ldm;cd ./stable-diffusion/;python optimizedSD/optimized_txt2img.py --prompt \"${q}\" --H 512 --W 512 --seed $seed --n_iter 1 --n_samples 1 --ddim_steps 50"
diff_dir=`ssh win ls stable-diffusion/outputs/txt2img-samples -Name|cut -b 1-2`
ssh win "cd stable-diffusion/outputs/txt2img-samples;mv $diff_dir* $diff_dir -Force;mv $diff_dir/*.png ~/msbot/t.png -Force"
ssh win ".\scoop\apps\imagemagick\current\convert.exe msbot/t.png msbot/t.jpg"
scp -r win:msbot/t.jpg $dir/png/t.jpg

85
src/at_reply_media.rs
Unescape Escape View File

@ -0,0 +1,85 @@
extern crate reqwest;
use crate::token_toml;
use crate::url;
use serde_json::json;
use iso8601_timestamp::Timestamp;
pub async fn post_request(text: String, cid: String, uri: String, mid: String, itype: String) -> String {
let token = token_toml(&"access");
let did = token_toml(&"did");
let url = url(&"record_create");
//let url = "https://bsky.social/xrpc/com.atproto.repo.createRecord";
let col = "app.bsky.feed.post".to_string();
let d = Timestamp::now_utc();
let d = d.to_string();
//{
// "did": "",
// "collection": "",
// "record": {
// "text": "",
// "createdAt": "",
// "": "",
// "embed": {
// "$type": "app.bsky.embed.images",
// "images": [
// {
// "image": {
// "cid": "",
// "mimeType": ""
// },
// "alt": ""
// }
// ]
// }
// }
//}
let post = Some(json!({
"did": did.to_string(),
"collection": col.to_string(),
"record": {
"text": text.to_string(),
"createdAt": d.to_string(),
"embed": {
"$type": "app.bsky.embed.images",
"images": [
{
"image": {
"cid": mid.to_string(),
"mimeType": itype.to_string()
},
"alt": ""
}
]
},
"reply": {
"root": {
"cid": cid.to_string(),
"uri": uri.to_string()
},
"parent": {
"cid": cid.to_string(),
"uri": uri.to_string()
}
}
},
}));
let client = reqwest::Client::new();
let res = client
.post(url)
.json(&post)
.header("Authorization", "Bearer ".to_owned() + &token)
.send()
.await
.unwrap()
.text()
.await
.unwrap();
return res
}

71
src/main.rs
Unescape Escape View File

@ -34,6 +34,7 @@ pub mod deepl;
pub mod at_notify_limit;
pub mod at_notify_read;
pub mod at_reply;
pub mod at_reply_media;
pub mod at_post;
pub mod at_post_link;
pub mod at_profile;
@ -804,6 +805,8 @@ fn first_start(c: &Context) -> io::Result<()> {
Ok(())
}
fn first(c: &Context) {
first_start(c).unwrap();
}
@ -979,6 +982,64 @@ fn bot_run(_c: &Context) {
let str_notify = at_notify_read::post_request(time.to_string()).await;
println!("{}", str_notify);
}
if com == "/sh" {
let prompt = &vec[2..].join(" ");
println!("cmd:{}, prompt:{}", com, prompt);
println!("cid:{}, uri:{}", cid, uri);
println!("{}", text);
let file = "/.config/atr/scpt/arch.zsh";
let mut f = shellexpand::tilde("~").to_string();
f.push_str(&file);
use std::process::Command;
let output = Command::new(&f).arg(&prompt).output().expect("zsh");
let d = String::from_utf8_lossy(&output.stdout);
let d = d.to_string();
println!("{}", d);
let text_limit = char_c(d);
let str_rep = at_reply::post_request(text_limit.to_string(), cid.to_string(), uri.to_string()).await;
println!("{}", str_rep);
let str_notify = at_notify_read::post_request(time.to_string()).await;
println!("{}", str_notify);
}
if com == "/diffusion" {
let prompt = &vec[2..].join(" ");
println!("cmd:{}, prompt:{}", com, prompt);
println!("cid:{}, uri:{}", cid, uri);
println!("{}", text);
let file = "/.config/atr/scpt/diffusion.zsh";
let mut f = shellexpand::tilde("~").to_string();
f.push_str(&file);
use std::process::Command;
let output = Command::new(&f).arg(&prompt).output().expect("zsh");
let d = String::from_utf8_lossy(&output.stdout);
let d = d.to_string();
println!("{}", d);
//media upload { #efactoring }
let file = "/.config/atr/scpt/png/t.jpg";
let mut f = shellexpand::tilde("~").to_string();
f.push_str(&file);
let token = token_toml(&"access");
let atoken = "Authorization: Bearer ".to_owned() + &token;
let con = "Content-Type: image/png";
let url = url(&"upload_blob");
let f = "@".to_owned() + &f;
let output = Command::new("curl").arg("-X").arg("POST").arg("-sL").arg("-H").arg(&con).arg("-H").arg(&atoken).arg("--data-binary").arg(&f).arg(&url).output().expect("curl");
let d = String::from_utf8_lossy(&output.stdout);
let d = d.to_string();
let mid: Cid = serde_json::from_str(&d).unwrap();
let mid = mid.cid;
println!("{}", mid);
let text_limit = "#stablediffusion";
let itype = "image/jpeg";
let str_rep = at_reply_media::post_request(text_limit.to_string(), cid.to_string(), uri.to_string(), mid.to_string(), itype.to_string()).await;
println!("{}", str_rep);
let str_notify = at_notify_read::post_request(time.to_string()).await;
println!("{}", str_notify);
}
}
}
}
@ -1009,6 +1070,8 @@ fn test(_c: &Context) {
let s = 0;
let e = link.chars().count();
let handle = "syui.cf";
let mid = "bafyreid27zk7lbis4zw5fz4podbvbs4fc5ivwji3dmrwa6zggnj4bnd57u";
let itype = "image/jpeg";
//pub mod openai;
let h = async {
@ -1042,6 +1105,14 @@ fn test(_c: &Context) {
};
tokio::runtime::Runtime::new().unwrap().block_on(h);
//pub mod at_reply_media;
let h = async {
println!("{}","at_reply_media");
let str = at_reply_media::post_request(prompt.to_string(), cid.to_string(), uri.to_string(), mid.to_string(), itype.to_string());
println!("{}",str.await);
};
tokio::runtime::Runtime::new().unwrap().block_on(h);
//pub mod at_notify_limit;
let h = async {
println!("{}","at_notify_limit");

Write Preview
Loading…
Cancel
Save