main.go

package main

import (
	"encoding/json"
	"io/ioutil"
	"log"
	"net/http"
	"os"

	"github.com/18F/e-QIP-prototype/api/cmd"
)

func main() {
	var payloads []json.RawMessage
	var buffer []byte
	var err error

	// Figure out if the input is coming from an argument or a pipe.
	if cmd.IsPiped() {
		// Read the contents from standard input
		buffer, err = ioutil.ReadAll(os.Stdin)
		if err != nil {
			log.Println("error reading from standard input:", err)
			return
		}

		// Marshal standard input contents in to JSON
		err = json.Unmarshal(buffer, &payloads)
		if err != nil {
			log.Println("error marshalling from standard input:", err)
			return
		}
	} else {
		for _, file := range os.Args[1:] {
			buffer, err = ioutil.ReadFile(file)
			if err != nil {
				log.Println("error reading from file:", err)
				return
			}
			payloads = append(payloads, buffer)
		}
	}

	// Create a web client to interface with the RESTful API.
	webclient := &cmd.WebClient{
		Client: &http.Client{},
	}

	// Loop through all sections received
	for _, payload := range payloads {
		// POST a save request for section
		webclient.Save(payload)
	}
	webclient.Logout()
}