goris.go

// Package main (goris.go) :
// This file is included all commands and options.
package main

import (
	"fmt"
	"os"

	"goris/ris"

	"github.com/urfave/cli"
)

const (
	appname = "goris"
)

// dispres : Display results
func dispres(r []string, c int) {
	if len(r) < c {
		c = len(r)
	}
	for i := 0; i < c; i++ {
		fmt.Printf("%s\n", r[i])
	}
}

// handler : Handler of goris
func handler(c *cli.Context) error {
	n := c.Int("number")
	if n > 100 {
		n = 100
	}
	if len(c.String("fromurl")) == 0 && len(c.String("fromfile")) == 0 {
		return fmt.Errorf("no parameters. You can see help by '$ %s -h'", appname)
	}
	var results []string
	var err error
	if len(c.String("fromurl")) > 0 && len(c.String("fromfile")) == 0 {
		results, err = ris.DefImg(c.Bool("webpages")).ImgFromURL(c.String("fromurl"))
		if err != nil {
			return err
		}
	}
	if len(c.String("fromurl")) == 0 && len(c.String("fromfile")) > 0 {
		fmt.Println("# Specification at Google side has been changed. By this, this command cannot be still modifying. So, in the current stage, this command cannot be used. I apologize for this situation.")
		fmt.Println("# Please use goris s -u URL")
		os.Exit(0)

		// results, err = ris.DefImg(c.Bool("webpages")).ImgFromFile(c.String("fromfile"))
		// if err != nil {
		// 	return err
		// }
	}
	if c.Bool("download") && (len(c.String("fromurl")) > 0 || len(c.String("fromfile")) > 0) && !c.Bool("webpages") {
		err := ris.Download(results, n)
		if err != nil {
			return err
		}
	}
	dispres(results, n)
	return nil
}

// createHelp : Create help document.
func createHelp() *cli.App {
	a := cli.NewApp()
	a.Name = appname
	a.Authors = []cli.Author{
		{Name: "tanaike [ https://github.com/tanaikech/goris ] ", Email: "tanaike@hotmail.com"},
	}
	a.Usage = "Search for images with Google Reverse Image Search."
	a.Version = "3.0.4"

	a.Commands = []cli.Command{
		{
			Name:        "search",
			Aliases:     []string{"s"},
			Usage:       "[ " + appname + " s -u URL ] or [ " + appname + " s -f file ]",
			Description: "Do search images.",
			Action:      handler,
			Flags: []cli.Flag{
				&cli.StringFlag{
					Name:  "fromurl, u",
					Usage: "Reverse Image Search from an URL.",
				},
				&cli.StringFlag{
					Name:  "fromfile, f",
					Usage: "Reverse Image Search from an image file.",
				},
				&cli.IntFlag{
					Name:  "number, n",
					Usage: "Number of retrieved image URLs. ( 1 - 100 )",
					Value: 10,
				},
				&cli.BoolFlag{
					Name:  "download, d",
					Usage: "Download images from retrieved URLs.",
				},
				&cli.BoolFlag{
					Name:  "webpages, w",
					Usage: "This is boolean. Retrieve web pages with matching images on Google top page. When this is not used, images are retrieved.",
				},
			},
		},
	}
	return a
}

// main : Main of goris
func main() {
	a := createHelp()
	err := a.Run(os.Args)
	if err != nil {
		fmt.Fprintln(os.Stderr, err)
		os.Exit(1)
	}
}