arozos-mirror/src/mod/agi/agi.share.go

package agi

import (
	"fmt"
	"os"
	"time"

	"github.com/robertkrimen/otto"
	"imuslab.com/arozos/mod/agi/static"
	"imuslab.com/arozos/mod/info/logger"
)

func (g *Gateway) ShareLibRegister() {
	err := g.RegisterLib("share", g.injectShareFunctions)
	if err != nil {
		logger.PrintAndLog("Agi", fmt.Sprint(err), nil)
		os.Exit(1)
	}
}

func (g *Gateway) injectShareFunctions(payload *static.AgiLibInjectionPayload) {
	vm := payload.VM
	u := payload.User
	//scriptFsh := payload.ScriptFsh
	//scriptPath := payload.ScriptPath
	//w := payload.Writer
	//r := payload.Request
	vm.Set("_share_file", func(call otto.FunctionCall) otto.Value {
		//Get the vpath of file to share
		vpath, err := call.Argument(0).ToString()
		if err != nil {
			return otto.New().MakeCustomError("Unable to decode filepath", "No given filepath for sharing")
		}

		//Get the timeout from the 2nd parameter for how long this share will exists
		timeout, err := call.Argument(1).ToInteger()
		if err != nil {
			//Not defined -> Do not expire
			timeout = 0
		}

		//Create a share object for this request
		vpathSourceFsh := u.GetRootFSHFromVpathInUserScope(vpath)
		shareID, err := g.Option.ShareManager.CreateNewShare(u, vpathSourceFsh, vpath)
		if err != nil {
			logger.PrintAndLog("Agi", "[AGI] Create Share Failed: "+err.Error(), nil)
			return otto.New().MakeCustomError("Share failed", err.Error())
		}

		if timeout > 0 {
			go func(timeout int) {
				time.Sleep(time.Duration(timeout) * time.Second)
				g.Option.ShareManager.RemoveShareByUUID(u, shareID.UUID)
				logger.PrintAndLog("Agi", "[AGI] Share auto-removed: "+shareID.UUID, nil)
			}(int(timeout))
		}

		r, _ := otto.ToValue(shareID.UUID)
		return r
	})

	vm.Set("_share_removeShare", func(call otto.FunctionCall) otto.Value {
		shareUUID, err := call.Argument(0).ToString()
		if err != nil {
			return otto.New().MakeCustomError("Failed to remove share", "No share UUID given")
		}
		err = g.Option.ShareManager.RemoveShareByUUID(u, shareUUID)
		if err != nil {
			logger.PrintAndLog("Agi", "[AGI] Share remove failed: "+err.Error(), nil)
			return otto.New().MakeCustomError("Failed to remove share", err.Error())
		}

		return otto.TrueValue()
	})

	vm.Set("_share_getShareUUID", func(call otto.FunctionCall) otto.Value {
		vpath, err := call.Argument(0).ToString()
		if err != nil {
			logger.PrintAndLog("Agi", "[AGI] Failed to get share UUID: filepath not given", nil)
			return otto.NullValue()
		}

		shareObject := g.Option.ShareManager.GetShareObjectFromUserAndVpath(u, vpath)
		if shareObject == nil {
			logger.PrintAndLog("Agi", "[AGI] Failed to get share UUID: File not shared", nil)
			return otto.NullValue()
		}

		shareUUID := shareObject.UUID
		val, _ := otto.ToValue(shareUUID)
		return val
	})

	vm.Set("_share_checkShareExists", func(call otto.FunctionCall) otto.Value {
		shareUUID, err := call.Argument(0).ToString()
		if err != nil {
			return otto.New().MakeCustomError("Failed to check share exists", "No share UUID given")
		}

		shareObject := g.Option.ShareManager.GetShareObjectFromUUID(shareUUID)
		r, _ := otto.ToValue(!(shareObject == nil))
		return r
	})

	vm.Set("_share_checkSharePermission", func(call otto.FunctionCall) otto.Value {
		shareUUID, err := call.Argument(0).ToString()
		if err != nil {
			return otto.New().MakeCustomError("Failed to check share permission", "No share UUID given")
		}

		shareObject := g.Option.ShareManager.GetShareObjectFromUUID(shareUUID)
		if shareObject == nil {
			return otto.NullValue()
		}
		r, _ := otto.ToValue(shareObject.Permission)
		return r
	})

	vm.Set("_share_fileIsShared", func(call otto.FunctionCall) otto.Value {
		vpath, err := call.Argument(0).ToString()
		if err != nil {
			return otto.New().MakeCustomError("Failed to check share exists", "No filepath given")
		}

		isShared := g.Option.ShareManager.FileIsShared(u, vpath)
		r, _ := otto.ToValue(isShared)
		return r
	})

	//Wrap all the native code function into an imagelib class
	vm.Run(`
		var share = {};
		share.shareFile = _share_file;
		share.removeShare = _share_removeShare;
		share.checkShareExists = _share_checkShareExists;
		share.fileIsShared = _share_fileIsShared;
		share.getFileShareUUID = _share_getShareUUID;
		share.checkSharePermission = _share_checkSharePermission;
	`)
}