better conditional compilation

Signed-off-by: dignow <linlong1265@gmail.com>
This commit is contained in:
dignow 2023-06-24 21:09:45 +08:00
parent b9652d165b
commit d15deabba5
10 changed files with 147 additions and 94 deletions

View File

@ -120,12 +120,11 @@ class ServerModel with ChangeNotifier {
_serverId = IDTextEditingController(text: _emptyIdShow); _serverId = IDTextEditingController(text: _emptyIdShow);
timerCallback() async { timerCallback() async {
var status = await bind.mainGetOnlineStatue(); final connectionStatus =
if (status > 0) { jsonDecode(await bind.mainGetConnectStatus()) as Map<String, dynamic>;
status = 1; final statusNum = connectionStatus['status_num'] as int;
} if (statusNum != _connectStatus) {
if (status != _connectStatus) { _connectStatus = statusNum;
_connectStatus = status;
notifyListeners(); notifyListeners();
} }

View File

@ -44,7 +44,7 @@ lazy_static::lazy_static! {
static ref CONFIG: Arc<RwLock<Config>> = Arc::new(RwLock::new(Config::load())); static ref CONFIG: Arc<RwLock<Config>> = Arc::new(RwLock::new(Config::load()));
static ref CONFIG2: Arc<RwLock<Config2>> = Arc::new(RwLock::new(Config2::load())); static ref CONFIG2: Arc<RwLock<Config2>> = Arc::new(RwLock::new(Config2::load()));
static ref LOCAL_CONFIG: Arc<RwLock<LocalConfig>> = Arc::new(RwLock::new(LocalConfig::load())); static ref LOCAL_CONFIG: Arc<RwLock<LocalConfig>> = Arc::new(RwLock::new(LocalConfig::load()));
pub static ref ONLINE: Arc<Mutex<HashMap<String, i64>>> = Default::default(); static ref ONLINE: Arc<Mutex<HashMap<String, i64>>> = Default::default();
pub static ref PROD_RENDEZVOUS_SERVER: Arc<RwLock<String>> = Arc::new(RwLock::new(match option_env!("RENDEZVOUS_SERVER") { pub static ref PROD_RENDEZVOUS_SERVER: Arc<RwLock<String>> = Arc::new(RwLock::new(match option_env!("RENDEZVOUS_SERVER") {
Some(key) if !key.is_empty() => key, Some(key) if !key.is_empty() => key,
_ => "", _ => "",
@ -309,6 +309,11 @@ pub struct TransferSerde {
pub read_jobs: Vec<String>, pub read_jobs: Vec<String>,
} }
#[inline]
pub fn get_online_statue() -> i64 {
*ONLINE.lock().unwrap().values().max().unwrap_or(&0)
}
#[cfg(not(any(target_os = "android", target_os = "ios")))] #[cfg(not(any(target_os = "android", target_os = "ios")))]
fn patch(path: PathBuf) -> PathBuf { fn patch(path: PathBuf) -> PathBuf {
if let Some(_tmp) = path.to_str() { if let Some(_tmp) = path.to_str() {

View File

@ -1,20 +1,21 @@
extern crate repng; extern crate repng;
extern crate scrap; extern crate scrap;
use std::fs::File; use scrap::Display;
use scrap::{i420_to_rgb, Display};
#[cfg(windows)] #[cfg(windows)]
use scrap::{CapturerMag, TraitCapturer}; use scrap::{i420_to_rgb, CapturerMag, TraitCapturer};
#[cfg(windows)]
use std::fs::File;
fn main() { fn main() {
let n = Display::all().unwrap().len(); let n = Display::all().unwrap().len();
for i in 0..n { for _i in 0..n {
#[cfg(windows)] #[cfg(windows)]
record(i); record(_i);
} }
} }
#[cfg(windows)]
fn get_display(i: usize) -> Display { fn get_display(i: usize) -> Display {
Display::all().unwrap().remove(i) Display::all().unwrap().remove(i)
} }

View File

@ -1,7 +1,8 @@
#[cfg(windows)] #[cfg(windows)]
pub mod win10; pub mod win10;
#[cfg(windows)]
use hbb_common::{bail, lazy_static, ResultType}; use hbb_common::lazy_static;
use hbb_common::{bail, ResultType};
use std::path::Path; use std::path::Path;
#[cfg(windows)] #[cfg(windows)]

View File

@ -14,12 +14,11 @@ use flutter_rust_bridge::{StreamSink, SyncReturn};
#[cfg(not(any(target_os = "android", target_os = "ios")))] #[cfg(not(any(target_os = "android", target_os = "ios")))]
use hbb_common::allow_err; use hbb_common::allow_err;
use hbb_common::{ use hbb_common::{
config::{self, LocalConfig, PeerConfig, PeerInfoSerde, ONLINE}, config::{self, LocalConfig, PeerConfig, PeerInfoSerde},
fs, log, fs, log,
message_proto::KeyboardMode, message_proto::KeyboardMode,
ResultType, ResultType,
}; };
use serde_json::json;
use std::{ use std::{
collections::HashMap, collections::HashMap,
ffi::{CStr, CString}, ffi::{CStr, CString},
@ -680,14 +679,18 @@ pub fn main_get_lan_peers() -> String {
} }
pub fn main_get_connect_status() -> String { pub fn main_get_connect_status() -> String {
let status = get_connect_status(); #[cfg(not(any(target_os = "android", target_os = "ios")))]
// (status_num, key_confirmed, mouse_time, id) {
let mut m = serde_json::Map::new(); serde_json::to_string(&get_connect_status()).unwrap_or("".to_string())
m.insert("status_num".to_string(), json!(status.0)); }
m.insert("key_confirmed".to_string(), json!(status.1)); #[cfg(any(target_os = "android", target_os = "ios"))]
m.insert("mouse_time".to_string(), json!(status.2)); {
m.insert("id".to_string(), json!(status.3)); let mut state = hbb_common::config::get_online_statue();
serde_json::to_string(&m).unwrap_or("".to_string()) if state > 0 {
state = 1;
}
serde_json::json!({ "status_num": state }).to_string()
}
} }
pub fn main_check_connect_status() { pub fn main_check_connect_status() {
@ -995,10 +998,6 @@ pub fn main_get_fingerprint() -> String {
get_fingerprint() get_fingerprint()
} }
pub fn main_get_online_statue() -> i64 {
ONLINE.lock().unwrap().values().max().unwrap_or(&0).clone()
}
pub fn cm_get_clients_state() -> String { pub fn cm_get_clients_state() -> String {
crate::ui_cm_interface::get_clients_state() crate::ui_cm_interface::get_clients_state()
} }
@ -1194,8 +1193,15 @@ pub fn main_check_mouse_time() {
} }
pub fn main_get_mouse_time() -> f64 { pub fn main_get_mouse_time() -> f64 {
#[cfg(not(any(target_os = "android", target_os = "ios")))]
{
get_mouse_time() get_mouse_time()
} }
#[cfg(any(target_os = "android", target_os = "ios"))]
{
0.0
}
}
pub fn main_wol(id: String) { pub fn main_wol(id: String) {
// TODO: move send_wol outside. // TODO: move send_wol outside.

View File

@ -186,6 +186,7 @@ pub enum Data {
}, },
SystemInfo(Option<String>), SystemInfo(Option<String>),
ClickTime(i64), ClickTime(i64),
#[cfg(not(any(target_os = "android", target_os = "ios")))]
MouseMoveTime(i64), MouseMoveTime(i64),
Authorize, Authorize,
Close, Close,
@ -332,6 +333,7 @@ async fn handle(data: Data, stream: &mut Connection) {
let t = crate::server::CLICK_TIME.load(Ordering::SeqCst); let t = crate::server::CLICK_TIME.load(Ordering::SeqCst);
allow_err!(stream.send(&Data::ClickTime(t)).await); allow_err!(stream.send(&Data::ClickTime(t)).await);
} }
#[cfg(not(any(target_os = "android", target_os = "ios")))]
Data::MouseMoveTime(_) => { Data::MouseMoveTime(_) => {
let t = crate::server::MOUSE_MOVE_TIME.load(Ordering::SeqCst); let t = crate::server::MOUSE_MOVE_TIME.load(Ordering::SeqCst);
allow_err!(stream.send(&Data::MouseMoveTime(t)).await); allow_err!(stream.send(&Data::MouseMoveTime(t)).await);
@ -345,13 +347,7 @@ async fn handle(data: Data, stream: &mut Connection) {
} }
} }
Data::OnlineStatus(_) => { Data::OnlineStatus(_) => {
let x = config::ONLINE let x = config::get_online_statue();
.lock()
.unwrap()
.values()
.max()
.unwrap_or(&0)
.clone();
let confirmed = Config::get_key_confirmed(); let confirmed = Config::get_key_confirmed();
allow_err!(stream.send(&Data::OnlineStatus(Some((x, confirmed)))).await); allow_err!(stream.send(&Data::OnlineStatus(Some((x, confirmed)))).await);
} }

View File

@ -63,6 +63,7 @@ lazy_static::lazy_static! {
static ref SWITCH_SIDES_UUID: Arc::<Mutex<HashMap<String, (Instant, uuid::Uuid)>>> = Default::default(); static ref SWITCH_SIDES_UUID: Arc::<Mutex<HashMap<String, (Instant, uuid::Uuid)>>> = Default::default();
} }
pub static CLICK_TIME: AtomicI64 = AtomicI64::new(0); pub static CLICK_TIME: AtomicI64 = AtomicI64::new(0);
#[cfg(not(any(target_os = "android", target_os = "ios")))]
pub static MOUSE_MOVE_TIME: AtomicI64 = AtomicI64::new(0); pub static MOUSE_MOVE_TIME: AtomicI64 = AtomicI64::new(0);
#[cfg(all(feature = "flutter", feature = "plugin_framework"))] #[cfg(all(feature = "flutter", feature = "plugin_framework"))]
@ -163,6 +164,7 @@ pub struct Connection {
// by peer // by peer
disable_audio: bool, disable_audio: bool,
// by peer // by peer
#[cfg(windows)]
enable_file_transfer: bool, enable_file_transfer: bool,
// by peer // by peer
audio_sender: Option<MediaSender>, audio_sender: Option<MediaSender>,
@ -291,6 +293,7 @@ impl Connection {
show_remote_cursor: false, show_remote_cursor: false,
ip: "".to_owned(), ip: "".to_owned(),
disable_audio: false, disable_audio: false,
#[cfg(windows)]
enable_file_transfer: false, enable_file_transfer: false,
disable_clipboard: false, disable_clipboard: false,
disable_keyboard: false, disable_keyboard: false,
@ -1112,6 +1115,7 @@ impl Connection {
self.audio && !self.disable_audio self.audio && !self.disable_audio
} }
#[cfg(windows)]
fn file_transfer_enabled(&self) -> bool { fn file_transfer_enabled(&self) -> bool {
self.file && self.enable_file_transfer self.file && self.enable_file_transfer
} }

View File

@ -362,9 +362,9 @@ impl UI {
fn get_connect_status(&mut self) -> Value { fn get_connect_status(&mut self) -> Value {
let mut v = Value::array(0); let mut v = Value::array(0);
let x = get_connect_status(); let x = get_connect_status();
v.push(x.0); v.push(x.status_num);
v.push(x.1); v.push(x.key_confirmed);
v.push(x.3); v.push(x.id);
v v
} }

View File

@ -454,10 +454,10 @@ impl<T: InvokeUiCM> IpcTaskRunner<T> {
} }
} }
Some(data) = self.rx.recv() => { Some(data) = self.rx.recv() => {
if let Data::SwitchPermission{name, enabled} = &data { if let Data::SwitchPermission{name: _name, enabled: _enabled} = &data {
#[cfg(windows)] #[cfg(windows)]
if name == "file" { if _name == "file" {
self.file_transfer_enabled = *enabled; self.file_transfer_enabled = *_enabled;
} }
} }
if self.stream.send(&data).await.is_err() { if self.stream.send(&data).await.is_err() {

View File

@ -1,9 +1,3 @@
use std::{
collections::HashMap,
process::Child,
sync::{Arc, Mutex},
};
#[cfg(any(target_os = "android", target_os = "ios"))] #[cfg(any(target_os = "android", target_os = "ios"))]
use hbb_common::password_security; use hbb_common::password_security;
use hbb_common::{ use hbb_common::{
@ -11,16 +5,21 @@ use hbb_common::{
config::{self, Config, LocalConfig, PeerConfig}, config::{self, Config, LocalConfig, PeerConfig},
directories_next, log, tokio, directories_next, log, tokio,
}; };
use hbb_common::{
config::{CONNECT_TIMEOUT, RENDEZVOUS_PORT},
futures::future::join_all,
rendezvous_proto::*,
};
#[cfg(not(any(target_os = "android", target_os = "ios")))] #[cfg(not(any(target_os = "android", target_os = "ios")))]
use hbb_common::{ use hbb_common::{
sleep, sleep,
tokio::{sync::mpsc, time}, tokio::{sync::mpsc, time},
}; };
use serde_derive::Serialize;
use hbb_common::{ use std::{
config::{CONNECT_TIMEOUT, RENDEZVOUS_PORT}, collections::HashMap,
futures::future::join_all, process::Child,
rendezvous_proto::*, sync::{Arc, Mutex},
}; };
use crate::common::SOFTWARE_UPDATE_URL; use crate::common::SOFTWARE_UPDATE_URL;
@ -32,10 +31,28 @@ use crate::ipc;
type Message = RendezvousMessage; type Message = RendezvousMessage;
pub type Children = Arc<Mutex<(bool, HashMap<(String, String), Child>)>>; pub type Children = Arc<Mutex<(bool, HashMap<(String, String), Child>)>>;
type Status = (i32, bool, i64, String); // (status_num, key_confirmed, mouse_time, id)
#[derive(Clone, Debug, Serialize)]
pub struct UiStatus {
pub status_num: i32,
#[cfg(not(feature = "flutter"))]
pub key_confirmed: bool,
#[cfg(not(any(target_os = "android", target_os = "ios")))]
pub mouse_time: i64,
#[cfg(not(feature = "flutter"))]
pub id: String,
}
lazy_static::lazy_static! { lazy_static::lazy_static! {
static ref UI_STATUS : Arc<Mutex<Status>> = Arc::new(Mutex::new((0, false, 0, "".to_owned()))); static ref UI_STATUS : Arc<Mutex<UiStatus>> = Arc::new(Mutex::new(UiStatus{
status_num: 0,
#[cfg(not(feature = "flutter"))]
key_confirmed: false,
#[cfg(not(any(target_os = "android", target_os = "ios")))]
mouse_time: 0,
#[cfg(not(feature = "flutter"))]
id: "".to_owned(),
}));
static ref ASYNC_JOB_STATUS : Arc<Mutex<String>> = Default::default(); static ref ASYNC_JOB_STATUS : Arc<Mutex<String>> = Default::default();
static ref TEMPORARY_PASSWD : Arc<Mutex<String>> = Arc::new(Mutex::new("".to_owned())); static ref TEMPORARY_PASSWD : Arc<Mutex<String>> = Arc::new(Mutex::new("".to_owned()));
} }
@ -408,10 +425,9 @@ pub fn is_installed_lower_version() -> bool {
} }
#[inline] #[inline]
#[cfg(not(any(target_os = "android", target_os = "ios")))]
pub fn get_mouse_time() -> f64 { pub fn get_mouse_time() -> f64 {
let ui_status = UI_STATUS.lock().unwrap(); UI_STATUS.lock().unwrap().mouse_time as f64
let res = ui_status.2 as f64;
return res;
} }
#[inline] #[inline]
@ -424,10 +440,9 @@ pub fn check_mouse_time() {
} }
#[inline] #[inline]
pub fn get_connect_status() -> Status { #[cfg(not(any(target_os = "android", target_os = "ios")))]
let ui_statue = UI_STATUS.lock().unwrap(); pub fn get_connect_status() -> UiStatus {
let res = ui_statue.clone(); UI_STATUS.lock().unwrap().clone()
res
} }
#[inline] #[inline]
@ -872,10 +887,13 @@ pub fn get_hostname() -> String {
#[cfg(not(any(target_os = "android", target_os = "ios")))] #[cfg(not(any(target_os = "android", target_os = "ios")))]
#[tokio::main(flavor = "current_thread")] #[tokio::main(flavor = "current_thread")]
async fn check_connect_status_(reconnect: bool, rx: mpsc::UnboundedReceiver<ipc::Data>) { async fn check_connect_status_(reconnect: bool, rx: mpsc::UnboundedReceiver<ipc::Data>) {
#[cfg(not(feature = "flutter"))]
let mut key_confirmed = false; let mut key_confirmed = false;
let mut rx = rx; let mut rx = rx;
let mut mouse_time = 0; let mut mouse_time = 0;
#[cfg(not(feature = "flutter"))]
let mut id = "".to_owned(); let mut id = "".to_owned();
#[cfg(target_os = "windows")]
let mut enable_file_transfer = "".to_owned(); let mut enable_file_transfer = "".to_owned();
loop { loop {
@ -889,9 +907,10 @@ async fn check_connect_status_(reconnect: bool, rx: mpsc::UnboundedReceiver<ipc:
log::error!("ipc connection closed: {}", err); log::error!("ipc connection closed: {}", err);
break; break;
} }
#[cfg(not(any(target_os = "android", target_os = "ios")))]
Ok(Some(ipc::Data::MouseMoveTime(v))) => { Ok(Some(ipc::Data::MouseMoveTime(v))) => {
mouse_time = v; mouse_time = v;
UI_STATUS.lock().unwrap().2 = v; UI_STATUS.lock().unwrap().mouse_time = v;
} }
Ok(Some(ipc::Data::Options(Some(v)))) => { Ok(Some(ipc::Data::Options(Some(v)))) => {
*OPTIONS.lock().unwrap() = v; *OPTIONS.lock().unwrap() = v;
@ -908,17 +927,31 @@ async fn check_connect_status_(reconnect: bool, rx: mpsc::UnboundedReceiver<ipc:
} }
Ok(Some(ipc::Data::Config((name, Some(value))))) => { Ok(Some(ipc::Data::Config((name, Some(value))))) => {
if name == "id" { if name == "id" {
#[cfg(not(feature = "flutter"))]
{
id = value; id = value;
}
} else if name == "temporary-password" { } else if name == "temporary-password" {
*TEMPORARY_PASSWD.lock().unwrap() = value; *TEMPORARY_PASSWD.lock().unwrap() = value;
} }
} }
Ok(Some(ipc::Data::OnlineStatus(Some((mut x, c))))) => { Ok(Some(ipc::Data::OnlineStatus(Some((mut x, _c))))) => {
if x > 0 { if x > 0 {
x = 1 x = 1
} }
key_confirmed = c; #[cfg(not(feature = "flutter"))]
*UI_STATUS.lock().unwrap() = (x as _, key_confirmed, mouse_time, id.clone()); {
key_confirmed = _c;
}
*UI_STATUS.lock().unwrap() = UiStatus {
status_num: x as _,
#[cfg(not(feature = "flutter"))]
key_confirmed: _c,
#[cfg(not(any(target_os = "android", target_os = "ios")))]
mouse_time,
#[cfg(not(feature = "flutter"))]
id: id.clone(),
};
} }
_ => {} _ => {}
} }
@ -942,7 +975,15 @@ async fn check_connect_status_(reconnect: bool, rx: mpsc::UnboundedReceiver<ipc:
.insert("ipc-closed".to_owned(), "Y".to_owned()); .insert("ipc-closed".to_owned(), "Y".to_owned());
break; break;
} }
*UI_STATUS.lock().unwrap() = (-1, key_confirmed, mouse_time, id.clone()); *UI_STATUS.lock().unwrap() = UiStatus {
status_num: -1,
#[cfg(not(feature = "flutter"))]
key_confirmed,
#[cfg(not(any(target_os = "android", target_os = "ios")))]
mouse_time,
#[cfg(not(feature = "flutter"))]
id: id.clone(),
};
sleep(1.).await; sleep(1.).await;
} }
} }