cmd/atlas: allow ref env:// to known attributes

ariga · Jan 31, 2024 · 4b91ddf · 4b91ddf
1 parent c43249c
commit 4b91ddf
Showing 1 changed file with 14 additions and 7 deletions.
diff --git a/cmd/atlas/internal/cmdapi/project.go b/cmd/atlas/internal/cmdapi/project.go
@@ -175,13 +175,20 @@ func (e *Env) VarFromURL(s string) (string, error) {
  if u.Host == "" || u.Path != "" || u.RawQuery != "" {
  return "", fmt.Errorf("invalid env:// variable %q", s)
  }
- attr, ok := e.Attr(u.Host)
- if !ok {
- return "", fmt.Errorf("env://%s (attribute) not found in env.%s", s, e.Name)
- }
- sv, err := attr.String()
- if err != nil {
- return "", fmt.Errorf("env://%s: %w", s, err)
+ var sv string
+ switch u.Host {
+ case "url":
+ sv = e.URL
+ case "dev":
+ sv = e.DevURL
+ default:
+ attr, ok := e.Attr(u.Host)
+ if !ok {
+ return "", fmt.Errorf("env://%s (attribute) not found in env.%s", u.Host, e.Name)
+ }
+ if sv, err = attr.String(); err != nil {
+ return "", fmt.Errorf("env://%s: %w", u.Host, err)
+ }
  }
  if strings.HasPrefix(sv, envAttrScheme+"://") {
  return "", fmt.Errorf("env://%s (attribute) cannot reference another env://", s)