瀏覽代碼

Make image upload resilient to crashes

Zed 7 年之前
父節點
當前提交
12cf772616
共有 1 個文件被更改,包括 6 次插入1 次删除
  1. 6 1
      src/nitter.nim

+ 6 - 1
src/nitter.nim

@@ -70,7 +70,12 @@ routes:
       await client.downloadFile($uri, filename)
       client.close()
 
-    sendFile(filename)
+    if not existsFile(filename):
+      resp Http404
+
+    let file = openAsync(filename)
+    defer: file.close()
+    resp await readAll(file), mimetype(filename)
 
   get "/video/@sig/@url":
     cond "http" in @"url"