Add error handling for arguments.

kentaro-m · May 5, 2017 · 0f57843 · 0f57843
1 parent 0eced20
commit 0f57843
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/cmd/copy.go b/cmd/copy.go
@@ -1,6 +1,8 @@
 package cmd
 
 import (
+	"errors"
+
 	"github.com/spf13/cobra"
 	"github.com/kentaro-m/md2confl/utils/util"
 	"github.com/kentaro-m/md2confl/utils/file"
@@ -15,6 +17,11 @@ var copyCmd = &cobra.Command{
 }
 
 func copy(cmd *cobra.Command, args []string) error {
+
+	if len(args) > 1 {
+		return errors.New("Too many arguments.")
+	}
+
 	file := file.File{}
 	if err := file.Open(args[0]); err != nil {
 		return err

diff --git a/cmd/preview.go b/cmd/preview.go
@@ -1,6 +1,8 @@
 package cmd
 
 import (
+	"errors"
+
 	"github.com/spf13/cobra"
 	"github.com/kentaro-m/md2confl/confluence"
 	"github.com/kentaro-m/md2confl/utils/file"
@@ -14,6 +16,11 @@ var previewCmd = &cobra.Command{
 }
 
 func preview(cmd *cobra.Command, args []string) error {
+
+	if len(args) > 1 {
+		return errors.New("Too many arguments.")
+	}
+
 	file := file.File{}
 	if err := file.Open(args[0]); err != nil {
 		return err