support windows

Author: 9001

copyparty/httpcli.py

@@ -7,6 +7,7 @@
 import time
 import json
 import socket
+import ctypes
 from datetime import datetime
 import calendar
 
@@ -867,7 +868,7 @@ def tx_file(self, req_path):
 
         if not is_compressed:
             self.out_headers["Cache-Control"] = "no-cache"
-        
+
         self.out_headers["Accept-Ranges"] = "bytes"
         self.send_headers(
             length=upper - lower,
@@ -1078,23 +1079,30 @@ def tx_browser(self):
                     break
 
         srv_info = []
-        
+
         try:
             if not self.args.nih:
-                srv_info.append(str(socket.gethostname()).split('.')[0])
+                srv_info.append(str(socket.gethostname()).split(".")[0])
         except:
             self.log("#wow #whoa")
             pass
 
         try:
             # some fuses misbehave
             if not self.args.nid:
-                sv = os.statvfs(abspath)
-                free = humansize(sv.f_frsize * sv.f_bfree, True)
-                total = humansize(sv.f_frsize * sv.f_blocks, True)
-                
-                srv_info.append(free + " free")
-                srv_info.append(total)
+                if WINDOWS:
+                    bfree = ctypes.c_ulonglong(0)
+                    ctypes.windll.kernel32.GetDiskFreeSpaceExW(
+                        ctypes.c_wchar_p(abspath), None, None, ctypes.pointer(bfree)
+                    )
+                    srv_info.append(humansize(bfree.value) + " free")
+                else:
+                    sv = os.statvfs(abspath)
+                    free = humansize(sv.f_frsize * sv.f_bfree, True)
+                    total = humansize(sv.f_frsize * sv.f_blocks, True)
+
+                    srv_info.append(free + " free")
+                    srv_info.append(total)
         except:
             pass
 
@@ -1112,7 +1120,7 @@ def tx_browser(self):
             prologue=logues[0],
             epilogue=logues[1],
             title=html_escape(self.vpath, quote=False),
-            srv_info='</span> /// <span>'.join(srv_info)
+            srv_info="</span> /// <span>".join(srv_info),
         )
         self.reply(html.encode("utf-8", "replace"))
         return True