refactoring.

client.erl

@@ -27,18 +27,7 @@ switch_command(Socket, Filename, Offset, Command) ->
       ClientOffset = filelib:file_size(?CLIENT_FOLDER ++ Filename),
       gen_tcp:send(Socket, <<?APPROVEMENT:8/integer, ClientOffset:32/integer>>),
       {ok, IoDevice} = file:open(?CLIENT_FOLDER ++ Filename, [append]),
-      wait_for_file(Socket, IoDevice)
-  end.
-
-wait_for_file(Socket, IoDevice) ->
-  case gen_tcp:recv(Socket, 0) of
-    {ok, <<?SENDING:8/integer, Packet/binary>>} ->
-      file:write(IoDevice, Packet),
-      wait_for_file(Socket, IoDevice);
-    {ok, <<?DOWNLOADED:8/integer>>} ->
-      io:format("File was downloaded.~n");
-    {ok, <<?FILE_NOT_EXIST:8/integer>>} ->
-      io:format("File does not exist.~n")
+      utils:wait_for_file(Socket, IoDevice)
   end.
 
 wait_for_command(Socket) ->

server.erl

@@ -27,7 +27,7 @@ handle(Socket) ->
       [Filename, Offset] = utils:get_file_info(Packet),
       gen_tcp:send(Socket, [<<Offset:32/integer, ?UPLOAD:8/integer>>, Filename]),
       {ok, IoDevice} = file:open(?SERVER_FOLDER ++ Filename, [append]),
-      wait_for_file(Socket, IoDevice),
+      utils:wait_for_file(Socket, IoDevice),
       handle(Socket);
     {ok, <<"EXIT\n">>} ->
       gen_tcp:close(Socket);
@@ -45,14 +45,3 @@ wait_for_approvement(Socket, Filename) ->
     {error, timeout} ->
       ok
   end.
-
-wait_for_file(Socket, IoDevice) ->
-  case gen_tcp:recv(Socket, 0) of
-    {ok, <<?SENDING:8/integer, Packet/binary>>} ->
-      file:write(IoDevice, Packet),
-      wait_for_file(Socket, IoDevice);
-    {ok, <<?FILE_NOT_EXIST:8/integer>>} ->
-      io:format("File not found.~n");
-    {ok, <<?DOWNLOADED:8/integer>>} ->
-      io:format("File received.~n")
-  end.

utils.erl

@@ -1,6 +1,6 @@
 -module(utils).
 -include("config.hrl").
--export([delete_n/1, get_file_info/1, send_file/4, send_file_binary/3]).
+-export([delete_n/1, get_file_info/1, send_file/4, send_file_binary/3, wait_for_file/2]).
 
 delete_n(Str) ->
   string:substr(Str, 1, string:len(Str) - 1).
@@ -28,3 +28,14 @@ send_file_binary(Socket, IoDevice, Offset) ->
       io:format("File was sent.~n"),
       gen_tcp:send(Socket, <<?DOWNLOADED:8/integer>>)
   end.
+
+wait_for_file(Socket, IoDevice) ->
+  case gen_tcp:recv(Socket, 0) of
+    {ok, <<?SENDING:8/integer, Packet/binary>>} ->
+      file:write(IoDevice, Packet),
+      wait_for_file(Socket, IoDevice);
+    {ok, <<?DOWNLOADED:8/integer>>} ->
+      io:format("File was downloaded.~n");
+    {ok, <<?FILE_NOT_EXIST:8/integer>>} ->
+      io:format("File does not exist.~n")
+  end.