server.go

package main

import (
	"encoding/json"
	"fmt"
	"math/rand"
	"net/http"
	"os"
	"time"

	"github.com/gin-gonic/gin"
	"github.com/glibs/gin-webserver"
	"github.com/olahol/melody"
	"github.com/skratchdot/open-golang/open"
)
var mel *melody.Melody
func web() {
	host := "0.0.0.0:8080"
	server := InitializeServer(host)
	server.Start()
	Log.Info("Gin web server started on " + host)
}

// InitializeServer gets our gin running front end poppin off
func InitializeServer(host string) (server *network.WebServer) {
	rand.Seed(time.Now().UTC().UnixNano())
	// Make sure folders exist that we want:
	if err := ensureBindDirs(); err != nil {
		Log.Error("Failed to have home working dir to put the files into at ~/Desktop/bind, err: ", err)
	} else {
		Log.Info("bind dirs ensured!")
	}
	if os.Args[0] != "d" { //development mode
		gin.SetMode(gin.ReleaseMode)
	}
	r := gin.New()
	r.LoadHTMLGlob("public/tmpl/*.html")
	r.StaticFS("/videos", http.Dir(basePath+"/videos"))
	r.StaticFS("/frames", http.Dir(basePath+"/frames"))
	r.Static("/public", "./public")
	r.GET("/", getIndex)
	r.POST("/g", postIndex)
	r.GET("/g", getIndex)
	r.GET("/about", getAbout)
	r.GET("/jobs", getJobs)
	r.GET("/code", getCode)
	mel = melody.New() // melody middleware

	// websocket route
	r.GET("/ws",func(ctx *gin.Context){
		// handle request with Melody
		mel.HandleRequest(ctx.Writer,ctx.Request)
	})

	// Melody message handler
	mel.HandleMessage(func(ses *melody.Session,msg []byte){
		// broadcast message to connected sockets
		mel.Broadcast(msg)
	})


	r.GET("/openframes", func(c *gin.Context) {
		open.Run(basePath + "/frames")
	})
	r.GET("/openvideos", func(c *gin.Context) {
		open.Run(basePath + "/videos")
	})
	r.GET("/openlogs", func(c *gin.Context) {
		open.Run(basePath + "/logs")
	})
	r.GET("/toggleClipYt", func(c *gin.Context) {
		open.Run(basePath + "/logs")
	})
	// go requests(mel)
	// go jobUpdates(mel)

	return network.InitializeWebServer(r, host)
}

type request struct {
	URL       []string
	Password  string `json:"-"`
	IsAdmin   bool
	CreatedAt time.Time
}



func requests(mel *melody.Melody) {
	time.Sleep(time.Second * 5) //wait for the server to warmup
	seen := make(map[string]bool)
	ticker := time.NewTicker(time.Second * 5).C
	request := request{} //initialize empty user
	for {
		select {
		case <-ticker:
			r, _ := http.Get("http://localhost:8081/bot")
			//Parse json request body and use it to set fields on user
			//Note that user is passed as a pointer variable so that it's fields can be modified
			err := json.NewDecoder(r.Body).Decode(&request.URL)
			if err != nil {
				panic(err)
			}
			for i := 0; len(request.URL) > i; i++ {
				if !seen[request.URL[i]] {
					s := fmt.Sprintf(`%s`, request.URL[i])
					mel.Broadcast([]byte(s))
					seen[request.URL[i]] = true
				}
			}
		}
	}
}