actions.go

package main

import (
	"log"
	"os"
	"runtime"

	"github.com/carlescere/scheduler"
	"github.com/maddevsio/spiderwoman/lib"
	"github.com/urfave/cli"
)

var path = Path{sqliteDBPath, lib.SourcesFilePath, lib.SourcesDefaultFilePath, lib.TypesFilePath, lib.TypesHDefaultFilePath}

func actionOnce(c *cli.Context) error {
	initialize(path)
	crawl(path)
	grab(path)
	return nil
}

func actionForever(c *cli.Context) error {
	job := func() {
		crawl(path)
		grab(path)
	}
	initialize(path)
	log.Print("All is OK. Starting cron job...")
	if config.GetString("box") == "dev" {
		log.Print("This is a dev box")
		scheduler.Every(1).Minutes().Run(job)
		//gocron.Every(1).Minute().Do(crawl) // this is for testing on dev box
	} else {
		log.Print("This is production")
		if config.GetString("start-time") == "" {
			log.Fatal("You need to set start-time value in config.yaml")
		}
		scheduler.Every().Day().At(config.GetString("start-time")).Run(job)
		//gocron.Every(1).Day().At().Do(crawl, path)
	}
	//<- gocron.Start()
	runtime.Goexit()
	return nil
}

func actionExcel(c *cli.Context) error {
	err := os.MkdirAll(config.GetString("xls-dir"), 0777)
	if err != nil {
		log.Fatalf("cannot create dir for excel files: %v", err)
	}
	if c.Args().Get(0) != "noinit" {
		initialize(path)
	}
	createAllXLSByDays(path.SqliteDBPath)
	return nil
}

func actionGrab(c *cli.Context) error {
	initialize(path)
	grab(path)
	return nil
}