libmutton/libmuttonserver.go

package main

import (
	"bufio"
	"encoding/json"
	"fmt"
	"os"
	"runtime/debug"
	"strings"

	"github.com/rwinkhart/go-boilerplate/back"
	"github.com/rwinkhart/go-boilerplate/other"
	"github.com/rwinkhart/libmutton/global"
	"github.com/rwinkhart/libmutton/synccommon"
	"github.com/rwinkhart/libmutton/syncserver"
)

func main() {
	args := os.Args
	if len(args) < 2 {
		helpServer()
	}

	// check if stdin was provided
	stdinInfo, _ := os.Stdin.Stat()
	stdinPresent := stdinInfo.Mode()&os.ModeNamedPipe != 0

	var stdin []string
	if stdinPresent {
		// read stdin, appending each line to the stdin slice
		scanner := bufio.NewScanner(os.Stdin)
		for scanner.Scan() {
			stdin = append(stdin, scanner.Text())
		}
	}

	// allow reporting panic details to clients
	defer func() {
		if r := recover(); r != nil {
			fmt.Printf(
				"{\"errMsg\":\"SERVER-SIDE PANIC OCCURRED: %v%sARGS: %v%sSTDIN: %v%sSTACK TRACE: %s\"}",
				r,
				global.FSSpace+global.FSSpace,
				global.FSSpace+strings.Join(os.Args, global.FSSpace),
				global.FSSpace+global.FSSpace,
				global.FSSpace+strings.Join(stdin, global.FSSpace),
				global.FSSpace+global.FSSpace,
				global.FSSpace+strings.ReplaceAll(strings.ReplaceAll(string(debug.Stack()), "\n", global.FSSpace), "\t", ""),
			)
			return
		}
	}()

	switch args[1] {
	case "fetch":
		// print all information needed for syncing to stdout for interpretation by the client
		// stdin[0] is expected to be the device ID
		syncserver.GetRemoteDataFromServer(stdin[0])
	case "rename":
		// move an entry to a new location before using fallthrough to add its previous iteration to the deletions directory
		// stdin[0] is evaluated after fallthrough
		// stdin[1] is expected to be the OLD vanityPath with FSPath representing path separators - Always pass in UNIX format
		// stdin[2] is expected to be the NEW vanityPath with FSPath representing path separators - Always pass in UNIX format
		err := synccommon.RenameLocal(strings.ReplaceAll(stdin[1], global.FSPath, "/"), strings.ReplaceAll(stdin[2], global.FSPath, "/"))
		if err != nil {
			fmt.Printf("{\"errMsg\":\"%s\"}", err.Error())
			return
		}
		fallthrough // fallthrough to add the old entry to the deletions directory
	case "shear":
		// shear an entry from the server and add it to the deletions directory
		// stdin[0] is expected to be the device ID
		// stdin[1] is expected to be the vanityPath with FSPath representing path separators - Always pass in UNIX format
		_, _, err := synccommon.ShearLocal(strings.ReplaceAll(stdin[1], global.FSPath, "/"), stdin[0], false)
		if err != nil {
			fmt.Printf("{\"errMsg\":\"%s\"}", err.Error())
			return
		}
	case "shear-age":
		// shear ONLY the age file associated with an entry from the server and add it to the deletions directory
		// stdin[0] is expected to be the device ID
		// stdin[1] is expected to be the vanityPath with FSPath representing path separators - Always pass in UNIX format
		_, _, err := synccommon.ShearLocal(strings.ReplaceAll(stdin[1], global.FSPath, "/"), stdin[0], true)
		if err != nil {
			fmt.Printf("{\"errMsg\":\"%s\"}", err.Error())
			return
		}
	case "addfolder":
		// add a new folder to the server
		// stdin[0] is expected to be the vanityPath with FSPath representing path separators - Always pass in UNIX format
		err := synccommon.AddFolderLocal(strings.ReplaceAll(stdin[0], global.FSPath, "/"))
		if err != nil {
			fmt.Printf("{\"errMsg\":\"%s\"}", err.Error())
			return
		}
	case "register":
		// register a new device ID
		// stdin[0] is expected to be the device ID
		// stdin[1] is expected to be the old device ID (for removal)
		f, err := os.OpenFile(global.CfgDir+global.PathSeparator+"devices"+global.PathSeparator+stdin[0], os.O_CREATE|os.O_WRONLY, 0600)
		if err != nil {
			fmt.Printf("{\"errMsg\":\"%s\"}", err.Error())
			return
		}
		_ = f.Close()
		if stdin[1] != global.FSMisc { // FSMisc is used to indicate that no device ID is being replaced
			// remove the old device ID file
			err = os.RemoveAll(global.CfgDir + global.PathSeparator + "devices" + global.PathSeparator + stdin[1])
			if err != nil {
				fmt.Printf("{\"errMsg\":\"%s\"}", err.Error())
				return
			}
			// carry over deletions from the old device ID to the new one
			deletionsDirRoot := global.CfgDir + global.PathSeparator + "deletions" + global.PathSeparator
			deletionsList, err := os.ReadDir(deletionsDirRoot)
			if err != nil {
				fmt.Printf("{\"errMsg\":\"%s\"}", err.Error())
				return
			}
			for _, deletion := range deletionsList {
				affectedIDVanityPath := strings.Split(deletion.Name(), global.FSSpace)
				if affectedIDVanityPath[0] == stdin[1] {
					err = os.Rename(deletionsDirRoot+deletion.Name(), deletionsDirRoot+stdin[0]+global.FSSpace+affectedIDVanityPath[1]+global.FSSpace+affectedIDVanityPath[2])
					if err != nil {
						fmt.Printf("{\"errMsg\":\"%s\"}", err.Error())
						return
					}
				}
			}
		}

		// print EntryRoot, AgeDir and bool indicating OS type to stdout for client to store in config
		registerResp := synccommon.RegisterResp{EntryRoot: global.EntryRoot, AgeDir: global.AgeDir, IsWindows: global.IsWindows}
		registerRespBytes, err := json.Marshal(registerResp)
		if err != nil {
			fmt.Printf("{\"errMsg\":\"%s\"}", err.Error())
			return
		}
		fmt.Print(string(registerRespBytes))
	case "init":
		// create the necessary directories for libmuttonserver to function
		_, err := global.DirInit(false)
		if err != nil {
			other.PrintError("Failed to initialize libmuttonserver directories: "+err.Error(), back.ErrorWrite)
		}
		_ = os.MkdirAll(global.CfgDir+global.PathSeparator+"deletions", 0700) // error ignored; failure would have occurred by this point in core.DirInit
		fmt.Println("libmuttonserver directories initialized")
	case "version":
		versionServer()
	default:
		helpServer()
	}
}

