lash/attack-of-the-clones #57
| @ -77,10 +77,10 @@ func main() { | |||||||
| 	pfp := path.Join(scriptDir, "pp.csv") | 	pfp := path.Join(scriptDir, "pp.csv") | ||||||
| 
 | 
 | ||||||
| 	cfg := engine.Config{ | 	cfg := engine.Config{ | ||||||
| 		Root:          "root", | 		Root:              "root", | ||||||
| 		OutputSize:    uint32(size), | 		OutputSize:        uint32(size), | ||||||
| 		FlagCount:     uint32(128), | 		FlagCount:         uint32(128), | ||||||
| 		MenuSeparator: menuSeparator, | 		MenuSeparator:     menuSeparator, | ||||||
| 		ResetOnEmptyInput: true, | 		ResetOnEmptyInput: true, | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
|  | |||||||
| @ -92,10 +92,10 @@ func main() { | |||||||
| 	pfp := path.Join(scriptDir, "pp.csv") | 	pfp := path.Join(scriptDir, "pp.csv") | ||||||
| 
 | 
 | ||||||
| 	cfg := engine.Config{ | 	cfg := engine.Config{ | ||||||
| 		Root:          "root", | 		Root:              "root", | ||||||
| 		OutputSize:    uint32(size), | 		OutputSize:        uint32(size), | ||||||
| 		FlagCount:     uint32(128), | 		FlagCount:         uint32(128), | ||||||
| 		MenuSeparator: menuSeparator, | 		MenuSeparator:     menuSeparator, | ||||||
| 		ResetOnEmptyInput: true, | 		ResetOnEmptyInput: true, | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
|  | |||||||
| @ -78,10 +78,10 @@ func main() { | |||||||
| 	pfp := path.Join(scriptDir, "pp.csv") | 	pfp := path.Join(scriptDir, "pp.csv") | ||||||
| 
 | 
 | ||||||
| 	cfg := engine.Config{ | 	cfg := engine.Config{ | ||||||
| 		Root:          "root", | 		Root:              "root", | ||||||
| 		OutputSize:    uint32(size), | 		OutputSize:        uint32(size), | ||||||
| 		FlagCount:     uint32(128), | 		FlagCount:         uint32(128), | ||||||
| 		MenuSeparator: menuSeparator, | 		MenuSeparator:     menuSeparator, | ||||||
| 		ResetOnEmptyInput: true, | 		ResetOnEmptyInput: true, | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
|  | |||||||
| @ -38,7 +38,7 @@ func main() { | |||||||
| 	var stateDebug bool | 	var stateDebug bool | ||||||
| 	var host string | 	var host string | ||||||
| 	var port uint | 	var port uint | ||||||
| 	 | 
 | ||||||
| 	flag.StringVar(&override.DbConn, "c", "?", "default connection string (replaces all unspecified strings)") | 	flag.StringVar(&override.DbConn, "c", "?", "default connection string (replaces all unspecified strings)") | ||||||
| 	flag.StringVar(&override.ResourceConn, "resource", "?", "resource connection string") | 	flag.StringVar(&override.ResourceConn, "resource", "?", "resource connection string") | ||||||
| 	flag.StringVar(&override.UserConn, "userdata", "?", "userdata store connection string") | 	flag.StringVar(&override.UserConn, "userdata", "?", "userdata store connection string") | ||||||
| @ -81,9 +81,9 @@ func main() { | |||||||
| 	pfp := path.Join(scriptDir, "pp.csv") | 	pfp := path.Join(scriptDir, "pp.csv") | ||||||
| 
 | 
 | ||||||
| 	cfg := engine.Config{ | 	cfg := engine.Config{ | ||||||
| 		Root:       "root", | 		Root:              "root", | ||||||
| 		OutputSize: uint32(size), | 		OutputSize:        uint32(size), | ||||||
| 		FlagCount:  uint32(128), | 		FlagCount:         uint32(128), | ||||||
| 		ResetOnEmptyInput: true, | 		ResetOnEmptyInput: true, | ||||||
| 	} | 	} | ||||||
| 	if stateDebug { | 	if stateDebug { | ||||||
|  | |||||||
| @ -25,7 +25,7 @@ const ( | |||||||
| 	defaultSSHHost  string = "127.0.0.1" | 	defaultSSHHost  string = "127.0.0.1" | ||||||
| 	defaultSSHPort  uint   = 7122 | 	defaultSSHPort  uint   = 7122 | ||||||
| 	defaultHTTPHost string = "127.0.0.1" | 	defaultHTTPHost string = "127.0.0.1" | ||||||
| 	defaultHTTPPort uint     = 7123 | 	defaultHTTPPort uint   = 7123 | ||||||
| 	defaultDomain          = "sarafu.local" | 	defaultDomain          = "sarafu.local" | ||||||
| ) | ) | ||||||
| 
 | 
 | ||||||
| @ -52,7 +52,6 @@ func SearchDomains() []string { | |||||||
| 	return ParsedDomains | 	return ParsedDomains | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| 
 |  | ||||||
| func Language() string { | func Language() string { | ||||||
| 	return viseconfig.DefaultLanguage | 	return viseconfig.DefaultLanguage | ||||||
| } | } | ||||||
|  | |||||||
| @ -7,8 +7,8 @@ import ( | |||||||
| 	"os" | 	"os" | ||||||
| 	"path" | 	"path" | ||||||
| 
 | 
 | ||||||
| 	"git.defalsify.org/vise.git/logging" |  | ||||||
| 	"git.defalsify.org/vise.git/engine" | 	"git.defalsify.org/vise.git/engine" | ||||||
|  | 	"git.defalsify.org/vise.git/logging" | ||||||
| 
 | 
 | ||||||
| 	"git.grassecon.net/grassrootseconomics/sarafu-vise/config" | 	"git.grassecon.net/grassrootseconomics/sarafu-vise/config" | ||||||
| 	"git.grassecon.net/grassrootseconomics/sarafu-vise/handlers/application" | 	"git.grassecon.net/grassrootseconomics/sarafu-vise/handlers/application" | ||||||
| @ -17,8 +17,8 @@ import ( | |||||||
| ) | ) | ||||||
| 
 | 
 | ||||||
| var ( | var ( | ||||||
| 	logg      = logging.NewVanilla().WithContextKey("SessionId") | 	logg          = logging.NewVanilla().WithContextKey("SessionId") | ||||||
| 	scriptDir = path.Join("services", "registration") | 	scriptDir     = path.Join("services", "registration") | ||||||
| 	menuSeparator = ": " | 	menuSeparator = ": " | ||||||
| ) | ) | ||||||
| 
 | 
 | ||||||
| @ -87,5 +87,4 @@ func main() { | |||||||
| 		fmt.Fprintf(os.Stderr, "cmd exec error: %v\n", err) | 		fmt.Fprintf(os.Stderr, "cmd exec error: %v\n", err) | ||||||
| 		os.Exit(1) | 		os.Exit(1) | ||||||
| 	} | 	} | ||||||
| 
 |  | ||||||
| } | } | ||||||
|  | |||||||
| @ -3,9 +3,11 @@ package cmd | |||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
| 	"fmt" | 	"fmt" | ||||||
|  | 	"regexp" | ||||||
| 
 | 
 | ||||||
| 	"git.defalsify.org/vise.git/logging" | 	"git.defalsify.org/vise.git/db" | ||||||
| 	"git.defalsify.org/vise.git/engine" | 	"git.defalsify.org/vise.git/engine" | ||||||
|  | 	"git.defalsify.org/vise.git/logging" | ||||||
| 	"git.defalsify.org/vise.git/persist" | 	"git.defalsify.org/vise.git/persist" | ||||||
| 	"git.defalsify.org/vise.git/resource" | 	"git.defalsify.org/vise.git/resource" | ||||||
| 	"git.defalsify.org/vise.git/state" | 	"git.defalsify.org/vise.git/state" | ||||||
| @ -16,21 +18,24 @@ import ( | |||||||
| var argc map[string]int = map[string]int{ | var argc map[string]int = map[string]int{ | ||||||
| 	"reset": 0, | 	"reset": 0, | ||||||
| 	"admin": 1, | 	"admin": 1, | ||||||
|  | 	"clone": 1, | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| var ( | var ( | ||||||
| 	logg = logging.NewVanilla().WithDomain("cmd").WithContextKey("SessionId") | 	logg             = logging.NewVanilla().WithDomain("cmd").WithContextKey("SessionId") | ||||||
|  | 	cloneTargetRegex = `^\+000` | ||||||
| ) | ) | ||||||
| 
 | 
 | ||||||
| type Cmd struct { | type Cmd struct { | ||||||
| 	sessionId  string | 	sessionId    string | ||||||
| 	conn       storage.ConnData | 	conn         storage.ConnData | ||||||
| 	flagParser *application.FlagManager | 	flagParser   *application.FlagManager | ||||||
| 	cmd        int | 	cmd          int | ||||||
| 	enable     bool | 	enable       bool | ||||||
| 	exec       func(ctx context.Context, ss storage.StorageService) error | 	param        string | ||||||
| 	engineConfig	*engine.Config | 	exec         func(ctx context.Context, ss storage.StorageService) error | ||||||
| 	st	*state.State | 	engineConfig *engine.Config | ||||||
|  | 	st           *state.State | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| func NewCmd(sessionId string, flagParser *application.FlagManager) *Cmd { | func NewCmd(sessionId string, flagParser *application.FlagManager) *Cmd { | ||||||
| @ -70,6 +75,50 @@ func (c *Cmd) engine(ctx context.Context, rs resource.Resource, pe *persist.Pers | |||||||
| 	return en, nil | 	return en, nil | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
|  | func (c *Cmd) execClone(ctx context.Context, ss storage.StorageService) error { | ||||||
|  | 	re := regexp.MustCompile(cloneTargetRegex) | ||||||
|  | 	if !re.MatchString(c.param) { | ||||||
|  | 		return fmt.Errorf("Clone sessionId must match target: %s", c.param) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	pe, err := ss.GetPersister(ctx) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return fmt.Errorf("get persister error: %v", err) | ||||||
|  | 	} | ||||||
|  | 	err = pe.Load(c.engineConfig.SessionId) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return fmt.Errorf("persister load error: %v", err) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	/// TODO consider DRY with devtools/store/dump
 | ||||||
|  | 	store, err := ss.GetUserdataDb(ctx) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return fmt.Errorf("store retrieve error: %v", err) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	store.SetSession(c.engineConfig.SessionId) | ||||||
|  | 	store.SetPrefix(db.DATATYPE_USERDATA) | ||||||
|  | 	dmp, err := store.Dump(ctx, []byte("")) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return fmt.Errorf("store dump fail: %v\n", err.Error()) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	for true { | ||||||
|  | 		store.SetSession(c.engineConfig.SessionId) | ||||||
|  | 		k, v := dmp.Next(ctx) | ||||||
|  | 		if k == nil { | ||||||
|  | 			break | ||||||
|  | 		} | ||||||
|  | 		store.SetSession(c.param) | ||||||
|  | 		err = store.Put(ctx, k, v) | ||||||
|  | 		if err != nil { | ||||||
|  | 			return fmt.Errorf("user data store clone failed on key: %x", k) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	return pe.Save(c.param) | ||||||
|  | } | ||||||
|  | 
 | ||||||
| func (c *Cmd) execReset(ctx context.Context, ss storage.StorageService) error { | func (c *Cmd) execReset(ctx context.Context, ss storage.StorageService) error { | ||||||
| 	pe, err := ss.GetPersister(ctx) | 	pe, err := ss.GetPersister(ctx) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| @ -157,6 +206,16 @@ func (c *Cmd) parseCmdReset(cmd string, param string, more []string) (bool, erro | |||||||
| 	return false, nil | 	return false, nil | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
|  | func (c *Cmd) parseCmdClone(cmd string, param string, more []string) (bool, error) { | ||||||
|  | 	if cmd == "clone" { | ||||||
|  | 		c.enable = false | ||||||
|  | 		c.param = param | ||||||
|  | 		c.exec = c.execClone | ||||||
|  | 		return true, nil | ||||||
|  | 	} | ||||||
|  | 	return false, nil | ||||||
|  | } | ||||||
|  | 
 | ||||||
| func (c *Cmd) Parse(args []string) error { | func (c *Cmd) Parse(args []string) error { | ||||||
| 	var param string | 	var param string | ||||||
| 	if len(args) < 1 { | 	if len(args) < 1 { | ||||||
| @ -169,7 +228,7 @@ func (c *Cmd) Parse(args []string) error { | |||||||
| 		return fmt.Errorf("invalid command: %v", cmd) | 		return fmt.Errorf("invalid command: %v", cmd) | ||||||
| 	} | 	} | ||||||
| 	if n > 0 { | 	if n > 0 { | ||||||
| 		if len(args) < n + 1 { | 		if len(args) < n+1 { | ||||||
| 			return fmt.Errorf("Wrong number of arguments, need: %d", n) | 			return fmt.Errorf("Wrong number of arguments, need: %d", n) | ||||||
| 		} | 		} | ||||||
| 		param = args[1] | 		param = args[1] | ||||||
| @ -192,6 +251,13 @@ func (c *Cmd) Parse(args []string) error { | |||||||
| 		return nil | 		return nil | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
|  | 	r, err = c.parseCmdClone(cmd, param, args) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	if r { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
| 
 | 
 | ||||||
| 	return fmt.Errorf("unknown subcommand: %s", cmd) | 	return fmt.Errorf("unknown subcommand: %s", cmd) | ||||||
| } | } | ||||||
|  | |||||||
| @ -10,9 +10,9 @@ import ( | |||||||
| func TestInsertOrShift(t *testing.T) { | func TestInsertOrShift(t *testing.T) { | ||||||
| 	tests := []struct { | 	tests := []struct { | ||||||
| 		name     string | 		name     string | ||||||
| 		profile Profile | 		profile  Profile | ||||||
| 		index   int | 		index    int | ||||||
| 		value   string | 		value    string | ||||||
| 		expected []string | 		expected []string | ||||||
| 	}{ | 	}{ | ||||||
| 		{ | 		{ | ||||||
|  | |||||||
| @ -181,8 +181,8 @@ func (s *SshRunner) GetEngine(sessionId string) (engine.Engine, func(), error) { | |||||||
| 	accountService := services.New(ctx, menuStorageService) | 	accountService := services.New(ctx, menuStorageService) | ||||||
| 	_, err = lhs.GetHandler(accountService) | 	_, err = lhs.GetHandler(accountService) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 	       fmt.Fprintf(os.Stderr, "get accounts service handler: %v\n", err) | 		fmt.Fprintf(os.Stderr, "get accounts service handler: %v\n", err) | ||||||
| 	       os.Exit(1) | 		os.Exit(1) | ||||||
| 	} | 	} | ||||||
| 	en := lhs.GetEngine(lhs.Cfg, rs, pe) | 	en := lhs.GetEngine(lhs.Cfg, rs, pe) | ||||||
| 	closer := func() { | 	closer := func() { | ||||||
|  | |||||||
		Loading…
	
		Reference in New Issue
	
	Block a user