Cache UnicodeData.txt to priv

noblesamurai · Jul 12, 2011 · c162fb1 · c162fb1
1 parent c27435a
commit c162fb1
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 1 deletion.
diff --git a/README.txt b/README.txt
@@ -1,5 +1,9 @@
 A pure Erlang IDNA implementation.
 
+An attempt will be made to read from priv/UnicodeData.txt. If the file does not exist,
+it is downloaded from unicode.org, and an attempt is made to save it to the above filename,
+though no checking is done as to whether or not this succeeded.
+
 Quick start:
 
   $ make

diff --git a/priv/.placeholder b/priv/.placeholder
diff --git a/src/idna.erl b/src/idna.erl
@@ -10,7 +10,14 @@
 
 start() ->
   idna_unicode_data:start(),
-  idna_unicode_data:load("http://www.unicode.org/Public/UNIDATA/UnicodeData.txt").
+  File = code:priv_dir('erlang') ++ "/UnicodeData.txt",
+  case file:read_file(File) of
+    {'ok', Data} -> idna_unicode_data:load(Data);
+	{'error', _} ->
+      {ok, {{_, 200, _}, _, GetData}} = http:request(get, {"http://www.unicode.org/Public/UNIDATA/UnicodeData.txt", []}, [], [{body_format, binary}]),
+	  file:write_file(File, GetData),
+	  idna_unicode_data:load(GetData)
+  end.
 
 to_ascii(Domain) ->
   to_ascii(string:tokens(idna_unicode:downcase(Domain), "."), []).