0.5.3?
This commit is contained in:
parent
ddbdde5ae6
commit
2ab3f0d77b
2
Cargo.lock
generated
2
Cargo.lock
generated
@ -910,7 +910,7 @@ dependencies = [
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "ica-rs"
|
name = "ica-rs"
|
||||||
version = "0.5.2"
|
version = "0.5.3"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"anyhow",
|
"anyhow",
|
||||||
"chrono",
|
"chrono",
|
||||||
|
@ -1,6 +1,6 @@
|
|||||||
[package]
|
[package]
|
||||||
name = "ica-rs"
|
name = "ica-rs"
|
||||||
version = "0.5.2"
|
version = "0.5.3"
|
||||||
edition = "2021"
|
edition = "2021"
|
||||||
|
|
||||||
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
|
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
|
||||||
|
@ -47,18 +47,18 @@ class IcaStatus:
|
|||||||
...
|
...
|
||||||
|
|
||||||
|
|
||||||
class ReplyMessage:
|
class IcaReplyMessage:
|
||||||
...
|
...
|
||||||
|
|
||||||
|
|
||||||
class SendMessage:
|
class IcaSendMessage:
|
||||||
@property
|
@property
|
||||||
def content(self) -> str:
|
def content(self) -> str:
|
||||||
...
|
...
|
||||||
@content.setter
|
@content.setter
|
||||||
def content(self, value: str) -> None:
|
def content(self, value: str) -> None:
|
||||||
...
|
...
|
||||||
def with_content(self, content: str) -> "SendMessage":
|
def with_content(self, content: str) -> "IcaSendMessage":
|
||||||
"""
|
"""
|
||||||
为了链式调用, 返回自身
|
为了链式调用, 返回自身
|
||||||
"""
|
"""
|
||||||
@ -66,15 +66,15 @@ class SendMessage:
|
|||||||
return self
|
return self
|
||||||
|
|
||||||
|
|
||||||
class DeleteMessage:
|
class IcaDeleteMessage:
|
||||||
def __str__(self):
|
def __str__(self):
|
||||||
...
|
...
|
||||||
|
|
||||||
|
|
||||||
class NewMessage:
|
class IcaNewMessage:
|
||||||
def reply_with(self, message: str) -> SendMessage:
|
def reply_with(self, message: str) -> IcaSendMessage:
|
||||||
...
|
...
|
||||||
def as_deleted(self) -> DeleteMessage:
|
def as_deleted(self) -> IcaDeleteMessage:
|
||||||
...
|
...
|
||||||
def __str__(self) -> str:
|
def __str__(self) -> str:
|
||||||
...
|
...
|
||||||
@ -93,6 +93,15 @@ class NewMessage:
|
|||||||
@property
|
@property
|
||||||
def is_reply(self) -> bool:
|
def is_reply(self) -> bool:
|
||||||
...
|
...
|
||||||
|
@property
|
||||||
|
def is_room_msg(self) -> bool:
|
||||||
|
...
|
||||||
|
@property
|
||||||
|
def is_chat_msg(self) -> bool:
|
||||||
|
...
|
||||||
|
@property
|
||||||
|
def room_id(self) -> RoomId:
|
||||||
|
...
|
||||||
|
|
||||||
|
|
||||||
class IcaClient:
|
class IcaClient:
|
||||||
@ -103,13 +112,13 @@ class IcaClient:
|
|||||||
# (因为目前来说, rust调用 Python端没法启动一个异步运行时
|
# (因为目前来说, rust调用 Python端没法启动一个异步运行时
|
||||||
# 所以只能 tokio::task::block_in_place 转换成同步调用)
|
# 所以只能 tokio::task::block_in_place 转换成同步调用)
|
||||||
# """
|
# """
|
||||||
def send_message(self, message: SendMessage) -> bool:
|
def send_message(self, message: IcaSendMessage) -> bool:
|
||||||
...
|
...
|
||||||
def send_and_warn(self, message: SendMessage) -> bool:
|
def send_and_warn(self, message: IcaSendMessage) -> bool:
|
||||||
"""发送消息, 并在日志中输出警告信息"""
|
"""发送消息, 并在日志中输出警告信息"""
|
||||||
self.warn(message.content)
|
self.warn(message.content)
|
||||||
return self.send_message(message)
|
return self.send_message(message)
|
||||||
def delete_message(self, message: DeleteMessage) -> bool:
|
def delete_message(self, message: IcaDeleteMessage) -> bool:
|
||||||
...
|
...
|
||||||
|
|
||||||
@property
|
@property
|
||||||
@ -144,11 +153,11 @@ on_load = Callable[[IcaClient], None]
|
|||||||
# def on_load(client: IcaClient) -> None:
|
# def on_load(client: IcaClient) -> None:
|
||||||
# ...
|
# ...
|
||||||
|
|
||||||
on_message = Callable[[NewMessage, IcaClient], None]
|
on_ica_message = Callable[[IcaNewMessage, IcaClient], None]
|
||||||
# def on_message(msg: NewMessage, client: IcaClient) -> None:
|
# def on_message(msg: NewMessage, client: IcaClient) -> None:
|
||||||
# ...
|
# ...
|
||||||
|
|
||||||
on_delete_message = Callable[[MessageId, IcaClient], None]
|
on_ica_delete_message = Callable[[MessageId, IcaClient], None]
|
||||||
# def on_delete_message(msg_id: MessageId, client: IcaClient) -> None:
|
# def on_delete_message(msg_id: MessageId, client: IcaClient) -> None:
|
||||||
# ...
|
# ...
|
||||||
|
|
||||||
|
@ -1,12 +1,12 @@
|
|||||||
from typing import TYPE_CHECKING, TypeVar
|
from typing import TYPE_CHECKING, TypeVar
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from ica_typing import NewMessage, IcaClient
|
from ica_typing import IcaNewMessage, IcaClient
|
||||||
else:
|
else:
|
||||||
NewMessage = TypeVar("NewMessage")
|
IcaNewMessage = TypeVar("NewMessage")
|
||||||
IcaClient = TypeVar("IcaClient")
|
IcaClient = TypeVar("IcaClient")
|
||||||
|
|
||||||
def on_message(msg: NewMessage, client: IcaClient) -> None:
|
def on_ica_message(msg: IcaNewMessage, client: IcaClient) -> None:
|
||||||
if not (msg.is_from_self or msg.is_reply):
|
if not (msg.is_from_self or msg.is_reply):
|
||||||
if msg.content == "/bot":
|
if msg.content == "/bot":
|
||||||
reply = msg.reply_with(f"ica-async-rs({client.version})-sync-py {client.ica_version}")
|
reply = msg.reply_with(f"ica-async-rs({client.version})-sync-py {client.ica_version}")
|
||||||
|
@ -7,11 +7,11 @@ import traceback
|
|||||||
from typing import TYPE_CHECKING, TypeVar, Optional, Tuple
|
from typing import TYPE_CHECKING, TypeVar, Optional, Tuple
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from ica_typing import NewMessage, IcaClient, ConfigData
|
from ica_typing import IcaNewMessage, IcaClient, ConfigData
|
||||||
CONFIG_DATA: ConfigData
|
CONFIG_DATA: ConfigData
|
||||||
else:
|
else:
|
||||||
CONFIG_DATA = None
|
CONFIG_DATA = None
|
||||||
NewMessage = TypeVar("NewMessage")
|
IcaNewMessage = TypeVar("NewMessage")
|
||||||
IcaClient = TypeVar("IcaClient")
|
IcaClient = TypeVar("IcaClient")
|
||||||
|
|
||||||
_version_ = "2.2.0-rs"
|
_version_ = "2.2.0-rs"
|
||||||
@ -57,7 +57,7 @@ def format_hit_count(count: int) -> str:
|
|||||||
return count_str
|
return count_str
|
||||||
|
|
||||||
|
|
||||||
def wrap_request(url: str, msg: NewMessage, client: IcaClient) -> Optional[dict]:
|
def wrap_request(url: str, msg: IcaNewMessage, client: IcaClient) -> Optional[dict]:
|
||||||
# if CONFIG_DATA
|
# if CONFIG_DATA
|
||||||
try:
|
try:
|
||||||
cookie = CONFIG_DATA["cookie"]
|
cookie = CONFIG_DATA["cookie"]
|
||||||
@ -83,7 +83,7 @@ def wrap_request(url: str, msg: NewMessage, client: IcaClient) -> Optional[dict]
|
|||||||
return response.json()
|
return response.json()
|
||||||
|
|
||||||
|
|
||||||
def bmcl_dashboard(msg: NewMessage, client: IcaClient) -> None:
|
def bmcl_dashboard(msg: IcaNewMessage, client: IcaClient) -> None:
|
||||||
req_time = time.time()
|
req_time = time.time()
|
||||||
# 记录请求时间
|
# 记录请求时间
|
||||||
data = wrap_request("https://bd.bangbang93.com/openbmclapi/metric/dashboard", msg, client)
|
data = wrap_request("https://bd.bangbang93.com/openbmclapi/metric/dashboard", msg, client)
|
||||||
@ -196,7 +196,7 @@ def bmcl_rank_general(msg, client):
|
|||||||
client.send_message(reply)
|
client.send_message(reply)
|
||||||
|
|
||||||
|
|
||||||
def bmcl_rank(msg: NewMessage, client: IcaClient, name: str) -> None:
|
def bmcl_rank(msg: IcaNewMessage, client: IcaClient, name: str) -> None:
|
||||||
req_time = time.time()
|
req_time = time.time()
|
||||||
# 记录请求时间
|
# 记录请求时间
|
||||||
rank_data = wrap_request("https://bd.bangbang93.com/openbmclapi/metric/rank", msg, client)
|
rank_data = wrap_request("https://bd.bangbang93.com/openbmclapi/metric/rank", msg, client)
|
||||||
@ -247,7 +247,7 @@ help = """/bmcl -> dashboard
|
|||||||
"""
|
"""
|
||||||
|
|
||||||
|
|
||||||
def on_message(msg: NewMessage, client: IcaClient) -> None:
|
def on_ica_message(msg: IcaNewMessage, client: IcaClient) -> None:
|
||||||
if not (msg.is_from_self or msg.is_reply):
|
if not (msg.is_from_self or msg.is_reply):
|
||||||
if '\n' in msg.content:
|
if '\n' in msg.content:
|
||||||
return
|
return
|
||||||
|
@ -5,12 +5,12 @@ import traceback
|
|||||||
from typing import TYPE_CHECKING, TypeVar
|
from typing import TYPE_CHECKING, TypeVar
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from ica_typing import NewMessage, IcaClient
|
from ica_typing import IcaNewMessage, IcaClient
|
||||||
else:
|
else:
|
||||||
NewMessage = TypeVar("NewMessage")
|
IcaNewMessage = TypeVar("NewMessage")
|
||||||
IcaClient = TypeVar("IcaClient")
|
IcaClient = TypeVar("IcaClient")
|
||||||
|
|
||||||
def safe_eval(code: str, msg: NewMessage) -> str:
|
def safe_eval(code: str, msg: IcaNewMessage) -> str:
|
||||||
try:
|
try:
|
||||||
# code = code.replace('help', '坏东西!\n')
|
# code = code.replace('help', '坏东西!\n')
|
||||||
# code = code.replace('bytes', '坏东西!\n')
|
# code = code.replace('bytes', '坏东西!\n')
|
||||||
@ -71,7 +71,7 @@ def safe_eval(code: str, msg: NewMessage) -> str:
|
|||||||
return result
|
return result
|
||||||
|
|
||||||
|
|
||||||
def on_message(message: NewMessage, client: IcaClient) -> None:
|
def on_message(message: IcaNewMessage, client: IcaClient) -> None:
|
||||||
if not (message.is_from_self or message.is_reply):
|
if not (message.is_from_self or message.is_reply):
|
||||||
if message.content.startswith("/="):
|
if message.content.startswith("/="):
|
||||||
code = message.content[2:]
|
code = message.content[2:]
|
||||||
|
@ -3,7 +3,7 @@ pub type ClientResult<T, E> = Result<T, E>;
|
|||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
pub enum IcaError {
|
pub enum IcaError {
|
||||||
/// Socket IO 链接错误
|
/// Socket IO 链接错误
|
||||||
SocketIoError(rust_socketio::Error),
|
SocketIoError(rust_socketio::error::Error),
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
|
@ -64,18 +64,30 @@ pub async fn start_ica(config: &IcaConfig, stop_reciver: StopGetter) -> ClientRe
|
|||||||
// 等待停止信号
|
// 等待停止信号
|
||||||
stop_reciver.await.ok();
|
stop_reciver.await.ok();
|
||||||
event!(Level::INFO, "socketio client stopping");
|
event!(Level::INFO, "socketio client stopping");
|
||||||
let disconnect = socket.disconnect().await;
|
match socket.disconnect().await {
|
||||||
match disconnect {
|
|
||||||
Ok(_) => {
|
Ok(_) => {
|
||||||
event!(Level::INFO, "socketio client stopped");
|
event!(Level::INFO, "socketio client stopped");
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
Err(e) => {
|
Err(e) => {
|
||||||
event!(Level::ERROR, "socketio client stopped with error: {}", e);
|
// 单独处理 SocketIoError(IncompleteResponseFromEngineIo(WebsocketError(AlreadyClosed)))
|
||||||
Err(IcaError::SocketIoError(e))
|
match e {
|
||||||
|
rust_socketio::Error::IncompleteResponseFromEngineIo(inner_e) => {
|
||||||
|
if inner_e.to_string().contains("AlreadyClosed") {
|
||||||
|
event!(Level::INFO, "socketio client stopped");
|
||||||
|
return Ok(());
|
||||||
|
} else {
|
||||||
|
event!(Level::ERROR, "socketio client stopped with error: {:?}", inner_e);
|
||||||
|
Err(IcaError::SocketIoError(
|
||||||
|
rust_socketio::Error::IncompleteResponseFromEngineIo(inner_e),
|
||||||
|
))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
e => {
|
||||||
|
event!(Level::ERROR, "socketio client stopped with error: {}", e);
|
||||||
|
Err(IcaError::SocketIoError(e))
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// event!(Level::INFO, "socketio client stopped");
|
|
||||||
// disconnect.into()
|
|
||||||
}
|
}
|
||||||
|
@ -73,6 +73,11 @@ pub fn verify_plugins() {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub const ICA_NEW_MESSAGE_FUNC: &str = "on_ica_message";
|
||||||
|
pub const ICA_DELETE_MESSAGE_FUNC: &str = "on_ica_delete_message";
|
||||||
|
|
||||||
|
pub const MATRIX_NEW_MESSAGE_FUNC: &str = "on_matrix_message";
|
||||||
|
|
||||||
/// 执行 new message 的 python 插件
|
/// 执行 new message 的 python 插件
|
||||||
pub async fn ica_new_message_py(message: &NewMessage, client: &Client) {
|
pub async fn ica_new_message_py(message: &NewMessage, client: &Client) {
|
||||||
// 验证插件是否改变
|
// 验证插件是否改变
|
||||||
@ -80,15 +85,17 @@ pub async fn ica_new_message_py(message: &NewMessage, client: &Client) {
|
|||||||
|
|
||||||
let plugins = PyStatus::get_files();
|
let plugins = PyStatus::get_files();
|
||||||
for (path, plugin) in plugins.iter() {
|
for (path, plugin) in plugins.iter() {
|
||||||
let msg = class::NewMessagePy::new(message);
|
let msg = class::ica::NewMessagePy::new(message);
|
||||||
let client = class::IcaClientPy::new(client);
|
let client = class::ica::IcaClientPy::new(client);
|
||||||
let args = (msg, client);
|
let args = (msg, client);
|
||||||
// 甚至实际上压根不需要await这个spawn, 直接让他自己跑就好了(离谱)
|
// 甚至实际上压根不需要await这个spawn, 直接让他自己跑就好了(离谱)
|
||||||
tokio::spawn(async move {
|
tokio::spawn(async move {
|
||||||
Python::with_gil(|py| {
|
Python::with_gil(|py| {
|
||||||
if let Some(py_func) = get_func(plugin.py_module.as_ref(py), path, "on_message") {
|
if let Some(py_func) =
|
||||||
|
get_func(plugin.py_module.as_ref(py), path, ICA_NEW_MESSAGE_FUNC)
|
||||||
|
{
|
||||||
if let Err(e) = py_func.call1(args) {
|
if let Err(e) = py_func.call1(args) {
|
||||||
warn!("failed to call function<on_message>: {:?}", e);
|
warn!("failed to call function<{}>: {:?}", ICA_NEW_MESSAGE_FUNC, e);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
@ -102,15 +109,34 @@ pub async fn ica_delete_message_py(msg_id: MessageId, client: &Client) {
|
|||||||
let plugins = PyStatus::get_files();
|
let plugins = PyStatus::get_files();
|
||||||
for (path, plugin) in plugins.iter() {
|
for (path, plugin) in plugins.iter() {
|
||||||
let msg_id = msg_id.clone();
|
let msg_id = msg_id.clone();
|
||||||
let client = class::IcaClientPy::new(client);
|
let client = class::ica::IcaClientPy::new(client);
|
||||||
let args = (msg_id.clone(), client);
|
let args = (msg_id.clone(), client);
|
||||||
tokio::spawn(async move {
|
tokio::spawn(async move {
|
||||||
Python::with_gil(|py| {
|
Python::with_gil(|py| {
|
||||||
if let Some(py_func) =
|
if let Some(py_func) =
|
||||||
get_func(plugin.py_module.as_ref(py), path, "on_delete_message")
|
get_func(plugin.py_module.as_ref(py), path, ICA_DELETE_MESSAGE_FUNC)
|
||||||
{
|
{
|
||||||
if let Err(e) = py_func.call1(args) {
|
if let Err(e) = py_func.call1(args) {
|
||||||
warn!("failed to call function<on_delete_message>: {:?}", e);
|
warn!("failed to call function<{}>: {:?}", ICA_DELETE_MESSAGE_FUNC, e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
})
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn matrix_new_message_py() {
|
||||||
|
verify_plugins();
|
||||||
|
|
||||||
|
let plugins = PyStatus::get_files();
|
||||||
|
for (path, plugin) in plugins.iter() {
|
||||||
|
tokio::spawn(async move {
|
||||||
|
Python::with_gil(|py| {
|
||||||
|
if let Some(py_func) =
|
||||||
|
get_func(plugin.py_module.as_ref(py), path, MATRIX_NEW_MESSAGE_FUNC)
|
||||||
|
{
|
||||||
|
if let Err(e) = py_func.call0() {
|
||||||
|
warn!("failed to call function<{}>: {:?}", MATRIX_NEW_MESSAGE_FUNC, e);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
|
@ -1,237 +1,7 @@
|
|||||||
|
pub mod ica;
|
||||||
|
|
||||||
use pyo3::prelude::*;
|
use pyo3::prelude::*;
|
||||||
use rust_socketio::asynchronous::Client;
|
|
||||||
use tokio::runtime::Runtime;
|
|
||||||
use toml::Value as TomlValue;
|
use toml::Value as TomlValue;
|
||||||
use tracing::{debug, info, warn};
|
|
||||||
|
|
||||||
use crate::data_struct::ica::messages::{
|
|
||||||
DeleteMessage, MessageTrait, NewMessage, ReplyMessage, SendMessage,
|
|
||||||
};
|
|
||||||
use crate::data_struct::ica::MessageId;
|
|
||||||
use crate::ica::client::{delete_message, send_message};
|
|
||||||
use crate::MainStatus;
|
|
||||||
|
|
||||||
#[pyclass]
|
|
||||||
#[pyo3(name = "IcaStatus")]
|
|
||||||
pub struct IcaStatusPy {}
|
|
||||||
|
|
||||||
#[pymethods]
|
|
||||||
impl IcaStatusPy {
|
|
||||||
#[new]
|
|
||||||
pub fn py_new() -> Self { Self {} }
|
|
||||||
#[getter]
|
|
||||||
pub fn get_qq_login(&self) -> bool { MainStatus::global_ica_status().qq_login }
|
|
||||||
#[getter]
|
|
||||||
pub fn get_online(&self) -> bool { MainStatus::global_ica_status().online_status.online }
|
|
||||||
#[getter]
|
|
||||||
pub fn get_self_id(&self) -> i64 { MainStatus::global_ica_status().online_status.qqid }
|
|
||||||
#[getter]
|
|
||||||
pub fn get_nick_name(&self) -> String {
|
|
||||||
MainStatus::global_ica_status().online_status.nick.clone()
|
|
||||||
}
|
|
||||||
#[getter]
|
|
||||||
pub fn get_loaded_messages_count(&self) -> u64 {
|
|
||||||
MainStatus::global_ica_status().current_loaded_messages_count
|
|
||||||
}
|
|
||||||
#[getter]
|
|
||||||
pub fn get_ica_version(&self) -> String {
|
|
||||||
MainStatus::global_ica_status().online_status.icalingua_info.ica_version.clone()
|
|
||||||
}
|
|
||||||
|
|
||||||
#[getter]
|
|
||||||
pub fn get_os_info(&self) -> String {
|
|
||||||
MainStatus::global_ica_status().online_status.icalingua_info.os_info.clone()
|
|
||||||
}
|
|
||||||
|
|
||||||
#[getter]
|
|
||||||
pub fn get_resident_set_size(&self) -> String {
|
|
||||||
MainStatus::global_ica_status()
|
|
||||||
.online_status
|
|
||||||
.icalingua_info
|
|
||||||
.resident_set_size
|
|
||||||
.clone()
|
|
||||||
}
|
|
||||||
|
|
||||||
#[getter]
|
|
||||||
pub fn get_heap_used(&self) -> String {
|
|
||||||
MainStatus::global_ica_status().online_status.icalingua_info.heap_used.clone()
|
|
||||||
}
|
|
||||||
|
|
||||||
#[getter]
|
|
||||||
pub fn get_load(&self) -> String {
|
|
||||||
MainStatus::global_ica_status().online_status.icalingua_info.load.clone()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Default for IcaStatusPy {
|
|
||||||
fn default() -> Self { Self::new() }
|
|
||||||
}
|
|
||||||
|
|
||||||
impl IcaStatusPy {
|
|
||||||
pub fn new() -> Self { Self {} }
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Clone)]
|
|
||||||
#[pyclass]
|
|
||||||
#[pyo3(name = "NewMessage")]
|
|
||||||
pub struct NewMessagePy {
|
|
||||||
pub msg: NewMessage,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[pymethods]
|
|
||||||
impl NewMessagePy {
|
|
||||||
pub fn reply_with(&self, content: String) -> SendMessagePy {
|
|
||||||
SendMessagePy::new(self.msg.reply_with(&content))
|
|
||||||
}
|
|
||||||
pub fn as_deleted(&self) -> DeleteMessagePy { DeleteMessagePy::new(self.msg.as_deleted()) }
|
|
||||||
pub fn __str__(&self) -> String { format!("{:?}", self.msg) }
|
|
||||||
#[getter]
|
|
||||||
pub fn get_id(&self) -> MessageId { self.msg.msg_id().clone() }
|
|
||||||
#[getter]
|
|
||||||
pub fn get_content(&self) -> String { self.msg.content().clone() }
|
|
||||||
#[getter]
|
|
||||||
pub fn get_sender_id(&self) -> i64 { self.msg.sender_id() }
|
|
||||||
#[getter]
|
|
||||||
pub fn get_is_from_self(&self) -> bool { self.msg.is_from_self() }
|
|
||||||
#[getter]
|
|
||||||
pub fn get_is_reply(&self) -> bool { self.msg.is_reply() }
|
|
||||||
}
|
|
||||||
|
|
||||||
impl NewMessagePy {
|
|
||||||
pub fn new(msg: &NewMessage) -> Self { Self { msg: msg.clone() } }
|
|
||||||
}
|
|
||||||
|
|
||||||
#[pyclass]
|
|
||||||
#[pyo3(name = "ReplyMessage")]
|
|
||||||
pub struct ReplyMessagePy {
|
|
||||||
pub msg: ReplyMessage,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[pymethods]
|
|
||||||
impl ReplyMessagePy {
|
|
||||||
pub fn __str__(&self) -> String { format!("{:?}", self.msg) }
|
|
||||||
}
|
|
||||||
|
|
||||||
impl ReplyMessagePy {
|
|
||||||
pub fn new(msg: ReplyMessage) -> Self { Self { msg } }
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Clone)]
|
|
||||||
#[pyclass]
|
|
||||||
#[pyo3(name = "SendMessage")]
|
|
||||||
pub struct SendMessagePy {
|
|
||||||
pub msg: SendMessage,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[pymethods]
|
|
||||||
impl SendMessagePy {
|
|
||||||
pub fn __str__(&self) -> String { format!("{:?}", self.msg) }
|
|
||||||
/// 设置消息内容
|
|
||||||
/// 用于链式调用
|
|
||||||
pub fn with_content(&mut self, content: String) -> Self {
|
|
||||||
self.msg.content = content;
|
|
||||||
self.clone()
|
|
||||||
}
|
|
||||||
#[getter]
|
|
||||||
pub fn get_content(&self) -> String { self.msg.content.clone() }
|
|
||||||
#[setter]
|
|
||||||
pub fn set_content(&mut self, content: String) { self.msg.content = content; }
|
|
||||||
}
|
|
||||||
|
|
||||||
impl SendMessagePy {
|
|
||||||
pub fn new(msg: SendMessage) -> Self { Self { msg } }
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Clone)]
|
|
||||||
#[pyclass]
|
|
||||||
#[pyo3(name = "DeleteMessage")]
|
|
||||||
pub struct DeleteMessagePy {
|
|
||||||
pub msg: DeleteMessage,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[pymethods]
|
|
||||||
impl DeleteMessagePy {
|
|
||||||
pub fn __str__(&self) -> String { format!("{:?}", self.msg) }
|
|
||||||
}
|
|
||||||
|
|
||||||
impl DeleteMessagePy {
|
|
||||||
pub fn new(msg: DeleteMessage) -> Self { Self { msg } }
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Clone)]
|
|
||||||
#[pyclass]
|
|
||||||
#[pyo3(name = "IcaClient")]
|
|
||||||
pub struct IcaClientPy {
|
|
||||||
pub client: Client,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[pymethods]
|
|
||||||
impl IcaClientPy {
|
|
||||||
pub fn send_message(&self, message: SendMessagePy) -> bool {
|
|
||||||
tokio::task::block_in_place(|| {
|
|
||||||
let rt = Runtime::new().unwrap();
|
|
||||||
rt.block_on(send_message(&self.client, &message.msg))
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn send_and_warn(&self, message: SendMessagePy) -> bool {
|
|
||||||
warn!(message.msg.content);
|
|
||||||
tokio::task::block_in_place(|| {
|
|
||||||
let rt = Runtime::new().unwrap();
|
|
||||||
rt.block_on(send_message(&self.client, &message.msg))
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn delete_message(&self, message: DeleteMessagePy) -> bool {
|
|
||||||
tokio::task::block_in_place(|| {
|
|
||||||
let rt = Runtime::new().unwrap();
|
|
||||||
rt.block_on(delete_message(&self.client, &message.msg))
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
/// 仅作占位
|
|
||||||
/// (因为目前来说, rust调用 Python端没法启动一个异步运行时
|
|
||||||
/// 所以只能 tokio::task::block_in_place 转换成同步调用)
|
|
||||||
// #[staticmethod]
|
|
||||||
// pub fn send_message_a(
|
|
||||||
// py: Python,
|
|
||||||
// client: IcaClientPy,
|
|
||||||
// message: SendMessagePy,
|
|
||||||
// ) -> PyResult<&PyAny> {
|
|
||||||
// pyo3_asyncio::tokio::future_into_py(py, async move {
|
|
||||||
// Ok(send_message(&client.client, &message.msg).await)
|
|
||||||
// })
|
|
||||||
// }
|
|
||||||
|
|
||||||
#[getter]
|
|
||||||
pub fn get_status(&self) -> IcaStatusPy { IcaStatusPy::new() }
|
|
||||||
#[getter]
|
|
||||||
pub fn get_version(&self) -> String { crate::VERSION.to_string() }
|
|
||||||
#[getter]
|
|
||||||
pub fn get_ica_version(&self) -> String { crate::ICA_VERSION.to_string() }
|
|
||||||
#[getter]
|
|
||||||
pub fn get_matrix_version(&self) -> String { crate::MATRIX_VERSION.to_string() }
|
|
||||||
|
|
||||||
pub fn debug(&self, content: String) {
|
|
||||||
debug!("{}", content);
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn info(&self, content: String) {
|
|
||||||
info!("{}", content);
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn warn(&self, content: String) {
|
|
||||||
warn!("{}", content);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl IcaClientPy {
|
|
||||||
pub fn new(client: &Client) -> Self {
|
|
||||||
Self {
|
|
||||||
client: client.clone(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Clone)]
|
#[derive(Clone)]
|
||||||
#[pyclass]
|
#[pyclass]
|
||||||
|
239
ica-rs/src/py/class/ica.rs
Normal file
239
ica-rs/src/py/class/ica.rs
Normal file
@ -0,0 +1,239 @@
|
|||||||
|
use pyo3::prelude::*;
|
||||||
|
use rust_socketio::asynchronous::Client;
|
||||||
|
use tokio::runtime::Runtime;
|
||||||
|
use tracing::{debug, info, warn};
|
||||||
|
|
||||||
|
use crate::data_struct::ica::messages::{
|
||||||
|
DeleteMessage, MessageTrait, NewMessage, ReplyMessage, SendMessage,
|
||||||
|
};
|
||||||
|
use crate::data_struct::ica::{MessageId, RoomId, RoomIdTrait};
|
||||||
|
use crate::ica::client::{delete_message, send_message};
|
||||||
|
use crate::MainStatus;
|
||||||
|
|
||||||
|
#[pyclass]
|
||||||
|
#[pyo3(name = "IcaStatus")]
|
||||||
|
pub struct IcaStatusPy {}
|
||||||
|
|
||||||
|
#[pymethods]
|
||||||
|
impl IcaStatusPy {
|
||||||
|
#[new]
|
||||||
|
pub fn py_new() -> Self { Self {} }
|
||||||
|
#[getter]
|
||||||
|
pub fn get_qq_login(&self) -> bool { MainStatus::global_ica_status().qq_login }
|
||||||
|
#[getter]
|
||||||
|
pub fn get_online(&self) -> bool { MainStatus::global_ica_status().online_status.online }
|
||||||
|
#[getter]
|
||||||
|
pub fn get_self_id(&self) -> i64 { MainStatus::global_ica_status().online_status.qqid }
|
||||||
|
#[getter]
|
||||||
|
pub fn get_nick_name(&self) -> String {
|
||||||
|
MainStatus::global_ica_status().online_status.nick.clone()
|
||||||
|
}
|
||||||
|
#[getter]
|
||||||
|
pub fn get_loaded_messages_count(&self) -> u64 {
|
||||||
|
MainStatus::global_ica_status().current_loaded_messages_count
|
||||||
|
}
|
||||||
|
#[getter]
|
||||||
|
pub fn get_ica_version(&self) -> String {
|
||||||
|
MainStatus::global_ica_status().online_status.icalingua_info.ica_version.clone()
|
||||||
|
}
|
||||||
|
|
||||||
|
#[getter]
|
||||||
|
pub fn get_os_info(&self) -> String {
|
||||||
|
MainStatus::global_ica_status().online_status.icalingua_info.os_info.clone()
|
||||||
|
}
|
||||||
|
|
||||||
|
#[getter]
|
||||||
|
pub fn get_resident_set_size(&self) -> String {
|
||||||
|
MainStatus::global_ica_status()
|
||||||
|
.online_status
|
||||||
|
.icalingua_info
|
||||||
|
.resident_set_size
|
||||||
|
.clone()
|
||||||
|
}
|
||||||
|
|
||||||
|
#[getter]
|
||||||
|
pub fn get_heap_used(&self) -> String {
|
||||||
|
MainStatus::global_ica_status().online_status.icalingua_info.heap_used.clone()
|
||||||
|
}
|
||||||
|
|
||||||
|
#[getter]
|
||||||
|
pub fn get_load(&self) -> String {
|
||||||
|
MainStatus::global_ica_status().online_status.icalingua_info.load.clone()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Default for IcaStatusPy {
|
||||||
|
fn default() -> Self { Self::new() }
|
||||||
|
}
|
||||||
|
|
||||||
|
impl IcaStatusPy {
|
||||||
|
pub fn new() -> Self { Self {} }
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Clone)]
|
||||||
|
#[pyclass]
|
||||||
|
#[pyo3(name = "NewMessage")]
|
||||||
|
pub struct NewMessagePy {
|
||||||
|
pub msg: NewMessage,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[pymethods]
|
||||||
|
impl NewMessagePy {
|
||||||
|
pub fn reply_with(&self, content: String) -> SendMessagePy {
|
||||||
|
SendMessagePy::new(self.msg.reply_with(&content))
|
||||||
|
}
|
||||||
|
pub fn as_deleted(&self) -> DeleteMessagePy { DeleteMessagePy::new(self.msg.as_deleted()) }
|
||||||
|
pub fn __str__(&self) -> String { format!("{:?}", self.msg) }
|
||||||
|
#[getter]
|
||||||
|
pub fn get_id(&self) -> MessageId { self.msg.msg_id().clone() }
|
||||||
|
#[getter]
|
||||||
|
pub fn get_content(&self) -> String { self.msg.content().clone() }
|
||||||
|
#[getter]
|
||||||
|
pub fn get_sender_id(&self) -> i64 { self.msg.sender_id() }
|
||||||
|
#[getter]
|
||||||
|
pub fn get_is_from_self(&self) -> bool { self.msg.is_from_self() }
|
||||||
|
#[getter]
|
||||||
|
pub fn get_is_reply(&self) -> bool { self.msg.is_reply() }
|
||||||
|
#[getter]
|
||||||
|
pub fn get_is_room_msg(&self) -> bool { self.msg.room_id.is_room() }
|
||||||
|
#[getter]
|
||||||
|
pub fn get_is_chat_msg(&self) -> bool { self.msg.room_id.is_chat() }
|
||||||
|
#[getter]
|
||||||
|
pub fn get_room_id(&self) -> RoomId { self.msg.room_id.clone() }
|
||||||
|
}
|
||||||
|
|
||||||
|
impl NewMessagePy {
|
||||||
|
pub fn new(msg: &NewMessage) -> Self { Self { msg: msg.clone() } }
|
||||||
|
}
|
||||||
|
|
||||||
|
#[pyclass]
|
||||||
|
#[pyo3(name = "ReplyMessage")]
|
||||||
|
pub struct ReplyMessagePy {
|
||||||
|
pub msg: ReplyMessage,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[pymethods]
|
||||||
|
impl ReplyMessagePy {
|
||||||
|
pub fn __str__(&self) -> String { format!("{:?}", self.msg) }
|
||||||
|
}
|
||||||
|
|
||||||
|
impl ReplyMessagePy {
|
||||||
|
pub fn new(msg: ReplyMessage) -> Self { Self { msg } }
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Clone)]
|
||||||
|
#[pyclass]
|
||||||
|
#[pyo3(name = "SendMessage")]
|
||||||
|
pub struct SendMessagePy {
|
||||||
|
pub msg: SendMessage,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[pymethods]
|
||||||
|
impl SendMessagePy {
|
||||||
|
pub fn __str__(&self) -> String { format!("{:?}", self.msg) }
|
||||||
|
/// 设置消息内容
|
||||||
|
/// 用于链式调用
|
||||||
|
pub fn with_content(&mut self, content: String) -> Self {
|
||||||
|
self.msg.content = content;
|
||||||
|
self.clone()
|
||||||
|
}
|
||||||
|
#[getter]
|
||||||
|
pub fn get_content(&self) -> String { self.msg.content.clone() }
|
||||||
|
#[setter]
|
||||||
|
pub fn set_content(&mut self, content: String) { self.msg.content = content; }
|
||||||
|
}
|
||||||
|
|
||||||
|
impl SendMessagePy {
|
||||||
|
pub fn new(msg: SendMessage) -> Self { Self { msg } }
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Clone)]
|
||||||
|
#[pyclass]
|
||||||
|
#[pyo3(name = "DeleteMessage")]
|
||||||
|
pub struct DeleteMessagePy {
|
||||||
|
pub msg: DeleteMessage,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[pymethods]
|
||||||
|
impl DeleteMessagePy {
|
||||||
|
pub fn __str__(&self) -> String { format!("{:?}", self.msg) }
|
||||||
|
}
|
||||||
|
|
||||||
|
impl DeleteMessagePy {
|
||||||
|
pub fn new(msg: DeleteMessage) -> Self { Self { msg } }
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Clone)]
|
||||||
|
#[pyclass]
|
||||||
|
#[pyo3(name = "IcaClient")]
|
||||||
|
pub struct IcaClientPy {
|
||||||
|
pub client: Client,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[pymethods]
|
||||||
|
impl IcaClientPy {
|
||||||
|
pub fn send_message(&self, message: SendMessagePy) -> bool {
|
||||||
|
tokio::task::block_in_place(|| {
|
||||||
|
let rt = Runtime::new().unwrap();
|
||||||
|
rt.block_on(send_message(&self.client, &message.msg))
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn send_and_warn(&self, message: SendMessagePy) -> bool {
|
||||||
|
warn!(message.msg.content);
|
||||||
|
tokio::task::block_in_place(|| {
|
||||||
|
let rt = Runtime::new().unwrap();
|
||||||
|
rt.block_on(send_message(&self.client, &message.msg))
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn delete_message(&self, message: DeleteMessagePy) -> bool {
|
||||||
|
tokio::task::block_in_place(|| {
|
||||||
|
let rt = Runtime::new().unwrap();
|
||||||
|
rt.block_on(delete_message(&self.client, &message.msg))
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
/// 仅作占位
|
||||||
|
/// (因为目前来说, rust调用 Python端没法启动一个异步运行时
|
||||||
|
/// 所以只能 tokio::task::block_in_place 转换成同步调用)
|
||||||
|
// #[staticmethod]
|
||||||
|
// pub fn send_message_a(
|
||||||
|
// py: Python,
|
||||||
|
// client: IcaClientPy,
|
||||||
|
// message: SendMessagePy,
|
||||||
|
// ) -> PyResult<&PyAny> {
|
||||||
|
// pyo3_asyncio::tokio::future_into_py(py, async move {
|
||||||
|
// Ok(send_message(&client.client, &message.msg).await)
|
||||||
|
// })
|
||||||
|
// }
|
||||||
|
|
||||||
|
#[getter]
|
||||||
|
pub fn get_status(&self) -> IcaStatusPy { IcaStatusPy::new() }
|
||||||
|
#[getter]
|
||||||
|
pub fn get_version(&self) -> String { crate::VERSION.to_string() }
|
||||||
|
#[getter]
|
||||||
|
pub fn get_ica_version(&self) -> String { crate::ICA_VERSION.to_string() }
|
||||||
|
#[getter]
|
||||||
|
pub fn get_matrix_version(&self) -> String { crate::MATRIX_VERSION.to_string() }
|
||||||
|
|
||||||
|
pub fn debug(&self, content: String) {
|
||||||
|
debug!("{}", content);
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn info(&self, content: String) {
|
||||||
|
info!("{}", content);
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn warn(&self, content: String) {
|
||||||
|
warn!("{}", content);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl IcaClientPy {
|
||||||
|
pub fn new(client: &Client) -> Self {
|
||||||
|
Self {
|
||||||
|
client: client.clone(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
10
news.md
10
news.md
@ -1,5 +1,15 @@
|
|||||||
# 更新日志
|
# 更新日志
|
||||||
|
|
||||||
|
## 0.5.3
|
||||||
|
|
||||||
|
修复了 Icalingua 断开时 如果 socketio 已经断开会导致程序 返回 Error 的问题
|
||||||
|
以及还有一些别的修复就是了
|
||||||
|
|
||||||
|
- Python 端修改
|
||||||
|
- `on_message` -> `on_ica_message`
|
||||||
|
- `on_delete_message` -> `on_ica_delete_message`
|
||||||
|
- 添加 `on_matrix_message`
|
||||||
|
|
||||||
## 0.5.1/2
|
## 0.5.1/2
|
||||||
|
|
||||||
重构了一整波, 还没改 `ica-typing.py` 的代码
|
重构了一整波, 还没改 `ica-typing.py` 的代码
|
||||||
|
Loading…
Reference in New Issue
Block a user