hopefully works?

1 files changed, 16 insertions, 0 deletions

diff --git a/init.go b/init.go
index 4d1c208..1470af5 100644
--- a/init.go
+++ b/init.go
@@ -150,6 +150,22 @@ func (f *Forge) InitPB() {
 	f.rillY = 20
 }
 
+// set the URL for forge otherwise fallback to ENV or to forge.wit.com
+func (f *Forge) SetForgeURL(url string) {
+	if url == "" {
+		url = os.Getenv("FORGE_URL")
+	}
+	if url == "" {
+		url = "https://forge.wit.com/"
+	}
+	f.forgeURL = url
+	log.Info("Forge URL has been set to", f.forgeURL)
+}
+
+func (f *Forge) GetForgeURL() string {
+	return f.forgeURL
+}
+
 func (f *Forge) InitMachine() {
 	if f.Config.Username == "" {
 		usr, _ := user.Current()