realize/server/main.go

package server

import (
	"encoding/json"
	"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
	Open      bool
}

func render(c echo.Context, path string, mime int) error {
	data, err := Asset(path)
	if err != nil {
		return echo.NewHTTPError(http.StatusNotFound)
	}
	rs := c.Response()
	// check content type by extensions
	switch mime {
	case 1:
		rs.Header().Set(echo.HeaderContentType, echo.MIMETextHTMLCharsetUTF8)
		break
	case 2:
		rs.Header().Set(echo.HeaderContentType, echo.MIMEApplicationJavaScriptCharsetUTF8)
		break
	case 3:
		rs.Header().Set(echo.HeaderContentType, "text/css")
		break
	case 4:
		rs.Header().Set(echo.HeaderContentType, "image/svg+xml")
		break
	}
	rs.WriteHeader(http.StatusOK)
	rs.Write(data)
	return nil
}

// Server starting
func (s *Server) Start() {
	e := echo.New()
	e.Use(middleware.Gzip())

	// web panel
	e.GET("/", func(c echo.Context) error {
		return render(c, "assets/index.html", 1)
	})
	e.GET("/assets/js/all.min.js", func(c echo.Context) error {
		return render(c, "assets/assets/js/all.min.js", 2)
	})
	e.GET("/assets/css/app.css", func(c echo.Context) error {
		return render(c, "assets/assets/css/app.css", 3)
	})
	e.GET("/app/components/projects/index.html", func(c echo.Context) error {
		return render(c, "assets/app/components/projects/index.html", 1)
	})
	e.GET("/app/components/project/index.html", func(c echo.Context) error {
		return render(c, "assets/app/components/project/index.html", 1)
	})
	e.GET("/app/components/index.html", func(c echo.Context) error {
		return render(c, "assets/app/components/index.html", 1)
	})
	e.GET("/assets/img/svg/github-logo.svg", func(c echo.Context) error {
		return render(c, "assets/assets/img/svg/github-logo.svg", 4)
	})
	e.GET("/assets/img/svg/ic_error_black_48px.svg", func(c echo.Context) error {
		return render(c, "assets/assets/img/svg/ic_error_black_48px.svg", 4)
	})
	e.GET("/assets/img/svg/ic_swap_vertical_circle_black_48px.svg", func(c echo.Context) error {
		return render(c, "assets/assets/img/svg/ic_swap_vertical_circle_black_48px.svg", 4)
	})

	//websocket
	//e.GET("/ws", standard.WrapHandler(s.projects()))
	e.GET("/ws", standard.WrapHandler(s.projects()))
	go e.Run(standard.New(":5000"))
	if s.Open {
		Open("http://localhost:5000")
	}
}

// The WebSocket for projects list
func (s *Server) projects() websocket.Handler {
	return websocket.Handler(func(ws *websocket.Conn) {
		msg := func() {

			message, _ := json.Marshal(s.Blueprint.Projects)
			err := websocket.Message.Send(ws, string(message))
			if err != nil {
				log.Fatal(err)
			}
		}
		msg()
		for {
			select {
			case <-s.Sync:
				msg()
			}
		}
	})
}