-module(rebar_prv_plugins_upgrade). -behaviour(provider). -export([init/1, do/1, format_error/1]). -include("rebar.hrl"). -include_lib("providers/include/providers.hrl"). -define(PROVIDER, upgrade). -define(NAMESPACE, plugins). -define(DEPS, []). -spec init(rebar_state:t()) -> {ok, rebar_state:t()}. init(State) -> State1 = rebar_state:add_provider( State, providers:create([ {name, ?PROVIDER}, {module, ?MODULE}, {namespace, ?NAMESPACE}, {bare, true}, {deps, ?DEPS}, {example, "rebar3 plugins upgrade "}, {short_desc, "Upgrade plugins"}, {desc, "List or upgrade plugins. Use the -a/--all option to upgrade" " all plugins."}, {opts, [{plugin, undefined, undefined, string, "Plugin to upgrade"}, {all, $a, "all", undefined, "Upgrade all plugins."}]}])), {ok, State1}. -spec do(rebar_state:t()) -> {ok, rebar_state:t()} | {error, string()}. do(State) -> case handle_args(State) of {false, undefined} -> throw(?PRV_ERROR(no_arg)); {true, _} -> {_, LocalPluginsNames} = rebar_prv_plugins:list_local_plugins(State), lists:foldl( fun (LocalPluginName, {ok, StateAcc}) -> upgrade(atom_to_list(LocalPluginName), StateAcc) end, {ok, State}, LocalPluginsNames); {false, Plugin} -> upgrade(Plugin, State) end. -spec format_error(any()) -> iolist(). format_error({not_found, Plugin}) -> io_lib:format("Plugin to upgrade not found: ~ts", [Plugin]); format_error(no_arg) -> "Specify a plugin to upgrade, or --all to upgrade them all"; format_error(Reason) -> io_lib:format("~p", [Reason]). handle_args(State) -> {Args, _} = rebar_state:command_parsed_args(State), All = proplists:get_value(all, Args, false), Plugin = proplists:get_value(plugin, Args), {All, Plugin}. upgrade(Plugin, State) -> Profiles = rebar_state:current_profiles(State), case find_plugin(Plugin, Profiles, State) of not_found -> Dep = find_plugin(Plugin, [global], State); Dep -> Dep end, case Dep of not_found -> ?PRV_ERROR({not_found, Plugin}); {ok, P, Profile} -> State1 = rebar_state:set(State, deps_dir, ?DEFAULT_PLUGINS_DIR), maybe_update_pkg(P, State1), {Apps, State2} = rebar_prv_install_deps:handle_deps_as_profile(Profile, State1, [P], true), {no_cycle, Sorted} = rebar_prv_install_deps:find_cycles(Apps), ToBuild = rebar_prv_install_deps:cull_compile(Sorted, []), %% Add already built plugin deps to the code path CodePaths = [rebar_app_info:ebin_dir(A) || A <- Apps -- ToBuild], code:add_pathsa(CodePaths), %% Build plugin and its deps _ = build_plugin(ToBuild, State2), {ok, State} end. find_plugin(Plugin, Profiles, State) -> ec_lists:search(fun(Profile) -> Plugins = rebar_state:get(State, {plugins, Profile}, []) ++ rebar_state:get(State, {project_plugins, Profile}, []), case rebar_utils:tup_find(list_to_atom(Plugin), Plugins) of false -> not_found; P -> {ok, P} end end, Profiles). build_plugin(ToBuild, State) -> Providers = rebar_state:providers(State), rebar_prv_compile:compile(State, Providers, ToBuild, plugins). maybe_update_pkg(Tup, State) when is_tuple(Tup) -> maybe_update_pkg(element(1, Tup), State); maybe_update_pkg(Name, State) -> try rebar_app_utils:parse_dep(Name, root, unicode:characters_to_list(?DEFAULT_PLUGINS_DIR), State, [], 0) of AppInfo -> Source = rebar_app_info:source(AppInfo), case element(1, Source) of pkg -> Resources = rebar_state:resources(State), #{repos := RepoConfigs} = rebar_resource_v2:find_resource_state(pkg, Resources), PkgName = element(2, Source), [update_package(PkgName, RepoConfig, State) || RepoConfig <- RepoConfigs]; _ -> skip end catch throw:?PRV_ERROR({parse_dep, _}) -> skip end. update_package(Name, RepoConfig, State) -> case rebar_packages:update_package(Name, RepoConfig, State) of fail -> ?WARN("Failed to fetch updates for package ~ts from repo ~ts", [Name, maps:get(name, RepoConfig)]); _ -> ok end.