ace-teknologi/abra

package cmd

import (
    "encoding/json"
    "encoding/xml"
    "fmt"
    "os"

    abra "github.com/ace-teknologi/abra/abra-lib"
    "github.com/spf13/cobra"
)

const (
    searchByNameStringFlag = "search"
)

var searchByNameString string

var searchByNameCmd = &cobra.Command{
    Use:   "search",
    Short: "Searches the ABR",
    Long:  `Searches the ABR`,
    RunE: func(cmd *cobra.Command, args []string) error {
        return searchByName()
    },
}

func init() {
    rootCmd.AddCommand(searchByNameCmd)
    searchByNameCmd.Flags().StringVarP(&searchByNameString, searchByNameStringFlag, "s", "", "The name of the company to search for")
    searchByNameCmd.Flags().StringVarP(&outputFormat, outputFormatFlag, "f", "", "Output format: json, xml, text")
    searchByNameCmd.Flags().StringVarP(&outputFormatTextTemplatePath, outputFormatTextTemplatePathFlag, "t", "", "Path to text output template")
}

func searchByName() error {
    // Ensure we have a GUID
    err := setGUID()
    if err != nil {
        return err
    }

    client, err := abra.NewWithGuid(GUID)
    if err != nil {
        return err
    }

    searchResults, err := client.SearchByName(searchByNameString, nil)
    if err != nil {
        return err
    }

    outputFormat, err := setOutputType(outputFormat)
    if err != nil {
        return err
    }

    switch outputFormat {
    case outputTypeTEXT:
        fmt.Printf("Found %d Business Entities:\n", searchResults.NumberOfRecords)

        t, err := setOutputTypeTextTemplate("search", outputFormatTextTemplatePath)
        if err != nil {
            return err
        }

        for i, entity := range searchResults.SearchResultsRecord {
            fmt.Printf("%d.\n", i+1)
            err = t.Execute(os.Stdout, entity)
            if err != nil {
                return err
            }
        }
    case outputTypeJSON:
        b, err := json.Marshal(searchResults)
        if err != nil {
            return err
        }
        fmt.Println(string(b))
    case outputTypeXML:
        b, err := xml.Marshal(searchResults)
        if err != nil {
            return err
        }
        fmt.Println(string(b))
    default:
        return ErrInvalidOutputTypeMessage
    }

    return nil
}