Add functionality to accept setting status non-interactively

main.go

@@ -100,11 +100,18 @@ func SetStatus(args []string) error {
 	// User status is written to ~/.checkin
 	outputPath := path.Join(curUser.HomeDir, ".checkin")
 
+	var input string
+
+	// Check if input was provided on the command line.
+	if len(setFlags.Args()) > 0 {
+		input = ArgsToStatus(setFlags.Args(), curUser)
+	} else {
 		// Prompt user for input
-	input, err := PromptInput()
+		input, err = PromptInput()
 		if err != nil {
 			return err
 		}
+	}
 
 	// Remove file on blank input
 	if input == "" {
@@ -137,6 +144,29 @@ func SetStatus(args []string) error {
 	return nil
 }
 
+// ArgsToStatus concatenates all arguments left over from a call to flag.Parse
+// and returns it as a well-formed status.
+func ArgsToStatus(args []string, curUser *user.User) string {
+	status := strings.Join(args, " ")
+	status = strings.TrimRight(status, " \n\t\r")
+	if status == "" {
+		return status
+	}
+
+	if strings.HasPrefix(status, "~"+curUser.Username) {
+		return status
+	}
+
+	if strings.HasPrefix(status, curUser.Username) {
+		return "~" + status
+	}
+
+	return fmt.Sprintf("~%s: %s", curUser.Username, status)
+}
+
+// GetFriendlyWd returns the friendliest equivalent to the working directory.
+// If we're in the user's home directory, we return a path relative to ~user.
+// If we're in that user's public_html directory, we return an http link to it.
 func GetFriendlyWd(curUser *user.User) (string, error) {
 	wd, err := os.Getwd()
 	if err != nil {