arozos-mirror/src/main.go

package main

import (
	"flag"
	"fmt"
	"net/http"
	"os"
	"os/signal"
	"path/filepath"
	"strconv"
	"syscall"
	"time"

	console "imuslab.com/arozos/mod/console"
	fs "imuslab.com/arozos/mod/filesystem"
	"imuslab.com/arozos/mod/info/logger"
)

/*
	arozos
	author: tobychui

	To edit startup flags, see main.flag.go
	To edit main routing logic, see main.router.go
	To edit startup sequence, see startup.go

	P.S. Try to keep this file < 300 lines
*/

// Close handler, close db and clearn up everything before exit
func SetupCloseHandler() {
	c := make(chan os.Signal, 2)
	signal.Notify(c, os.Interrupt, syscall.SIGTERM)
	go func() {
		<-c
		executeShutdownSequence()
	}()
}

func executeShutdownSequence() {
	//Shutdown authAgent
	systemWideLogger.PrintAndLog("System", "<!> Shutting down auth gateway", nil)
	authAgent.Close()

	//Shutdown all storage pools
	systemWideLogger.PrintAndLog("System", "<!> Shutting down storage pools", nil)
	closeAllStoragePools()

	//Shutdown Logger
	systemWideLogger.Close()

	//Shutdown database
	systemWideLogger.PrintAndLog("System", "<!> Shutting down database", nil)
	sysdb.Close()

	//Shutdown network services
	StopNetworkServices()

	//Shutdown FTP Server
	if FTPManager != nil {
		systemWideLogger.PrintAndLog("System", "<!> Shutting down FTP Server", nil)
		FTPManager.StopFtpServer()
	}

	//Cleaning up tmp files
	systemWideLogger.PrintAndLog("System", "<!> Cleaning up tmp folder", nil)
	os.RemoveAll(*tmp_directory)
	//Do other things
	os.Exit(0)
}

func main() {
	//Parse startup flags and paramters
	flag.Parse()
	logger.SetGlobalJSONOutput(*log_format == "json")

	//Handle version printing
	if *show_version {
		fmt.Println("ArozOS " + build_version + " Revision " + internal_version)
		fmt.Println("Developed by tobychui and other co-developers, Licensed to " + deviceVendor)
		//fmt.Println("THE SOFTWARE IS PROVIDED 'AS IS', WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.")
		os.Exit(0)
	}

	//Handle flag assignments
	max_upload_size = int64(*max_upload) << 20 //Parse the max upload size

	//Clean up previous tmp files
	final_tmp_directory := filepath.Clean(*tmp_directory) + "/tmp/"
	tmp_directory = &final_tmp_directory
	os.RemoveAll(*tmp_directory)
	os.Mkdir(*tmp_directory, 0777)

	// Initialize a temporary stdout-only logger so calls before RunStartup are safe.
	// RunStartup will replace this with a file-backed logger.
	systemWideLogger, _ = logger.NewTmpLogger()
	logger.SetDefaultLogger(systemWideLogger)

	//Print copyRight information
	systemWideLogger.PrintAndLog("System", "ArozOS(C) "+strconv.Itoa(time.Now().Year())+" "+deviceVendor+".", nil)
	systemWideLogger.PrintAndLog("System", "ArozOS "+build_version+" Revision "+internal_version, nil)

	/*
		New Implementation of the ArOZ Online System, Sept 2020
	*/
	RunStartup()

	/*
		Development build test execution
	*/
	Run_Test()

	//Initiate all the static files transfer
	baseFileServer := http.FileServer(http.Dir("./web"))
	vendorWebDir := filepath.Join(vendorResRoot, "web")
	// Wrap the base file server: serve from vendorResRoot/web/ when a matching
	// file exists there, falling through to ./web/ otherwise.
	fileServer := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		vendorPath := filepath.Join(vendorWebDir, filepath.FromSlash(r.URL.Path))
		if !vendorWebExists {
			// Early exit if the vendor web directory doesn't exist, to avoid unnecessary file existence checks on every request.
			baseFileServer.ServeHTTP(w, r)
			return
		}
		if info, err := os.Stat(vendorPath); err == nil && !info.IsDir() {
			http.ServeFile(w, r, vendorPath)
			return
		}
		baseFileServer.ServeHTTP(w, r)
	})
	vendorWebExists = fs.FileExists(vendorWebDir)

	//Updates 2022-09-06: Gzip handler moved inside the master router
	http.Handle("/", mrouter(fileServer))

	//Setup handler for Ctrl +C
	SetupCloseHandler()

	//Start http server
	go func() {
		if *use_tls {
			if !*disable_http {
				go func() {
					address := fmt.Sprintf("%s:%d", *listen_host, *listen_port)
					systemWideLogger.PrintAndLog("System", fmt.Sprint("Standard (HTTP) Web server listening at", address), nil)
					http.ListenAndServe(address, nil)
				}()
			}
			address := fmt.Sprintf("%s:%d", *listen_host, *tls_listen_port)
			systemWideLogger.PrintAndLog("System", fmt.Sprint("Secure (HTTPS) Web server listening at", address), nil)
			http.ListenAndServeTLS(address, *tls_cert, *tls_key, nil)
		} else {
			address := fmt.Sprintf("%s:%d", *listen_host, *listen_port)
			systemWideLogger.PrintAndLog("System", fmt.Sprint("Web server listening at", address), nil)
			http.ListenAndServe(address, nil)
		}
	}()

	if *enable_console {
		//Startup interactive shell for debug and basic controls
		Console := console.NewConsole(consoleCommandHandler)
		Console.ListenAndHandle()
	} else {
		//Just do a blocking loop here
		select {}
	}

}