69 lines
1.4 KiB
Go
69 lines
1.4 KiB
Go
package server
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"github.com/labstack/echo"
|
|
"github.com/labstack/echo/engine/standard"
|
|
"github.com/labstack/echo/middleware"
|
|
c "github.com/tockins/realize/cli"
|
|
"golang.org/x/net/websocket"
|
|
"log"
|
|
"net/http"
|
|
)
|
|
|
|
// Server struct contains server informations
|
|
type Server struct {
|
|
Blueprint *c.Blueprint
|
|
Files map[string]string
|
|
Sync chan string
|
|
}
|
|
|
|
func render(c echo.Context, path string) error {
|
|
data, err := Asset(path)
|
|
if err != nil {
|
|
return echo.NewHTTPError(http.StatusNotFound)
|
|
}
|
|
rs := c.Response()
|
|
rs.Header().Set(echo.HeaderContentType, echo.MIMETextHTMLCharsetUTF8)
|
|
rs.WriteHeader(http.StatusOK)
|
|
rs.Write(data)
|
|
return nil
|
|
}
|
|
|
|
// Server starting
|
|
func (s *Server) Start() {
|
|
e := echo.New()
|
|
e.Use(middleware.Gzip())
|
|
e.GET("/", func(c echo.Context) error {
|
|
return c.JSON(200, s.Blueprint.Projects)
|
|
//return render(c, "server/assets/index.html")
|
|
})
|
|
|
|
e.GET("/projects", standard.WrapHandler(s.projects()))
|
|
go e.Run(standard.New(":5000"))
|
|
}
|
|
|
|
// The WebSocket for projects list
|
|
func (s *Server) projects() websocket.Handler {
|
|
return websocket.Handler(func(ws *websocket.Conn) {
|
|
msg := func() {
|
|
fmt.Println("tick")
|
|
message, _ := json.Marshal(s.Blueprint.Projects)
|
|
err := websocket.Message.Send(ws, string(message))
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
msg()
|
|
for {
|
|
select {
|
|
default:
|
|
continue
|
|
case <-s.Sync:
|
|
msg()
|
|
}
|
|
}
|
|
})
|
|
}
|