func helpServer() {
	fmt.Print(back.AnsiBold + "\nlibmuttonserver | Copyright (c) 2024-2025 Randall Winkhart\n" + back.AnsiReset + `
This software exists under the MIT license; you may redistribute it under certain conditions.
This program comes with absolutely no warranty; type "libmuttonserver version" for details.

` + back.AnsiBold + "Usage:" + back.AnsiReset + ` libmuttonserver <argument>

` + back.AnsiBold + "Arguments (user):" + back.AnsiReset + `
 help                    Bring up this menu
 version                 Display version and license information
 init                    Create the necessary directories for libmuttonserver to function` + "\n\n")
	os.Exit(0)
}

func versionServer() {
	fmt.Print(back.AnsiBold + "\n                    MIT License" + back.AnsiReset + `

  Permission is hereby granted, free of charge, to any
person obtaining a copy of this software and associated
  documentation files (the "Software"), to deal in the
    Software without restriction, including without
   limitation the rights to use, copy, modify, merge,
 publish, distribute, sublicense, and/or sell copies of
the Software, and to permit persons to whom the Software
    is furnished to do so, subject to the following
                      conditions:

 The above copyright notice and this permission notice
shall be included in all copies or substantial portions
                   of the Software.

 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.` + "\n\n---------------------------------------------------------")
	fmt.Print(back.AnsiBold + "\n\n              libmuttonserver" + back.AnsiReset + " Version " + global.LibmuttonVersion + "\n\n        Copyright (c) 2024-2025: Randall Winkhart" + "\n\n")
}