th7/api/api.go

package api

import (
	"fmt"
	"log"

	"github.com/gin-gonic/gin"

	"th7/data/config"
	"th7/ports"
)

type GinAdapter struct {
	router   *gin.Engine
	corePort ports.CorePort
	cfg      config.Config
	port     int
}

func (g *GinAdapter) IndexHandler(c *gin.Context) {
	c.File("./static/index.html")
}

func NewGinAdapter(corePort ports.CorePort, cfg config.Config) *GinAdapter {
	var adapter GinAdapter

	adapter.corePort = corePort
	adapter.port = cfg.API.Port
	adapter.cfg = cfg

	//gin.SetMode(gin.ReleaseMode)

	adapter.router = gin.New()

	// websockets
	manager := NewManager()
	go manager.start()

	// API
	adapter.router.GET("/v1/data/channel/:id", adapter.GetChannelByIDHandler)
	adapter.router.GET("/v1/data/channels", adapter.GetChannelsHandler)
	adapter.router.GET("/v1/data/ratio", adapter.GetRatioHandler)
	adapter.router.GET("/v1/data", adapter.GetDataHandler)

	adapter.router.GET("/v1/config/channel/:id", adapter.GetConfigChannelByIdHandler)
	adapter.router.GET("/v1/config/channels", adapter.GetConfigChannelsHandler)
	adapter.router.GET("/v1/config/device", adapter.GetConfigDeviceHandler)
	adapter.router.GET("/v1/config/db", adapter.GetConfigDBHandler)

	adapter.router.GET("/ws", func(c *gin.Context) {
		conn, err := upgrader.Upgrade(c.Writer, c.Request, nil)
		if err != nil {
			log.Println(err)
		}

		client := &Client{
			socket:   conn,
			send:     make(chan []byte),
			manager:  manager,
			all:      false,
			channels: false,
			config:   false,
		}

		manager.register <- client

		go client.read()
		go client.write()
	})

	adapter.router.POST("/v1/config/device", adapter.SetConfigDeviceHandler)
	adapter.router.POST("/v1/config/channel/:id", adapter.SetConfigChannelByIdHandler)
	adapter.router.POST("/v1/config/db", adapter.SetConfigDBHandler)

	// start service that dumps new data to websocket chan
	go startService(manager, corePort)

	// TH7 demo code. html file and javascript
	adapter.router.GET("/", adapter.IndexHandler)
	adapter.router.Static("/assets", "./static")

	return &adapter
}

func (g *GinAdapter) Run() {
	portStr := fmt.Sprintf(":%d", g.port)
	g.router.Run(portStr)
}