package main

import (
	"os"
	"os/signal"
	"syscall"
	"fmt"
	"strings"

	log "github.com/Sirupsen/logrus"

	"github.com/munnerz/plex-elastic-transcoder/common"
	"github.com/munnerz/plex-elastic-transcoder/executors"

	_ "github.com/munnerz/plex-elastic-transcoder/executors/kubernetes"
)

const (
	cmdPath = "/plexmediaserver/bootstrap.sh"
	logFilePath = "/var/log/plex/plex-elastic-transcoder.log"
	plexServerURL = "10.12.14.16:32400"
)

var executor executors.Executor

func signals() {
	// Signal handling
	c := make(chan os.Signal, 1)

	signal.Notify(c, os.Interrupt)
	signal.Notify(c, syscall.SIGTERM)
	go func() {
		<-c
		log.Print("Shutting down...")

		if executor != nil {
			log.Print("Terminating running job...")
			err := executor.Stop()
			if err != nil {
				log.Fatal("Error terminating job: ", err)
				os.Exit(1)
			}

			log.Print("Successfully terminated running job.")
		}

		os.Exit(0)
	}()
}

func main() {
	// Setup signals
	signals()

	// Setup logs
	fo, err := os.Create(logFilePath)

	if err != nil {
		panic(fmt.Sprintf("Error opening log file: %s", err))
	}

	defer func() {
		if err := fo.Close(); err != nil {
			panic(fmt.Sprintf("Error closing file: %s", err))
		}
	}()

	log.SetOutput(fo)

	// Get the arguments passed to Plex New Transcoder
	args := os.Args[1:]
	wd, _ := os.Getwd()
	for i, arg := range args {
		if arg == "-progressurl" {
			// Change the progress URL to report to about the transcode
			args[i + 1] = strings.Replace(args[i+1], "127.0.0.1:32400", plexServerURL, 1)
		}
	}
	args = append([]string{wd}, args)

	log.Print("In WD: ", wd)
	log.Print("Dispatching job with args: ", args)

	job := executors.Job{
		Command: []string{cmdPath},
		Args: args,
	}

	executor = common.CreateExecutor(job)

	log.Print("Created executor: ", executor)

	err = executor.Start()
	if err != nil {
		log.Fatal("Job start failed with error: ", err)
	}

	log.Print("Waiting for build pod to enter Running state...")

	err = executor.WaitForState(executors.ExecutorRunning)
	if err != nil {
		log.Fatal("Error waiting for pod to enter running state: ", err)
	}

	log.Print("Job has started running...")
	log.Print("Waiting for job to complete...")

	err = executor.WaitForState(executors.ExecutorSucceeded)
	if err != nil {
		log.Fatal("Error waiting for job to complete: ", err)
	}

	log.Print("Job completed. Destroying pod...")

	err = executor.Stop()
	if err != nil {
		log.Fatal("Error stopping job: ", err)
	}

	log.Print("Pod destroyed. Exiting.")
}