make hosts command run

Signed-off-by: Jason McCallister <jason@craftcms.com>
craftcms · Apr 8, 2020 · 4170422 · 4170422
1 parent 445e77c
commit 4170422
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 8 deletions.
diff --git a/internal/cmd/hosts.go b/internal/cmd/hosts.go
@@ -2,7 +2,6 @@ package cmd
 
 import (
 	"errors"
-	"fmt"
 	"os"
 
 	"github.com/spf13/cobra"
@@ -49,8 +48,6 @@ var hostsCommand = &cobra.Command{
 
 		hosts.AddHosts(ip, domains)
 
-		fmt.Println(hosts.RenderHostsFile())
-
-		return nil
+		return hosts.Save()
 	},
 }
diff --git a/internal/cmd/hosts_remove.go b/internal/cmd/hosts_remove.go
@@ -42,10 +42,8 @@ var hostsRemoveCommand = &cobra.Command{
 			domains = append(domains, site.Domain)
 		}
 
-		hosts.RemoveAddresses(domains)
+		hosts.Remove(domains)
 
-		// fmt.Println(hosts.RenderHostsFile())
-
-		return nil
+		return hosts.Save()
 	},
 }