main.go

package main

import (
	"fmt"
	"io"
	"log"
	"os"

	"github.com/gin-gonic/gin"
	"github.com/urfave/cli"
)

func setupRoutes(router *gin.Engine) {
	router.GET("/ping", func(c *gin.Context) {
		c.JSON(200, gin.H{
			"message": "pong",
		})
	})
	router.POST("/login", LoginHandler)
	router.POST("/adddl", AddDownloadHandler)
	router.POST("/canceldl", CancelDownloadHandler)
	router.GET("/dlinfo/:gid", GetDownloadInfoHandler)
}

func setupLoggingToFile(logfile string) {
	log.Println("Setting logToFile: ", logfile)
	os.Remove(logfile)
	handle, err := GetLogFileHandle(logfile)
	if err != nil {
		log.Println("Cannot open log file: ", err.Error())
	} else {
		log.SetOutput(io.MultiWriter(os.Stdout, handle))
	}
	gin.DefaultWriter = io.MultiWriter(os.Stdout, handle)
}

func callback(c *cli.Context) error {
	ip := c.String("ip")
	port := c.String("port")
	apikey := c.String("apikey")
	logfile := c.String("logfile")
	if logfile != "" {
		setupLoggingToFile(logfile)
	}
	if apikey == "" {
		return fmt.Errorf("No mega.nz api key provided, exiting.")
	}
	megaClient = getMegaClient(apikey)
	debug := c.Bool("debug")
	if !debug {
		gin.SetMode(gin.ReleaseMode)
	}
	log.Printf("Serving on %s:%s\n", ip, port)
	r := gin.Default()
	setupRoutes(r)
	return r.Run(fmt.Sprintf("%s:%s", ip, port))
}

func main() {
	app := cli.NewApp()
	app.Name = "MegaSDK-REST"
	app.Usage = "A web server encapsulating the downloading functionality of megasdk written in Go."
	app.Authors = []*cli.Author{
		{Name: "JaskaranSM"},
	}
	app.Action = callback
	app.Flags = []cli.Flag{
		&cli.StringFlag{
			Name:  "port",
			Value: "6090",
			Usage: "port to listen on",
		},
		&cli.StringFlag{
			Name:  "ip",
			Value: "",
			Usage: "ip to listen on, by default webserver listens on localhost",
		},
		&cli.StringFlag{
			Name:  "apikey",
			Value: "",
			Usage: "API Key for MegaSDK. (mandatory)",
		},
		&cli.BoolFlag{
			Name:  "debug",
			Value: false,
			Usage: "run webserver in debug mode",
		},
		&cli.StringFlag{
			Name:  "logfile",
			Value: "",
			Usage: "log to file provided",
		},
	}
	app.Version = "0.1"
	err := app.Run(os.Args)
	if err != nil {
		log.Fatal(err)
	}
}