place.go

package main

import (
	"bytes"
	"context"
	"fmt"
	"io"
	"os"
	"path/filepath"

	"github.com/aabizri/transit/maps"

	"github.com/aabizri/navitia"
	"github.com/aabizri/navitia/pretty"
	"github.com/fatih/color"
	"github.com/fogleman/gg"
	"github.com/pkg/errors"
	"github.com/urfave/cli"
)

var placeFlags = []cli.Flag{
	cli.UintFlag{
		Name:  "count, c",
		Value: 6,
	},
}

var placeCommand = cli.Command{
	Name:    "place",
	Aliases: []string{"places,p"},
	Usage:   "Search for places",
	Action:  placeAction,
	Flags:   placeFlags,
}

func placeAction(c *cli.Context) error {
	gctx := context.Background()
	for i, query := range c.Args() {
		ctx, cancel := context.WithTimeout(gctx, requestTimeout)
		defer cancel()

		// Create a request
		req := navitia.PlacesRequest{Query: query, Count: c.Uint("count")}

		// Execute it
		pr, err := session.Places(ctx, req)
		if err != nil {
			return errors.Wrap(err, "Error while requesting places")
		}

		// Now let's print
		msg := fmt.Sprintf("\n[%d/%d] Query \"%s\" ", i+1, len(c.Args()), color.New(color.Underline, color.FgHiCyan).Sprint(query))
		buf := bytes.NewBuffer([]byte(msg))
		err = pretty.DefaultPlacesResultsConf.PrettyWrite(pr, buf)
		if err != nil {
			return errors.Wrapf(err, "Error while preparing result output for query #%d", i)
		}

		// And copy
		_, err = io.Copy(os.Stdout, buf)
		if err != nil {
			return errors.Wrapf(err, "error while copying buffer to stdout")
		}

		// If enabled, draw !
		if path := c.GlobalString("map"); path != "" {
			if !filepath.IsAbs(path) {
				wd, err := os.Getwd()
				if err != nil {
					return errors.Wrap(err, "couldn't retrieve working directory, consider giving an absolute path")
				}

				path = filepath.Join(wd, path)
			}

			img, err := maps.DrawPlaces(pr.Places)
			if err != nil {
				return errors.Wrap(err, "error while drawing places")
			}

			err = gg.SavePNG(path, img)
			if err != nil {
				return errors.Wrapf(err, "error while saving image to %s", path)
			}
		}
	}

	return nil
}