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/settings"
	w "github.com/tockins/realize/watcher"
	"golang.org/x/net/websocket"
	"log"
	"net/http"
	"strconv"
	"gopkg.in/urfave/cli.v2"
)

// Server struct contains server informations
type Server struct {
	*c.Settings  `yaml:"-"`
	*w.Blueprint `yaml:"-"`
	Sync         chan string `yaml:"-"`
}

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(p *cli.Context) (err error) {
	if !p.Bool("no-server") && s.Enabled {
		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()))

		go e.Run(standard.New(string(s.Settings.Server.Host) + ":" + strconv.Itoa(s.Settings.Server.Port)))
		if s.Open || p.Bool("open") {
			_, err = Open("http://" + string(s.Settings.Server.Host) + ":" + strconv.Itoa(s.Settings.Server.Port))
			if err != nil {
				return err
			}
		}
	}
	return nil
}

// 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()
			}
		}
	})
}