添加文本复制功能

Author: pysrc

Cargo.lock

@@ -10,4 +10,5 @@ dscom = {path = "../dscom"}
 websocket = "0.26.5"
 scrap = "0.5"
 enigo = "0.1.2"
-flate2 = "1.0"
+flate2 = "1.0"
+clipboard = "0.5"

dsweb-core/Cargo.toml

@@ -1,11 +1,14 @@
 use crate::config;
 use crate::key_mouse;
 use crate::screen::Cap;
+use clipboard::ClipboardContext;
+use clipboard::ClipboardProvider;
 use enigo::Enigo;
 use enigo::KeyboardControllable;
 use enigo::MouseControllable;
 use flate2::write::DeflateEncoder;
 use flate2::Compression;
+use websocket::sync::Client;
 use std::io::Write;
 use std::net::Ipv4Addr;
 use std::net::SocketAddrV4;
@@ -49,8 +52,12 @@ pub fn run(port: u16) {
                     println!("Break !");
                 });
             }
-            "paste" => {
-                // 处理粘贴
+            "diffscreen-transfer" => {
+                // 处理复制粘贴、文件
+                let client = request.use_protocol("diffscreen-transfer").accept().unwrap();
+                std::thread::spawn(move ||{
+                    handle_transfer(client);
+                });
             }
             _ => {
                 request.reject().unwrap();
@@ -137,7 +144,7 @@ fn get_rgb_block(
     let srw = (w / sw) + if w % sw == 0 { 0usize } else { 1usize };
     let srh = (h / sh) + if h % sh == 0 { 0usize } else { 1usize };
     let num = srw * srh;
-    let mut buf = vec![vec![0u8; offset + sw * sh * 3]; num];
+    let mut buf = vec![vec![255u8; offset + sw * sh * 3]; num];
     for i in 0..(srw * srh) {
         buf[i][offset - 2] = (i >> 8) as u8;
         buf[i][offset - 1] = i as u8;
@@ -190,3 +197,40 @@ fn screen_stream(mut stream: Writer<TcpStream>, running: Arc<AtomicBool>) {
         (a, b) = (b, a);
     }
 }
+
+
+fn handle_transfer(client: Client<TcpStream>) {
+    let (mut receiver, mut sender) = client.split().unwrap();
+    let mut cbctx: ClipboardContext = ClipboardProvider::new().unwrap();
+    for message in receiver.incoming_messages() {
+        let message = match message {
+            Ok(message) => message,
+            Err(e) => {
+                eprintln!("Msg err {}", e);
+                return;
+            }
+        };
+        match message {
+            OwnedMessage::Text(ctx) => {
+                if ctx.starts_with("paste-text") {
+                    let ctx = ctx.replacen("paste-text ", "", 1);
+                    cbctx.set_contents(ctx).unwrap();
+                } else if ctx.starts_with("copy-text") {
+                    if let Ok(mut txt) = cbctx.get_contents() {
+                        txt.insert_str(0, "copy-text ");
+                        sender.send_message(&OwnedMessage::Text(txt)).unwrap();
+                    }
+                }
+            }
+            OwnedMessage::Ping(ping) => {
+                let message = OwnedMessage::Pong(ping);
+                sender.send_message(&message).unwrap();
+            }
+            OwnedMessage::Close(_) => {
+                println!("Front transfer close !");
+                return;
+            }
+            _ => {}
+        }
+    }
+}

dsweb-core/src/server.rs

@@ -16,12 +16,19 @@
         <button onclick="add()">+</button>
         <button onclick="sub()">-</button>
     </div>
+    <div style="margin-top: 10px;margin-left: 30%;">
+        <div>
+            <textarea id="transfer" style="height: 100px;width: 50%;"></textarea>
+        </div>
+        <button onclick="dsweb.copy_text()">复制出来（文本）</button>
+        <button onclick="dsweb.paste_text()">复制进去（文本）</button>
+    </div>
 </body>
 <script>
     var scale_rate = 0.05;
     var hostdom = document.getElementById("host");
     function start() {
-        dsweb.start("canvas", hostdom.value);
+        dsweb.start("canvas", "transfer", hostdom.value);
     }
     function stop() {
         dsweb.stop();

dsweb-front/index.html

@@ -1,18 +1,44 @@
 import init, { start_websocket } from "../pkg/dsweb_front"
 
 var ws = null;
+var transfer_ws = null;
+var transferdom = null;
 
-const start = (canvasid, url) => {    
+const start = (canvasid, transferid, url) => {    
     init().then(()=>{
         ws = start_websocket(canvasid, url);
+        transferdom = document.getElementById(transferid);
+        transfer_ws = new WebSocket(url, "diffscreen-transfer");
+        transfer_ws.onopen = function(event) {
+            console.log('Transfer Websocket 已经连接');
+        };
+        transfer_ws.onmessage = function(event) {
+            var d = event.data;
+            if(d.startsWith("copy-text")) {
+                d = d.replace("copy-text ", "");
+                transferdom.value = d;
+            }
+        };
     });
 }
 
 const stop = () => {
     ws.close();
+    transfer_ws.close();
+}
+
+const paste_text = () => {
+    var msg = transferdom.value;
+    transfer_ws.send(`paste-text ${msg}`);
+};
+
+const copy_text = () => {
+    transfer_ws.send("copy-text");
 }
 
 export {
     start,
-    stop
+    stop,
+    paste_text,
+    copy_text,
 };