add rebar update provider

1 files changed, 56 insertions, 0 deletions

diff --git a/src/rebar_prv_update.erl b/src/rebar_prv_update.erl
new file mode 100644
index 0000000..ef6270a
--- /dev/null
+++ b/src/rebar_prv_update.erl
@@ -0,0 +1,56 @@
+%% -*- erlang-indent-level: 4;indent-tabs-mode: nil -*-
+%% ex: ts=4 sw=4 et
+
+-module(rebar_prv_update).
+
+-behaviour(rebar_provider).
+
+-export([init/1,
+         do/1]).
+
+-include("rebar.hrl").
+
+-define(PROVIDER, update).
+-define(DEPS, []).
+
+%% ===================================================================
+%% Public API
+%% ===================================================================
+
+-spec init(rebar_config:config()) -> {ok, rebar_config:config()}.
+init(State) ->
+    State1 = rebar_config:add_provider(State, #provider{name = ?PROVIDER,
+                                                        provider_impl = ?MODULE,
+                                                        bare = false,
+                                                        deps = ?DEPS,
+                                                        example = "rebar update cowboy",
+                                                        short_desc = "",
+                                                        desc = "",
+                                                        opts = []}),
+    {ok, State1}.
+
+-spec do(rebar_config:config()) -> {ok, rebar_config:config()} | relx:error().
+do(Config) ->
+    [Name] = rebar_config:command_args(Config),
+    ?INFO("Updating ~s~n", [Name]),
+
+    DepsDir = rebar_deps:get_deps_dir(Config),
+    Deps = rebar_config:get_local(Config, deps, []),
+    {_, _, Source} = lists:keyfind(list_to_atom(Name), 1, Deps),
+    TargetDir = rebar_deps:get_deps_dir(DepsDir, Name),
+    rebar_fetch:update_source1(TargetDir, Source),
+
+    [App] = rebar_app_discover:find_apps([TargetDir]),
+
+    {ok, AppInfo1} = rebar_otp_app:compile(Config, App),
+    Config1 = rebar_config:replace_app(Config, rebar_app_info:name(AppInfo1), AppInfo1),
+    rebar_erlc_compiler:compile(Config, rebar_app_info:dir(AppInfo1)),
+    update_lock_file(Config, AppInfo1, Source),
+
+    {ok, Config}.
+
+update_lock_file(Config, App, Source) ->
+    New = rebar_fetch:new(rebar_app_info:dir(App), rebar_app_info:name(App), rebar_app_info:original_vsn(App), Source),
+    {ok, [Terms]} = file:consult("./rebar.lock"),
+    LockDeps = lists:keyreplace(rebar_app_info:name(App), 1, Terms, New),
+    ok = file:write_file("./rebar.lock", io_lib:format("~p.~n", [LockDeps])).