package master

import (
	"analysis/app"
	"analysis/logo"
	"analysis/util"
	"context"
	"encoding/json"
	"os"
	"strconv"
	"strings"
	"time"

	"basic.com/libgowrapper/sdkstruct.git"
	"basic.com/valib/pubsub.git"
)

func reaper(ctxt context.Context) {
	pidChan := make(chan int, 1)
	Reap(pidChan)
	go waitForRestart(ctxt, pidChan)
}

// Run run
func Run(ctx context.Context, soFile, configPath string) bool {
	reaper(ctx)

	fetcher := NewFetcher(soFile)
	if fetcher == nil {
		logo.Errorln("New Fetcher Load so File Funcs Error From File: ", soFile)
		return false
	}

	logo.Infoln("~~~~~~Created Fetcher, Now Sync From DB")

	ip := "tcp://" + util.FSI.IP
	url := ip + ":" + strconv.Itoa(util.FSI.DataPort)
	hearturl := ip + ":" + strconv.Itoa(util.FSI.HBPort)

	chMsg, err := fetcher.fnInit(url, hearturl, 0, []string{pubsub.Topic_Sdk}, "analysis-master"+strconv.Itoa(os.Getpid()))
	for {
		if err == nil {
			break
		}
		logo.Infoln("Analysis Fetcher INIT Error! URL:", url)
		time.Sleep(time.Second)
		chMsg, err = fetcher.fnInit(url, hearturl, 0, []string{pubsub.Topic_Sdk}, "analysis-master"+strconv.Itoa(os.Getpid()))
	}

	for {
		select {
		case <-ctx.Done():
			return true
		case msg := <-chMsg:
			//      		sdktype	   process_name   topic		null
			//				yolo/face  yolo_0/yolo_1  channel
			var sdk map[string](map[string](map[string]interface{}))

			if err := json.Unmarshal(msg.Msg, &sdk); err != nil {
				logo.Infoln("Fetcher SDK unmarshal err:", err)
				continue
			}

			logo.Infoln("~~~~~~Recv New SDKInfos")
			chCameras <- CameraInfo{
				Cameras: cameras,
			}
			logo.Infoln("~~~~~~Recv New SDKInfos Over")

		default:
			time.Sleep(10 * time.Millisecond)
		}
	}

}

func manualStart(ctx context.Context, sdks []sdkstruct.SDKInfo, configPath string) bool {
	rPath := configPath

	params := app.GetParams()

	for _, v := range sdks {

		file := rPath + v.SdkType + ".json"
		if rPath[len(rPath)-1] != '/' {
			file = rPath + "/" + v.SdkType + ".json"
		}

		cfg, err := app.ReadConfig(file)
		if err != nil {
			logo.Errorln("Master Read: ", file, " Config Error: ", err)
			continue
		}

		if len(cfg.Env) > 0 {
			envs := strings.Split(cfg.Env, ":")
			normal := true
			for _, v := range envs {
				if !util.IsFileExist(v) {
					normal = false
					break
				}
			}
			if !normal {
				logo.Infoln("Can't Find Runtime Path, Skip It: ", file)
				continue
			}
		}

		logo.Infoln(file, " CONFIG: ", cfg)

		args := []string{
			`-role=slave`,
			"-sdk=" + v.SdkType,
			"-id=" + v.IpcID,
			"-" + util.ConfigPath + "=" + file,
		}

		args = append(args, params...)
		pid, err := runProc(ctx, "./analysis", args, cfg.Env)

		if err != nil {
			logo.Errorf("ANALYSIS START SLAVE PROC %s IPC: %s error %+v\n", v.SdkType, v.IpcID, err)
		}
		logo.Infof("START SDK %s ID %s PID %d Env: %s\n", v.SdkType, v.IpcID, pid, cfg.Env)
	}
	return true

}