all: rename rescached_httpd.go to httpd.go

1 files changed, 209 insertions, 0 deletions

diff --git a/httpd.go b/httpd.go
new file mode 100644
index 0000000..29ecc5c
--- /dev/null
+++ b/httpd.go
@@ -0,0 +1,209 @@
+// Copyright 2020, Shulhan <ms@kilabit.info>. All rights reserved.
+// Use of this source code is governed by a BSD-style
+// license that can be found in the LICENSE file.
+
+package rescached
+
+import (
+	"encoding/json"
+	"fmt"
+	"log"
+	stdhttp "net/http"
+
+	liberrors "github.com/shuLhan/share/lib/errors"
+	"github.com/shuLhan/share/lib/http"
+)
+
+const (
+	defHTTPDRootDir = "_www/public/"
+)
+
+func (srv *Server) httpdInit() (err error) {
+	env := &http.ServerOptions{
+		Root:    defHTTPDRootDir,
+		Address: srv.env.WuiListen,
+		Includes: []string{
+			`.*\.css`,
+			`.*\.html`,
+			`.*\.js`,
+			`.*\.png`,
+		},
+		CORSAllowOrigins: []string{
+			"http://127.0.0.1:5000",
+		},
+		CORSAllowHeaders: []string{
+			http.HeaderContentType,
+		},
+		Development: srv.env.Debug >= 3,
+	}
+
+	srv.httpd, err = http.NewServer(env)
+	if err != nil {
+		return fmt.Errorf("newHTTPServer: %w", err)
+	}
+
+	err = srv.httpdRegisterEndpoints()
+	if err != nil {
+		return fmt.Errorf("newHTTPServer: %w", err)
+	}
+
+	return nil
+}
+
+func (srv *Server) httpdRegisterEndpoints() (err error) {
+	epAPIGetEnvironment := &http.Endpoint{
+		Method:       http.RequestMethodGet,
+		Path:         "/api/environment",
+		RequestType:  http.RequestTypeJSON,
+		ResponseType: http.ResponseTypeJSON,
+		Call:         srv.httpdAPIGetEnvironment,
+	}
+
+	err = srv.httpd.RegisterEndpoint(epAPIGetEnvironment)
+	if err != nil {
+		return err
+	}
+
+	epAPIPostEnvironment := &http.Endpoint{
+		Method:       http.RequestMethodPost,
+		Path:         "/api/environment",
+		RequestType:  http.RequestTypeJSON,
+		ResponseType: http.ResponseTypeJSON,
+		Call:         srv.httpdAPIPostEnvironment,
+	}
+
+	err = srv.httpd.RegisterEndpoint(epAPIPostEnvironment)
+	if err != nil {
+		return err
+	}
+
+	epAPIPostHostsBlock := &http.Endpoint{
+		Method:       http.RequestMethodPost,
+		Path:         "/api/hosts_block",
+		RequestType:  http.RequestTypeJSON,
+		ResponseType: http.ResponseTypeJSON,
+		Call:         srv.apiPostHostsBlock,
+	}
+
+	err = srv.httpd.RegisterEndpoint(epAPIPostHostsBlock)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func (srv *Server) httpdRun() {
+	defer func() {
+		err := recover()
+		if err != nil {
+			log.Printf("httpServer: %s", err)
+		}
+	}()
+
+	log.Printf("=== rescached: httpd listening at %s", srv.env.WuiListen)
+
+	err := srv.httpd.Start()
+	if err != nil {
+		log.Printf("httpServer.run: %s", err)
+	}
+}
+
+func (srv *Server) httpdAPIGetEnvironment(
+	httpRes stdhttp.ResponseWriter, req *stdhttp.Request, reqBody []byte,
+) (
+	resBody []byte, err error,
+) {
+	return json.Marshal(srv.env)
+}
+
+func (srv *Server) httpdAPIPostEnvironment(
+	httpRes stdhttp.ResponseWriter, req *stdhttp.Request, reqBody []byte,
+) (
+	resBody []byte, err error,
+) {
+	newOpts := new(environment)
+	err = json.Unmarshal(reqBody, newOpts)
+	if err != nil {
+		return nil, err
+	}
+
+	newOpts.init()
+
+	fmt.Printf("new options: %+v\n", newOpts)
+
+	res := &liberrors.E{
+		Code:    stdhttp.StatusOK,
+		Message: "Restarting DNS server",
+	}
+
+	err = newOpts.write(srv.fileConfig)
+	if err != nil {
+		log.Println("httpdAPIPostEnvironment:", err.Error())
+		res.Code = stdhttp.StatusInternalServerError
+		res.Message = err.Error()
+		return json.Marshal(res)
+	}
+
+	srv.env = newOpts
+
+	srv.Stop()
+	err = srv.Start()
+	if err != nil {
+		log.Println("httpdAPIPostEnvironment:", err.Error())
+		res.Code = stdhttp.StatusInternalServerError
+		res.Message = err.Error()
+	}
+
+	return json.Marshal(res)
+}
+
+func (srv *Server) apiPostHostsBlock(
+	httpRes stdhttp.ResponseWriter, req *stdhttp.Request, reqBody []byte,
+) (
+	resBody []byte, err error,
+) {
+	hostsBlocks := make([]*hostsBlock, 0)
+
+	err = json.Unmarshal(reqBody, &hostsBlocks)
+	if err != nil {
+		return nil, err
+	}
+
+	res := &liberrors.E{
+		Code:    stdhttp.StatusOK,
+		Message: "Restarting DNS server",
+	}
+
+	for x, hb := range hostsBlocks {
+		fmt.Printf("apiPostHostsBlock[%d]: %+v\n", x, hb)
+	}
+
+	var mustRestart bool
+	for _, hb := range srv.env.HostsBlocks {
+		isUpdated := hb.update(hostsBlocks)
+		if isUpdated {
+			mustRestart = true
+		}
+	}
+
+	err = srv.env.write(srv.fileConfig)
+	if err != nil {
+		log.Println("apiPostHostsBlock:", err.Error())
+		res.Code = stdhttp.StatusInternalServerError
+		res.Message = err.Error()
+		return json.Marshal(res)
+	}
+
+	if mustRestart {
+		srv.Stop()
+		err = srv.Start()
+		if err != nil {
+			log.Println("apiPostHostsBlock:", err.Error())
+			res.Code = stdhttp.StatusInternalServerError
+			res.Message = err.Error()
+		}
+	}
+
+	return json.Marshal(res)
+}