apps/efka/src/efka_downloader.erl

@@ -12,14 +12,15 @@
 -behaviour(gen_server).
 
 %% API
--export([start_link/0, download/3]).
+-export([start_link/2, download/1]).
 -export([test/0]).
 
 %% gen_server callbacks
 -export([init/1, handle_call/3, handle_cast/2, handle_info/2, terminate/2, code_change/3]).
 
 -record(state, {
-
+    url :: string(),
+    target_file :: string()
 }).
 
 %%%===================================================================
@@ -27,28 +28,26 @@
 %%%===================================================================
 
 test() ->
-    {ok, Pid} = start_link(),
+    {ok, Pid} = start_link("https://codeload.github.com/genadyo/LivePhotoDemo/zip/refs/heads/master", "/tmp/test.tar"),
-    Url = "https://codeload.github.com/genadyo/LivePhotoDemo/zip/refs/heads/master",
+    Ref = download(Pid),
-    TargetDir = "/tmp/",
-    Ref = download(Pid, Url, TargetDir),
     receive
         {download_response, Ref, Info} ->
             lager:debug("info is: ~p", [Info])
     end,
     ok.
 
--spec download(Pid :: pid(), Url :: string(), TargetDir :: string()) -> reference().
+-spec download(Pid :: pid()) -> reference().
-download(Pid, Url, TargetDir) when is_pid(Pid), is_list(Url), is_list(TargetDir) ->
+download(Pid) when is_pid(Pid) ->
     Ref = make_ref(),
     ReceiverPid = self(),
-    gen_server:cast(Pid, {download, ReceiverPid, Ref, Url, TargetDir}),
+    gen_server:cast(Pid, {download, ReceiverPid, Ref}),
     Ref.
 
 %% @doc Spawns the server and registers the local name (unique)
--spec(start_link() ->
+-spec(start_link(Url :: string(), TargetFile :: string()) ->
     {ok, Pid :: pid()} | ignore | {error, Reason :: term()}).
-start_link() ->
+start_link(Url, TargetFile) when is_list(Url), is_list(TargetFile) ->
-    gen_server:start_link(?MODULE, [], []).
+    gen_server:start_link(?MODULE, [Url, TargetFile], []).
 
 %%%===================================================================
 %%% gen_server callbacks
@@ -59,8 +58,8 @@ start_link() ->
 -spec(init(Args :: term()) ->
     {ok, State :: #state{}} | {ok, State :: #state{}, timeout() | hibernate} |
     {stop, Reason :: term()} | ignore).
-init([]) ->
+init([Url, TargetFile]) ->
-    {ok, #state{}}.
+    {ok, #state{url = Url, target_file = TargetFile}}.
 
 %% @private
 %% @doc Handling call messages
@@ -81,16 +80,11 @@ handle_call(_Request, _From, State = #state{}) ->
     {noreply, NewState :: #state{}} |
     {noreply, NewState :: #state{}, timeout() | hibernate} |
     {stop, Reason :: term(), NewState :: #state{}}).
-handle_cast({download, ReceiverPid, Ref, Url, TargetDir}, State = #state{}) ->
+handle_cast({download, ReceiverPid, Ref}, State = #state{url = Url, target_file = TargetFile}) ->
-    SSLOpts = {ssl_options, [
+    case hackney:get(Url, [], <<>>, [async]) of
-        % 完全禁用证书验证
-        {verify, verify_none}
-    ]},
-
-    TargetFile = get_filename_from_url(Url),
-    case hackney:request(get, Url, [], <<>>, [async, {stream_to, self()}, {pool, false}, SSLOpts]) of
         {ok, ClientRef} ->
-            case receive_data(ClientRef, TargetDir, TargetFile) of
+            {ok, File} = file:open(TargetFile, [write, binary]),
+            case receive_data(ClientRef, File) of
                 ok ->
                     ReceiverPid ! {download_response, Ref, ok};
                 {error, Reason} ->
@@ -132,56 +126,23 @@ code_change(_OldVsn, State = #state{}, _Extra) ->
 %%% Internal functions
 %%%===================================================================
 
-%% 读取请求 ResponseLine
+receive_data(ClientRef, File) ->
-receive_data(ClientRef, TargetDir, DefaultFile) when is_list(TargetDir), is_list(DefaultFile) ->
     receive
+        {hackney_response, ClientRef, {headers, _Headers}} ->
+            receive_data(ClientRef, File);
         {hackney_response, ClientRef, {status, 200, _Reason}} ->
-            receive_data0(ClientRef, TargetDir, DefaultFile);
+            receive_data(ClientRef, File);
         {hackney_response, ClientRef, {status, StatusCode, _Reason}} ->
-            {error, {http_status, StatusCode}}
+            lager:debug("call me here"),
-    end.
+            {error, {http_status, StatusCode}};
-%% 处理头部信息, 解析可能的文件名
+        {hackney_response, ClientRef, Data} ->
-receive_data0(ClientRef, TargetDir, DefaultFile) ->
+            file:write(File, Data),
-    receive
+            receive_data(ClientRef, File);
-        {hackney_response, ClientRef, {headers, Headers}} ->
-            TargetFilename = extra_filename(Headers, DefaultFile),
-            FullFilename = TargetDir ++ TargetFilename,
-            {ok, File} = file:open(FullFilename, [write, binary]),
-            receive_data1(ClientRef, File)
-    end.
-%% 接受文件数据
-receive_data1(ClientRef, File) ->
-    receive
-        {hackney_response, ClientRef, {error, Reason}} ->
-            ok = file:close(File),
-            {error, Reason};
         {hackney_response, ClientRef, done} ->
             ok = file:close(File),
             hackney:close(ClientRef),
             ok;
-        {hackney_response, ClientRef, Data} ->
+        {hackney_response, ClientRef, {error, Reason}} ->
-            file:write(File, Data),
+            ok = file:close(File),
-            receive_data1(ClientRef, File)
+            {error, Reason}
     end.
-
--spec extra_filename(Headers :: list(), Default :: string()) -> Filename :: string().
-extra_filename(Headers, Default) when is_list(Headers), is_list(Default) ->
-    case lists:filter(fun({K, _}) -> string:lowercase(K) =:= <<"content-disposition">> end, Headers) of
-        [{_, <<"attachment; ", Rest/binary>>}|_] ->
-            Params0 = binary:split(Rest, <<";">>, [global]),
-            Params = lists:map(fun(P) -> list_to_tuple(binary:split(P, <<"=">>)) end, Params0),
-            case proplists:get_value(<<"filename">>, Params) of
-                undefined ->
-                    Default;
-                Filename ->
-                    binary_to_list(Filename)
-            end;
-        _ ->
-            Default
-    end.
-
--spec get_filename_from_url(Url :: string()) -> string().
-get_filename_from_url(Url) when is_list(Url) ->
-    URIMap = uri_string:parse(Url),
-    Path = maps:get(path, URIMap),
-    filename:basename(Path).