From 95847ea10b20042aabdf873732a5d4e61f4e7be9 Mon Sep 17 00:00:00 2001
From: Zaiming Shi <zmstone@gmail.com>
Date: Tue, 9 Mar 2021 16:35:31 +0100
Subject: [PATCH] chore(build): Support vars file for enterprise

---
 rebar.config.erl | 20 +++++++++-----------
 1 file changed, 9 insertions(+), 11 deletions(-)

diff --git a/rebar.config.erl b/rebar.config.erl
index b73e9522d..3e3173c99 100644
--- a/rebar.config.erl
+++ b/rebar.config.erl
@@ -127,6 +127,7 @@ profiles() ->
 %% RelType: cloud (full size) | edge (slim size)
 %% PkgType: bin | pkg
 relx(Vsn, RelType, PkgType) ->
+    IsEnterprise = is_enterprise(),
     [ {include_src,false}
     , {include_erts, true}
     , {extended_start_script,false}
@@ -136,22 +137,19 @@ relx(Vsn, RelType, PkgType) ->
     , {release, {emqx, Vsn}, relx_apps(RelType)}
     , {overlay, relx_overlay(RelType)}
     , {overlay_vars, [ {built_on_arch, rebar_utils:get_arch()}
-                     , {emqx_description, emqx_description(RelType)}
-                     , overlay_vars_rel(RelType)
+                     , {emqx_description, emqx_description(RelType, IsEnterprise)}
+                     , overlay_vars_rel(RelType, IsEnterprise)
                      , overlay_vars_pkg(PkgType)
                      ]}
     ].
 
-emqx_description(cloud) ->
-    case is_enterprise() of
-        true -> "EMQ X Enterprise";
-        false -> "EMQ X Broker"
-    end;
-emqx_description(edge) ->
-    "EMQ X Edge".
+emqx_description(cloud, true) -> "EMQ X Enterprise";
+emqx_description(cloud, false) -> "EMQ X Broker";
+emqx_description(edge, _) -> "EMQ X Edge".
 
-overlay_vars_rel(cloud) -> "vars/vars-cloud.config";
-overlay_vars_rel(edge) -> "vars/vars-edge.config".
+overlay_vars_rel(cloud, true) -> "vars/vars-enterprise.config";
+overlay_vars_rel(cloud, false) -> "vars/vars-cloud.config";
+overlay_vars_rel(edge, _) -> "vars/vars-edge.config".
 
 overlay_vars_pkg(bin) -> "vars/vars-bin.config";
 overlay_vars_pkg(pkg) -> "vars/vars-pkg.config".