goapi.go

package main

import (
	"fmt"
	"log"
	"net"
	"net/http"
	"os"
	"path"
	"strings"

	"github.com/grpc-ecosystem/grpc-gateway/runtime"
	"github.com/mch1307/go-ws-api/db"
	"github.com/mch1307/go-ws-api/pb"
	context "golang.org/x/net/context"
	grpc "google.golang.org/grpc"
)

const grpcPort = ":8082"
const httpPort = ":8081"

func main() {
	log.Println("Starting application")
	db.InitDB()
	fmt.Println(db.GetAllDevices())
	// start listening for grpc
	listen, err := net.Listen("tcp", grpcPort)
	if err != nil {
		log.Fatal(err)
	}
	// Instanciate new gRPC server
	server := grpc.NewServer()
	//register service
	pb.RegisterDeviceServiceServer(server, new(DeviceService))

	log.Println("Starting grpc server on port " + grpcPort)

	// Start the gRPC server in goroutine
	go server.Serve(listen)

	// Start the HTTP server for Rest
	log.Println("Starting HTTP server on port " + httpPort)
	run()
}

func serveSwagger(w http.ResponseWriter, r *http.Request) {
	//swagger := http.FileServer(http.Dir("./3rdparty/swagger-ui"))
	fmt.Println("request", r.URL.Path)
	p := strings.TrimPrefix(r.URL.Path, "/swagger/")
	p = path.Join("3rdparty/swagger-ui/", p)
	fmt.Println("request map ", p)
	http.ServeFile(w, r, p)

}

func run() error {
	ctx := context.Background()
	ctx, cancel := context.WithCancel(ctx)
	defer cancel()

	gw := runtime.NewServeMux()
	opts := []grpc.DialOption{grpc.WithInsecure()}
	err := pb.RegisterDeviceServiceHandlerFromEndpoint(ctx, gw, "localhost"+grpcPort, opts)
	if err != nil {
		return err
	}
	mux := http.NewServeMux()
	mux.HandleFunc("/swagger/", serveSwagger)
	curdir, _ := os.Getwd()
	fmt.Println("cur dir", curdir)
	//swagger := http.FileServer(http.Dir(filepath.Join(curdir, "3rdparty", "swagger-ui")))
	//mux.Handle("/swagger/", swagger)
	mux.Handle("/api/", gw)

	return http.ListenAndServe(httpPort, mux)
}

// DeviceService service for devices
type DeviceService struct{}

// GetAllDevices returns all registered devices
func (s *DeviceService) GetAllDevices(ctx context.Context, req *pb.Empty) (*pb.Devices, error) {
	devices := db.GetAllDevices()
	return &devices, nil
}

// GetDeviceByID returns a specific device
func (s *DeviceService) GetDeviceByID(ctx context.Context, id *pb.ID) (*pb.Device, error) {
	device := db.GetDeviceByID(id.Id)
	return device, nil
}

// SwitchDevice turns a device on or off
func (s *DeviceService) SwitchDevice(ctx context.Context, device *pb.UpdateDevice) (*pb.Device, error) {
	updatedDevice, err := db.SwitchDevice(device.Id, device.Value)
	if err != nil {
		log.Println("error updating device ", err)
	}
	return updatedDevice, err
}

// RegisterDevice registers a new device
func (s *DeviceService) RegisterDevice(ctx context.Context, device *pb.Device) (*pb.Device, error) {
	return nil, nil
}