Exemplo n.º 1
0
func NewGame() *Game {
	g := &Game{ishell.NewShell(), NewPlayer(), NewPlaces()}

	g.Setup()

	return g
}
Exemplo n.º 2
0
func main() {
	shell := ishell.NewShell()

	// display info
	shell.Println("Sample Interactive Shell")

	// handle exit
	shell.Register("exit", func(cmd string, args []string) (string, error) {
		shell.Println("Do you want to do more ? y/n:")
		line, _ := shell.ReadLine()
		if strings.ToLower(line) == "y" {
			doLogin(shell)
		}
		shell.Stop()
		return "bye!", nil
	})

	// register a function for "greet" command.
	shell.Register("greet", func(cmd string, args []string) (string, error) {
		name := "Stranger"
		if len(args) > 0 {
			name = strings.Join(args, " ")
		}
		return "Hello " + name, nil
	})

	// start shell
	shell.Start()
}
Exemplo n.º 3
0
func main() {
	shell := ishell.NewShell()

	// display info
	shell.Println("Sample Interactive Shell")

	// handle login
	shell.Register("login", func(args ...string) (string, error) {
		doLogin(shell)
		return "", nil
	})

	// handle "greet".
	shell.Register("greet", func(args ...string) (string, error) {
		name := "Stranger"
		if len(args) > 0 {
			name = strings.Join(args, " ")
		}
		return "Hello " + name, nil
	})

	// read multiple lines with "multi" command
	shell.Register("multi", func(args ...string) (string, error) {
		shell.Println("Input multiple lines and end with semicolon ';'.")
		lines := shell.ReadMultiLines(";")
		shell.Println("Done reading. You wrote:")
		return lines, nil
	})

	// start shell
	shell.Start()
}
Exemplo n.º 4
0
func main() {
	shell := ishell.NewShell()
	connection := connect()
	shell.Println("Connected")
	registerCommands(shell, connection)
	shell.Start()
}
Exemplo n.º 5
0
func NewWorld() *World {
	// https://www.youtube.com/watch?v=GXBDvzz_jC8
	w := &World{AllGrids: NewGridList(), Density: defaultDensity,
		Shell: ishell.NewShell()}
	w.initialize()
	return w
}
Exemplo n.º 6
0
func enter_shell(m2m *xrm2m.M2MClient) {
	const done = "<done>"
	shell := ishell.NewShell()
	shell.Println("Welcome to the telemetry management demo")

	shell.Register("collectors", func(args ...string) (string, error) {
		var cols []Collector
		for _, arg := range args {
			host, port, _ := net.SplitHostPort(arg)
			portnum, _ := strconv.Atoi(port)
			cols = append(cols, Collector{Address: host, Port: portnum})
		}
		update_collectors(m2m, &cols)
		return "Updated collectors", nil
	})

	shell.Register("policy", func(args ...string) (string, error) {
		if len(args) != 1 {
			return "Please specify a policy name", nil
		}
		err := set_policy(m2m, args[0])
		return "Updated policy", err
	})
	shell.Start()
}
Exemplo n.º 7
0
func init() {
	flag.IntVar(&flagvar, "flagname", 1234, "help message for flagname")
	shell = ishell.NewShell()
}
Exemplo n.º 8
0
func main() {
	// create new shell.
	// by default, new shell includes 'exit', 'help' and 'clear' commands.
	shell := ishell.NewShell()

	// display welcome info.
	shell.Println("iKettle 2.0 shell")

	var settings settings
	var loadSettings = func() {
		b, err := ioutil.ReadFile("./.ikettle")
		if err == nil {
			err = json.Unmarshal(b, &settings)
			if err != nil {
				shell.Println("Error reading .ikettle settings file:", err)
			}
			return
		}
		shell.Println("Error reading .ikettle settings file:", err)
	}
	var saveSettings = func() {
		b, err := json.Marshal(&settings)
		if err != nil {
			shell.Println("Error saving .ikettle settings file:", err)
			return
		}
		err = ioutil.WriteFile("./.ikettle", b, 0660)
		if err != nil {
			shell.Println("Error saving .ikettle settings file:", err)
		}
	}
	loadSettings()

	// register a function for "connect" command.
	shell.Register("connect", func(args ...string) (string, error) {
		if remoteConn != nil {
			return "", errors.New("already connected - use 'disconnect' first")
		}

		if len(args) > 0 {
			settings.RemoteAddr = args[0]
			saveSettings()
		}

		if len(settings.RemoteAddr) == 0 {
			return "", errors.New("connect address missing, e.g. 'connect 192.168.0.10'")
		}

		var err error
		remoteConn, err = net.DialTimeout("tcp", settings.RemoteAddr+":2081", time.Duration(5)*time.Second)

		if err != nil {
			remoteConn = nil
			return "", err
		}

		go func() {
			rdr := bufio.NewReader(remoteConn)
			var buf []byte
			for {
				b, err := rdr.ReadByte()
				if err != nil {
					shell.Println("Disconnected")
					if remoteConn != nil {
						go func(rc net.Conn) {
							defer func() {
								recover()
							}()
							rc.Close()
						}(remoteConn)
					}
					return
				}
				if b != endMessage {
					buf = append(buf, b)
					continue
				}

				switch buf[0] {
				case commandSentAck:
					shell.Println("OK!")
					buf = []byte{}
				case autoDacKettleResponse:
					if len(buf) != 3 {
						shell.Println("invalid length for kettle dac response")
						buf = []byte{}
						continue
					}

					shell.Println("DAC response:", buf)
					buf = []byte{}
				case wifiNetworkResponse:
					// SSID,-db}SSID,-db}
					shell.Println("wifi response:", buf)
					s := string(buf)
					p := strings.Split(s, "}")
					for _, wn := range p {
						if len(wn) == 0 {
							buf = []byte{}
							continue
						}
						p2 := strings.SplitN(wn, ",", 2)
						shell.Println("=>", fmt.Sprintf("%s (%s dBm)", p2[0], p2[1]))
					}
					buf = []byte{}
				case deviceInfoResponse:
					if len(buf) != 3 {
						shell.Println("invalid length for device info response")
						buf = []byte{}
						continue
					}
					deviceType := int(buf[1])
					deviceSDK := int(buf[2])
					var deviceName string
					switch deviceType {
					case deviceKettle:
						deviceName = "iKettle 2.0"
					case deviceCoffee:
						deviceName = "Coffee"
					default:
						deviceName = "Unknown"
					}
					shell.Println("Device info:")
					shell.Println("=>", fmt.Sprintf("Type = %s [%d]", deviceName, deviceType))
					shell.Println("=>", fmt.Sprintf("SDK = %d", deviceSDK))
					buf = []byte{}
				case kettleStatus:
					if len(buf) != 6 {
						shell.Println("invalid length for kettle status update:", buf)
						shell.Println("`" + string(buf) + "`")
						buf = []byte{}
						continue
					}

					flag := buf[1]
					var status string
					//var kettleReady, boilingInProgress, keepWarmInProgress, cycleFinished, babyCooling bool

					switch flag {
					case 0:
						status = "READY"
						//kettleReady = true
					case 1:
						status = "BOILING"
						//boilingInProgress = true
					case 2:
						status = "KEEP WARM"
						//keepWarmInProgress = true
					case 3:
						status = "FINISHED"
						//cycleFinished = true
					case 4:
						status = "COOLING"
						//babyCooling = true
					default:
						shell.Println("Invalid flag in kettle status:", flag)
					}

					temp := buf[2]
					waterlevel1 := int(buf[3])
					waterlevel2 := int(buf[4])

					wlevel := (waterlevel1 << 8) + waterlevel2
					wlevel2 := wlevel - settings.OffBaseWeight

					shell.Println("Status =>", status, fmt.Sprintf("(%d)", flag))
					shell.Println("Temperature =>", temp)

					var wlstatus string
					if wlevel2 < 130 {
						wlstatus = "EMPTY"
					} else if wlevel2 < 163 {
						wlstatus = "LOW"
					} else if wlevel2 < 203 {
						wlstatus = "HALF"
					} else if wlevel2 < 270 {
						wlstatus = "FULL"
					} else {
						wlstatus = "OVERFILLED"
					}

					shell.Println("Water level =>", wlevel, fmt.Sprintf("(%s)", wlstatus))

					if pendingDAC {
						settings.OffBaseWeight = int((waterlevel1 << 8) + waterlevel2)
						saveSettings()

						_, err := remoteConn.Write([]byte{autoDacKettle, endMessage})
						if err != nil {
							shell.Println("Error requesting AutoDAC:", err)
							return
						}

						pendingDAC = false
						shell.Println("Wait for DAC response")
					}
					buf = []byte{}
				default:
					shell.Println("Got command:", buf)
					shell.Println("ASCII: `" + string(buf) + "`")
					buf = []byte{}
				}
			}
		}()

		return "Connected to " + settings.RemoteAddr, nil
	})

	shell.Register("disconnect", func(args ...string) (string, error) {
		if remoteConn == nil {
			return "", errors.New("not connected")
		}

		err := remoteConn.Close()
		remoteConn = nil
		if err != nil {
			return "", err
		}

		return "Disconnected from " + settings.RemoteAddr, nil
	})

	shell.Register("calibrate", func(args ...string) (string, error) {
		if remoteConn == nil {
			return "", errors.New("not connected")
		}

		shell.Println("Take kettle off base and press ENTER")
		_ = shell.ReadLine()
		pendingDAC = true

		return "Wait for status response", nil
	})

	shell.Register("on", func(args ...string) (string, error) {
		if remoteConn == nil {
			return "", errors.New("not connected")
		}

		temperature := 100

		if len(args) > 0 {
			tmp := args[0]
			v, err := strconv.Atoi(tmp)
			if err != nil {
				return "", errors.New("temperature must be valid number")
			}
			temperature = v
		}

		_, err := remoteConn.Write([]byte{turnOnKettle, byte(temperature), 0, endMessage})
		if err != nil {
			return "", err
		}

		return "switched on", nil
	})

	shell.Register("off", func(args ...string) (string, error) {
		if remoteConn == nil {
			return "", errors.New("not connected")
		}

		_, err := remoteConn.Write([]byte{turnOffKettle, endMessage})
		if err != nil {
			return "", err
		}

		return "switched off", nil
	})

	shell.Register("cmd", func(args ...string) (string, error) {
		if remoteConn == nil {
			return "", errors.New("not connected")
		}

		var buf []byte
		for _, v := range args {
			i, e := strconv.Atoi(v)
			if e != nil {
				return "", fmt.Errorf("arg must be numeric byte value: %s", v)
			}
			buf = append(buf, byte(i))
		}

		if buf[len(buf)-1] != 126 {
			buf = append(buf, byte(126))
		}

		_, err := remoteConn.Write(buf)
		if err != nil {
			return "", err
		}

		return fmt.Sprintf("command sent: %v", buf), nil
	})

	shell.Register("wifi", func(args ...string) (string, error) {
		if remoteConn == nil {
			return "", errors.New("not connected")
		}

		_, err := remoteConn.Write([]byte{listWifiNetworks, endMessage})
		if err != nil {
			return "", err
		}

		return "wifi network list requested", nil
	})

	shell.Register("info", func(args ...string) (string, error) {
		if remoteConn == nil {
			return "", errors.New("not connected")
		}

		_, err := remoteConn.Write([]byte{deviceInfo, endMessage})
		if err != nil {
			return "", err
		}

		return "wifi network list requested", nil
	})

	shell.Register("setup", func(args ...string) (string, error) {
		if remoteConn == nil {
			return "", errors.New("not connected: switch to iKettle network, then `connect 192.168.4.1`")
		}

		shell.ShowPrompt(false)
		defer shell.ShowPrompt(true)

		shell.Print("Enter network SSID:")
		ssid := shell.ReadLine()

		shell.Print("Enter network password:"******"Bytes: %v", bSsid))
		_, err := remoteConn.Write(bSsid)
		if err != nil {
			return "", fmt.Errorf("Error sending SSID command: %s", err)
		}

		bPasswd := []byte{configureWifiPassword}
		bPasswd = append(bPasswd, []byte(passwd)...)
		bPasswd = append(bPasswd, endMessage)
		_, err = remoteConn.Write(bPasswd)
		shell.Println(fmt.Sprintf("Bytes: %v", bPasswd))
		if err != nil {
			return "", fmt.Errorf("Error sending password command: %s", err)
		}

		bEnd := []byte{configureWifi, endMessage}
		shell.Println(fmt.Sprintf("Bytes: %v", bEnd))
		_, err = remoteConn.Write(bEnd)
		if err != nil {
			return "", fmt.Errorf("Error sending setup command: %s", err)
		}

		return "After ACK, switch to normal wifi network and `connect [kettle-ip]`", nil
	})

	// start shell
	shell.Start()
}
Exemplo n.º 9
0
func enter_shell(m2m *xrm2m.M2MClient) {
	const done = "<done>"
	shell := ishell.NewShell()
	shell.Println("Welcome to the XR M2M CLI")

	// CLI transition functions

	shell.Register("cli_exec", func(args ...string) (string, error) {
		return m2m.CliExec(strings.Join(args, " ")), m2m.Error
	})
	shell.Register("cli_get", func(args ...string) (string, error) {
		return pretty_print(m2m.CliGet(strings.Join(args, " ")), m2m.Error)
	})
	shell.Register("cli_set", func(args ...string) (string, error) {
		m2m.CliSet(strings.Join(args, " "))
		return done, m2m.Error
	})
	shell.Register("write_file", func(args ...string) (string, error) {
		shell.Println("Input multiple lines and end with semicolon ';'.")
		data := shell.ReadMultiLines(";")
		data = strings.TrimSuffix(data, ";") // @@@ bug in ishell
		m2m.WriteFile(args[0], []byte(data))
		return done, m2m.Error
	})

	// Basic schema ops

	shell.Register("get", func(args ...string) (string, error) {
		return pretty_print(m2m.Get(strings.Join(args, " ")), m2m.Error)
	})

	shell.Register("get_children", func(args ...string) (string, error) {
		return pretty_print(m2m.GetChildren(strings.Join(args, " ")), m2m.Error)
	})

	shell.Register("set", func(args ...string) (string, error) {
		m2m.Set(args[0], args[1])
		return done, m2m.Error
	})

	shell.Register("delete", func(args ...string) (string, error) {
		m2m.Delete(strings.Join(args, " "))
		return done, m2m.Error
	})

	shell.Register("replace", func(args ...string) (string, error) {
		m2m.Replace(strings.Join(args, " "))
		return done, m2m.Error
	})

	// Commit operations. Both "comment" and "label" are optional

	shell.Register("commit", func(args ...string) (string, error) {
		comment, label := optional(args)
		return m2m.Commit(comment, label), m2m.Error
	})

	shell.Register("commit_replace", func(args ...string) (string, error) {
		comment, label := optional(args)
		return m2m.CommitReplace(comment, label), m2m.Error
	})

	shell.Register("discard_changes", func(args ...string) (string, error) {
		m2m.DiscardChanges()
		return done, m2m.Error
	})

	shell.Register("get_changes", func(args ...string) (string, error) {
		return pretty_print(m2m.GetChanges(), m2m.Error)
	})

	// Schema

	shell.Register("get_schema", func(args ...string) (string, error) {
		path := args[0]
		var fields string
		if len(args) == 2 {
			fields = args[1]
		}
		return pretty_print(m2m.GetSchema(path, fields), m2m.Error)
	})

	shell.Register("get_version", func(args ...string) (string, error) {
		return pretty_print(m2m.GetVersion(), m2m.Error)
	})

	shell.Start()
}