words.go

/*
   Copyright 2018 Assetnote

   Licensed under the Apache License, Version 2.0 (the "License");
   you may not use this file except in compliance with the License.
   You may obtain a copy of the License at

       http://www.apache.org/licenses/LICENSE-2.0

   Unless required by applicable law or agreed to in writing, software
   distributed under the License is distributed on an "AS IS" BASIS,
   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   See the License for the specific language governing permissions and
   limitations under the License.
*/

package wordswithext

import (
	"github.com/urfave/cli"
	"github.com/assetnote/commonspeak2/assets"
	"github.com/assetnote/commonspeak2/log"
	"github.com/valyala/fasttemplate"
	"golang.org/x/net/context"
	"google.golang.org/api/option"
	"cloud.google.com/go/bigquery"
	"os"
)

func CmdStatus(c *cli.Context) error {
	verboseOpt := c.GlobalBool("verbose")
	silentOpt := c.GlobalBool("silent")
	project := c.GlobalString("project")
	credentials := c.GlobalString("credentials")
	extensions := c.String("extensions")
	limitArg := c.String("limit")
	outputFile := c.String("output")

	if _, err := os.Stat(credentials); os.IsNotExist(err) {
		log.Fatalln(`credentials file not found, either have credentials.json saved in this folder, or provide the
						the credentials file path via the --credentials parameter.`)
	}

	if credentials == "" {
		log.Fatalln("credentials are required, provide it using the --credentials parameter.")
	}

	if project == "" {
		log.Fatalln("project is required, provide it using the --project parameter.")
	}

	if extensions == "" {
		log.Fatalln("extensions are required, provide it using the -e parameter.")
	}

	extensionsRegex := convertExtensionsToRegex(extensions)
	
	wordsSqlAsset, err := assets.Asset("data/sql/github/words-with-ext.sql")
	if err != nil {
		// Asset was not found.
	}
	wordsTemplate := string(wordsSqlAsset[:])
	template := fasttemplate.New(wordsTemplate, "{{", "}}")
	compiledSql := template.ExecuteString(map[string]interface{}{
		"extensions":  extensionsRegex,
		"limit": 	   limitArg,
	})
	
	fields := log.Fields{
		"Mode":       "WordsWithExt",
		"Source":     "Github",
		"Limit":      limitArg,
		"Extensions": extensions,
	}
	
	if verboseOpt {
		log.WithFields(fields).Infof("Compiled SQL Template: %s", compiledSql)
	}

	ctx := context.Background()

	client, err := bigquery.NewClient(ctx, project, option.WithCredentialsFile(credentials))
	if err != nil {
		log.WithFields(fields).Warning("Failed to create a BigQuery client. Please check your credentials.")
	}

	log.WithFields(fields).Info("Executing BigQuery SQL... this could take some time.")
	rows, err := query(client, ctx, compiledSql)
	
	if err != nil {
		fields["Error"] = err.Error()
		log.WithFields(fields).Fatal("Error executing BigQuery SQL.")
	}
	
	resultsErr := handleResults(os.Stdout, rows, outputFile, silentOpt, verboseOpt)
	
	if resultsErr != nil {
		fields["Error"] = err.Error()
		log.WithFields(fields).Fatal("Error handling results.")
	}

	return nil
}