添加文件列表展示

Author: pysrc

README.md

@@ -14,6 +14,8 @@
 
 * https://github.com/quadrupleslap/scrap
 * https://github.com/enigo-rs/enigo
+* https://github.com/aweinstock314/rust-clipboard
+
 
 ## 计划中的功能
 

dsweb-core/src/server.rs

@@ -9,6 +9,7 @@ use enigo::MouseControllable;
 use flate2::write::DeflateEncoder;
 use flate2::Compression;
 use websocket::sync::Client;
+use std::fs;
 use std::io::Write;
 use std::net::Ipv4Addr;
 use std::net::SocketAddrV4;
@@ -198,6 +199,22 @@ fn screen_stream(mut stream: Writer<TcpStream>, running: Arc<AtomicBool>) {
     }
 }
 
+fn get_files(dir: &str) -> Vec<String> {
+    let mut res = Vec::<String>::new();
+    let mt = fs::metadata(dir);
+    if mt.is_err() {
+        fs::create_dir(dir).unwrap();
+        return res;
+    }
+    let folder = fs::read_dir(dir).unwrap();
+    for file in folder {
+        let f = file.unwrap();
+        if f.file_type().unwrap().is_file() {
+            res.push(f.path().file_name().unwrap().to_string_lossy().to_string());
+        }
+    }
+    return res;
+}
 
 fn handle_transfer(client: Client<TcpStream>) {
     let (mut receiver, mut sender) = client.split().unwrap();
@@ -220,6 +237,11 @@ fn handle_transfer(client: Client<TcpStream>) {
                         txt.insert_str(0, "copy-text ");
                         sender.send_message(&OwnedMessage::Text(txt)).unwrap();
                     }
+                } else if ctx.starts_with("file-list") {
+                    let fss = get_files("files");
+                    let mut res = fss.join("&");
+                    res.insert_str(0, "file-list ");
+                    sender.send_message(&OwnedMessage::Text(res)).unwrap();
                 }
             }
             OwnedMessage::Ping(ping) => {

dsweb-front/index.html

@@ -23,12 +23,27 @@
         <button onclick="dsweb.copy_text()">复制出来（文本）</button>
         <button onclick="dsweb.paste_text()">复制进去（文本）</button>
     </div>
+    <div style="margin-top: 10px;margin-left: 30%;">
+        <button onclick="file_list_refresh()">文件列表刷新</button>
+        <select id="files" style="width: 200px;"></select>
+        <button onclick="">下载选中文件</button>
+        <button>文件上传</button>
+    </div>
 </body>
 <script>
     var scale_rate = 0.05;
     var hostdom = document.getElementById("host");
     function start() {
-        dsweb.start("canvas", "transfer", hostdom.value);
+        dsweb.start("canvas", "transfer", hostdom.value, (files) => {
+            var doms = "";
+            for (var index = 0; index < files.length; index++) {
+                const element = files[index];
+                var sdom = `<option value="${element}">${element}</option>`;
+                doms += sdom;
+            }
+            var files = document.getElementById("files");
+            files.innerHTML = doms;
+        });
     }
     function stop() {
         dsweb.stop();
@@ -45,5 +60,8 @@
         c.style.width = `${(1 - scale_rate) * rect.width}px`;
         c.style.height = `${(1 - scale_rate) * rect.height}px`;
     }
+    function file_list_refresh() {
+        dsweb.file_list_refresh();
+    }
 </script>
 </html>

dsweb-front/src/main.js

@@ -4,7 +4,7 @@ var ws = null;
 var transfer_ws = null;
 var transferdom = null;
 
-const start = (canvasid, transferid, url) => {    
+const start = (canvasid, transferid, url, files_callback) => {    
     init().then(()=>{
         ws = start_websocket(canvasid, url);
         transferdom = document.getElementById(transferid);
@@ -17,6 +17,12 @@ const start = (canvasid, transferid, url) => {
             if(d.startsWith("copy-text")) {
                 d = d.replace("copy-text ", "");
                 transferdom.value = d;
+            } else if(d.startsWith("file-list")) {
+                d = d.replace("file-list ", "");
+                if(d) {
+                    var files = d.split("&");
+                    files_callback(files);
+                }
             }
         };
     });
@@ -36,9 +42,14 @@ const copy_text = () => {
     transfer_ws.send("copy-text");
 }
 
+const file_list_refresh = () => {
+    transfer_ws.send("file-list");
+}
+
 export {
     start,
     stop,
     paste_text,
     copy_text,
+    file_list_refresh,
 };