From 848742974fe409020b5a91a564e46de0ea8a2454 Mon Sep 17 00:00:00 2001
From: chengchingwen <adgjl5645@hotmail.com>
Date: Sun, 20 Feb 2022 11:31:53 +0800
Subject: [PATCH] Fix google drive download: remove gcode

---
 src/gdrive.jl | 20 ++++++--------------
 1 file changed, 6 insertions(+), 14 deletions(-)

diff --git a/src/gdrive.jl b/src/gdrive.jl
index 12e58e9..f7a5c51 100644
--- a/src/gdrive.jl
+++ b/src/gdrive.jl
@@ -43,14 +43,6 @@ function gfile_handler(url)
     return "https://docs.google.com/uc?export=download&id=$(h.captures[])"
 end
 
-function find_gcode(cookies)
-    for cookie in cookies
-        (match(r"download_warning_", cookie.name) !== nothing) && (return cookie.value)
-    end
-
-    return
-end
-
 function find_filename(header)
     m = match(r"filename=\\\"(.*)\\\"", header)
     if isnothing(m)
@@ -64,16 +56,16 @@ function find_filename(header)
 end
 
 function download_gdrive(url, localdir)
-    cookiejars = Dict{String, Set{HTTP.Cookies.Cookie}}()
-    HTTP.request("GET", url; cookies=true, cookiejar=cookiejars)
-    gcode = find_gcode(cookiejars["docs.google.com"])
+    m = match(r"confirm=([^;&]+)", url)
+    isnothing(m) && (url = "$url&confirm=pbef")
 
-    !isnothing(gcode) && (url = "$url&confirm=$gcode")
+    r = HTTP.head(url; status_exception = false)
+    HTTP.iserror(r) && r.status != 303 && throw(HTTP.StatusError(r.status, r.request.method, r.request.target, r))
+    r.status == 303 && (url = r["Location"])
 
     filepath = Ref{String}("")
     HTTP.open(
-        "GET", url, ["Range"=>"bytes=0-"],
-        cookies=true, cookiejar=cookiejars, redirect_limit=10
+        "GET", url, ["Range"=>"bytes=0-"], redirect_limit=10
     ) do stream
         response = HTTP.startread(stream)
         eof(stream) && return