1
1
mirror of https://github.com/plausible/analytics.git synced 2024-12-19 07:31:50 +03:00
analytics/lib/mix/tasks/download_country_database.ex
Uku Taht 83a759c60a
Use db-ip lite database for countries ()
* Use DBIP database for countries

* Add fake data for geolocation tests
2021-04-12 12:21:07 +03:00

39 lines
1.2 KiB
Elixir

defmodule Mix.Tasks.DownloadCountryDatabase do
use Mix.Task
use Plausible.Repo
require Logger
# coveralls-ignore-start
def run(_) do
Application.ensure_all_started(:httpoison)
Application.ensure_all_started(:timex)
this_month = Timex.today()
last_month = Timex.shift(this_month, months: -1)
this_month = this_month |> Date.to_iso8601() |> binary_part(0, 7)
last_month = last_month |> Date.to_iso8601() |> binary_part(0, 7)
this_month_url = "https://download.db-ip.com/free/dbip-country-lite-#{this_month}.mmdb.gz"
last_month_url = "https://download.db-ip.com/free/dbip-country-lite-#{last_month}.mmdb.gz"
Logger.info("Downloading #{this_month_url}")
res = HTTPoison.get!(this_month_url)
res =
case res.status_code do
404 ->
Logger.info("Got 404 for #{this_month_url}, trying #{last_month_url}")
HTTPoison.get!(last_month_url)
_ ->
res
end
if res.status_code == 200 do
File.mkdir("priv/geodb")
File.write!("priv/geodb/dbip-country.mmdb", res.body)
Logger.info("Downloaded and saved the database successfully")
else
Logger.error("Unable to download and save the database. Response: #{inspect(res)}")
end
end
end