forked from Elara6331/itd
		
	Initial Commit
This commit is contained in:
		
							
								
								
									
										87
									
								
								cmd/itctl/cmd/address.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										87
									
								
								cmd/itctl/cmd/address.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,87 @@ | ||||
| /* | ||||
|  *	itd uses bluetooth low energy to communicate with InfiniTime devices | ||||
|  *	Copyright (C) 2021 Arsen Musayelyan | ||||
|  * | ||||
|  *	This program is free software: you can redistribute it and/or modify | ||||
|  *	it under the terms of the GNU General Public License as published by | ||||
|  *	the Free Software Foundation, either version 3 of the License, or | ||||
|  *	(at your option) any later version. | ||||
|  * | ||||
|  *	This program is distributed in the hope that it will be useful, | ||||
|  *	but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||
|  *	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | ||||
|  *	GNU General Public License for more details. | ||||
|  * | ||||
|  *	You should have received a copy of the GNU General Public License | ||||
|  *	along with this program.  If not, see <http://www.gnu.org/licenses/>. | ||||
|  */ | ||||
|  | ||||
| package cmd | ||||
|  | ||||
| import ( | ||||
| 	"bufio" | ||||
| 	"encoding/json" | ||||
| 	"fmt" | ||||
| 	"net" | ||||
|  | ||||
| 	"github.com/rs/zerolog/log" | ||||
| 	"github.com/spf13/cobra" | ||||
| 	"go.arsenm.dev/itd/internal/types" | ||||
| ) | ||||
|  | ||||
| // addressCmd represents the address command | ||||
| var addressCmd = &cobra.Command{ | ||||
| 	Use:     "address", | ||||
| 	Aliases: []string{"addr"}, | ||||
| 	Short:   "Get InfiniTime's bluetooth address", | ||||
| 	Run: func(cmd *cobra.Command, args []string) { | ||||
| 		// Connect to itd UNIX socket | ||||
| 		conn, err := net.Dial("unix", SockPath) | ||||
| 		if err != nil { | ||||
| 			log.Fatal().Err(err).Msg("Error dialing socket. Is itd running?") | ||||
| 		} | ||||
| 		defer conn.Close() | ||||
|  | ||||
| 		// Encode request into connection | ||||
| 		err = json.NewEncoder(conn).Encode(types.Request{ | ||||
| 			Type: ReqTypeBtAddress, | ||||
| 		}) | ||||
| 		if err != nil { | ||||
| 			log.Fatal().Err(err).Msg("Error making request") | ||||
| 		} | ||||
|  | ||||
| 		// Read one line from connection | ||||
| 		line, _, err := bufio.NewReader(conn).ReadLine() | ||||
| 		if err != nil { | ||||
| 			log.Fatal().Err(err).Msg("Error reading line from connection") | ||||
| 		} | ||||
|  | ||||
| 		var res types.Response | ||||
| 		// Decode line into response | ||||
| 		err = json.Unmarshal(line, &res) | ||||
| 		if err != nil { | ||||
| 			log.Fatal().Err(err).Msg("Error decoding JSON data") | ||||
| 		} | ||||
|  | ||||
| 		if res.Error { | ||||
| 			log.Fatal().Msg(res.Message) | ||||
| 		} | ||||
|  | ||||
| 		// Print returned value | ||||
| 		fmt.Println(res.Value) | ||||
| 	}, | ||||
| } | ||||
|  | ||||
| func init() { | ||||
| 	getCmd.AddCommand(addressCmd) | ||||
|  | ||||
| 	// Here you will define your flags and configuration settings. | ||||
|  | ||||
| 	// Cobra supports Persistent Flags which will work for this command | ||||
| 	// and all subcommands, e.g.: | ||||
| 	// addressCmd.PersistentFlags().String("foo", "", "A help for foo") | ||||
|  | ||||
| 	// Cobra supports local flags which will only run when this command | ||||
| 	// is called directly, e.g.: | ||||
| 	// addressCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle") | ||||
| } | ||||
							
								
								
									
										77
									
								
								cmd/itctl/cmd/battery.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										77
									
								
								cmd/itctl/cmd/battery.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,77 @@ | ||||
| /* | ||||
|  *	itd uses bluetooth low energy to communicate with InfiniTime devices | ||||
|  *	Copyright (C) 2021 Arsen Musayelyan | ||||
|  * | ||||
|  *	This program is free software: you can redistribute it and/or modify | ||||
|  *	it under the terms of the GNU General Public License as published by | ||||
|  *	the Free Software Foundation, either version 3 of the License, or | ||||
|  *	(at your option) any later version. | ||||
|  * | ||||
|  *	This program is distributed in the hope that it will be useful, | ||||
|  *	but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||
|  *	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | ||||
|  *	GNU General Public License for more details. | ||||
|  * | ||||
|  *	You should have received a copy of the GNU General Public License | ||||
|  *	along with this program.  If not, see <http://www.gnu.org/licenses/>. | ||||
|  */ | ||||
|  | ||||
| package cmd | ||||
|  | ||||
| import ( | ||||
| 	"bufio" | ||||
| 	"encoding/json" | ||||
| 	"fmt" | ||||
| 	"net" | ||||
|  | ||||
| 	"github.com/rs/zerolog/log" | ||||
| 	"github.com/spf13/cobra" | ||||
| 	"go.arsenm.dev/itd/internal/types" | ||||
| ) | ||||
|  | ||||
| // batteryCmd represents the batt command | ||||
| var batteryCmd = &cobra.Command{ | ||||
| 	Use:     "battery", | ||||
| 	Aliases: []string{"batt"}, | ||||
| 	Short:   "Get battery level from InfiniTime", | ||||
| 	Run: func(cmd *cobra.Command, args []string) { | ||||
| 		// Connect to itd UNIX socket | ||||
| 		conn, err := net.Dial("unix", SockPath) | ||||
| 		if err != nil { | ||||
| 			log.Fatal().Err(err).Msg("Error dialing socket. Is itd running?") | ||||
| 		} | ||||
| 		defer conn.Close() | ||||
|  | ||||
| 		// Encode request into connection | ||||
| 		err = json.NewEncoder(conn).Encode(types.Request{ | ||||
| 			Type: ReqTypeBattLevel, | ||||
| 		}) | ||||
| 		if err != nil { | ||||
| 			log.Fatal().Err(err).Msg("Error making request") | ||||
| 		} | ||||
|  | ||||
| 		// Read one line from connection | ||||
| 		line, _, err := bufio.NewReader(conn).ReadLine() | ||||
| 		if err != nil { | ||||
| 			log.Fatal().Err(err).Msg("Error reading line from connection") | ||||
| 		} | ||||
|  | ||||
| 		var res types.Response | ||||
| 		// Deocde line into response | ||||
| 		err = json.Unmarshal(line, &res) | ||||
| 		if err != nil { | ||||
| 			log.Fatal().Err(err).Msg("Error decoding JSON data") | ||||
| 		} | ||||
|  | ||||
| 		if res.Error { | ||||
| 			log.Fatal().Msg(res.Message) | ||||
| 		} | ||||
|  | ||||
| 		// Print returned percentage | ||||
| 		fmt.Printf("%d%%\n", int(res.Value.(float64))) | ||||
| 	}, | ||||
| } | ||||
|  | ||||
| func init() { | ||||
| 	getCmd.AddCommand(batteryCmd) | ||||
| } | ||||
							
								
								
									
										41
									
								
								cmd/itctl/cmd/common.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										41
									
								
								cmd/itctl/cmd/common.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,41 @@ | ||||
| /* | ||||
|  *	itd uses bluetooth low energy to communicate with InfiniTime devices | ||||
|  *	Copyright (C) 2021 Arsen Musayelyan | ||||
|  * | ||||
|  *	This program is free software: you can redistribute it and/or modify | ||||
|  *	it under the terms of the GNU General Public License as published by | ||||
|  *	the Free Software Foundation, either version 3 of the License, or | ||||
|  *	(at your option) any later version. | ||||
|  * | ||||
|  *	This program is distributed in the hope that it will be useful, | ||||
|  *	but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||
|  *	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | ||||
|  *	GNU General Public License for more details. | ||||
|  * | ||||
|  *	You should have received a copy of the GNU General Public License | ||||
|  *	along with this program.  If not, see <http://www.gnu.org/licenses/>. | ||||
|  */ | ||||
|  | ||||
| package cmd | ||||
|  | ||||
| const SockPath = "/tmp/itd/socket" | ||||
|  | ||||
| const ( | ||||
| 	ReqTypeHeartRate = "hrt" | ||||
| 	ReqTypeBattLevel = "battlvl" | ||||
| 	ReqTypeFwVersion = "fwver" | ||||
| 	ReqTypeFwUpgrade = "fwupg" | ||||
| 	ReqTypeBtAddress = "btaddr" | ||||
| 	ReqTypeNotify    = "notify" | ||||
| 	ReqTypeSetTime   = "settime" | ||||
| ) | ||||
|  | ||||
| const ( | ||||
| 	UpgradeTypeArchive = iota | ||||
| 	UpgradeTypeFiles | ||||
| ) | ||||
|  | ||||
| type DFUProgress struct { | ||||
| 	Received int64 `mapstructure:"recvd"` | ||||
| 	Total    int64 `mapstructure:"total"` | ||||
| } | ||||
							
								
								
									
										34
									
								
								cmd/itctl/cmd/firmware.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										34
									
								
								cmd/itctl/cmd/firmware.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,34 @@ | ||||
| /* | ||||
|  *	itd uses bluetooth low energy to communicate with InfiniTime devices | ||||
|  *	Copyright (C) 2021 Arsen Musayelyan | ||||
|  * | ||||
|  *	This program is free software: you can redistribute it and/or modify | ||||
|  *	it under the terms of the GNU General Public License as published by | ||||
|  *	the Free Software Foundation, either version 3 of the License, or | ||||
|  *	(at your option) any later version. | ||||
|  * | ||||
|  *	This program is distributed in the hope that it will be useful, | ||||
|  *	but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||
|  *	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | ||||
|  *	GNU General Public License for more details. | ||||
|  * | ||||
|  *	You should have received a copy of the GNU General Public License | ||||
|  *	along with this program.  If not, see <http://www.gnu.org/licenses/>. | ||||
|  */ | ||||
|  | ||||
| package cmd | ||||
|  | ||||
| import ( | ||||
| 	"github.com/spf13/cobra" | ||||
| ) | ||||
|  | ||||
| // firmwareCmd represents the firmware command | ||||
| var firmwareCmd = &cobra.Command{ | ||||
| 	Use:     "firmware", | ||||
| 	Short:   "Manage InfiniTime firmware", | ||||
| 	Aliases: []string{"fw"}, | ||||
| } | ||||
|  | ||||
| func init() { | ||||
| 	rootCmd.AddCommand(firmwareCmd) | ||||
| } | ||||
							
								
								
									
										33
									
								
								cmd/itctl/cmd/get.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										33
									
								
								cmd/itctl/cmd/get.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,33 @@ | ||||
| /* | ||||
|  *	itd uses bluetooth low energy to communicate with InfiniTime devices | ||||
|  *	Copyright (C) 2021 Arsen Musayelyan | ||||
|  * | ||||
|  *	This program is free software: you can redistribute it and/or modify | ||||
|  *	it under the terms of the GNU General Public License as published by | ||||
|  *	the Free Software Foundation, either version 3 of the License, or | ||||
|  *	(at your option) any later version. | ||||
|  * | ||||
|  *	This program is distributed in the hope that it will be useful, | ||||
|  *	but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||
|  *	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | ||||
|  *	GNU General Public License for more details. | ||||
|  * | ||||
|  *	You should have received a copy of the GNU General Public License | ||||
|  *	along with this program.  If not, see <http://www.gnu.org/licenses/>. | ||||
|  */ | ||||
|  | ||||
| package cmd | ||||
|  | ||||
| import ( | ||||
| 	"github.com/spf13/cobra" | ||||
| ) | ||||
|  | ||||
| // getCmd represents the get command | ||||
| var getCmd = &cobra.Command{ | ||||
| 	Use:   "get", | ||||
| 	Short: "Get information from InfiniTime", | ||||
| } | ||||
|  | ||||
| func init() { | ||||
| 	rootCmd.AddCommand(getCmd) | ||||
| } | ||||
							
								
								
									
										76
									
								
								cmd/itctl/cmd/heart.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										76
									
								
								cmd/itctl/cmd/heart.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,76 @@ | ||||
| /* | ||||
|  *	itd uses bluetooth low energy to communicate with InfiniTime devices | ||||
|  *	Copyright (C) 2021 Arsen Musayelyan | ||||
|  * | ||||
|  *	This program is free software: you can redistribute it and/or modify | ||||
|  *	it under the terms of the GNU General Public License as published by | ||||
|  *	the Free Software Foundation, either version 3 of the License, or | ||||
|  *	(at your option) any later version. | ||||
|  * | ||||
|  *	This program is distributed in the hope that it will be useful, | ||||
|  *	but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||
|  *	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | ||||
|  *	GNU General Public License for more details. | ||||
|  * | ||||
|  *	You should have received a copy of the GNU General Public License | ||||
|  *	along with this program.  If not, see <http://www.gnu.org/licenses/>. | ||||
|  */ | ||||
|  | ||||
| package cmd | ||||
|  | ||||
| import ( | ||||
| 	"bufio" | ||||
| 	"encoding/json" | ||||
| 	"fmt" | ||||
| 	"net" | ||||
|  | ||||
| 	"github.com/rs/zerolog/log" | ||||
| 	"github.com/spf13/cobra" | ||||
| 	"go.arsenm.dev/itd/internal/types" | ||||
| ) | ||||
|  | ||||
| // heartCmd represents the heart command | ||||
| var heartCmd = &cobra.Command{ | ||||
| 	Use:   "heart", | ||||
| 	Short: "Get heart rate from InfiniTime", | ||||
| 	Run: func(cmd *cobra.Command, args []string) { | ||||
| 		// Connect to itd UNIX socket | ||||
| 		conn, err := net.Dial("unix", SockPath) | ||||
| 		if err != nil { | ||||
| 			log.Fatal().Err(err).Msg("Error dialing socket. Is itd running?") | ||||
| 		} | ||||
| 		defer conn.Close() | ||||
|  | ||||
| 		// Encode request into connection | ||||
| 		err = json.NewEncoder(conn).Encode(types.Request{ | ||||
| 			Type: ReqTypeHeartRate, | ||||
| 		}) | ||||
| 		if err != nil { | ||||
| 			log.Fatal().Err(err).Msg("Error making request") | ||||
| 		} | ||||
|  | ||||
| 		// Read one line from connection | ||||
| 		line, _, err := bufio.NewReader(conn).ReadLine() | ||||
| 		if err != nil { | ||||
| 			log.Fatal().Err(err).Msg("Error reading line from connection") | ||||
| 		} | ||||
|  | ||||
| 		var res types.Response | ||||
| 		// Decode line into response | ||||
| 		err = json.Unmarshal(line, &res) | ||||
| 		if err != nil { | ||||
| 			log.Fatal().Err(err).Msg("Error decoding JSON data") | ||||
| 		} | ||||
|  | ||||
| 		if res.Error { | ||||
| 			log.Fatal().Msg(res.Message) | ||||
| 		} | ||||
|  | ||||
| 		// Print returned BPM | ||||
| 		fmt.Printf("%d BPM\n", int(res.Value.(float64))) | ||||
| 	}, | ||||
| } | ||||
|  | ||||
| func init() { | ||||
| 	getCmd.AddCommand(heartCmd) | ||||
| } | ||||
							
								
								
									
										82
									
								
								cmd/itctl/cmd/notify.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										82
									
								
								cmd/itctl/cmd/notify.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,82 @@ | ||||
| /* | ||||
|  *	itd uses bluetooth low energy to communicate with InfiniTime devices | ||||
|  *	Copyright (C) 2021 Arsen Musayelyan | ||||
|  * | ||||
|  *	This program is free software: you can redistribute it and/or modify | ||||
|  *	it under the terms of the GNU General Public License as published by | ||||
|  *	the Free Software Foundation, either version 3 of the License, or | ||||
|  *	(at your option) any later version. | ||||
|  * | ||||
|  *	This program is distributed in the hope that it will be useful, | ||||
|  *	but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||
|  *	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | ||||
|  *	GNU General Public License for more details. | ||||
|  * | ||||
|  *	You should have received a copy of the GNU General Public License | ||||
|  *	along with this program.  If not, see <http://www.gnu.org/licenses/>. | ||||
|  */ | ||||
|  | ||||
| package cmd | ||||
|  | ||||
| import ( | ||||
| 	"bufio" | ||||
| 	"encoding/json" | ||||
| 	"net" | ||||
|  | ||||
| 	"github.com/rs/zerolog/log" | ||||
| 	"github.com/spf13/cobra" | ||||
| 	"go.arsenm.dev/itd/internal/types" | ||||
| ) | ||||
|  | ||||
| // notifyCmd represents the notify command | ||||
| var notifyCmd = &cobra.Command{ | ||||
| 	Use:   "notify <title> <body>", | ||||
| 	Short: "Send notification to InfiniTime", | ||||
| 	Run: func(cmd *cobra.Command, args []string) { | ||||
| 		// Ensure required arguments | ||||
| 		if len(args) != 2 { | ||||
| 			cmd.Usage() | ||||
| 			log.Fatal().Msg("Command notify requires two arguments") | ||||
| 		} | ||||
|  | ||||
| 		// Connect to itd UNIX socket | ||||
| 		conn, err := net.Dial("unix", SockPath) | ||||
| 		if err != nil { | ||||
| 			log.Fatal().Err(err).Msg("Error dialing socket. Is itd running?") | ||||
| 		} | ||||
| 		defer conn.Close() | ||||
|  | ||||
| 		// Encode request into connection | ||||
| 		err = json.NewEncoder(conn).Encode(types.Request{ | ||||
| 			Type: ReqTypeNotify, | ||||
| 			Data: types.ReqDataNotify{ | ||||
| 				Title: args[0], | ||||
| 				Body:  args[1], | ||||
| 			}, | ||||
| 		}) | ||||
| 		if err != nil { | ||||
| 			log.Fatal().Err(err).Msg("Error making request") | ||||
| 		} | ||||
|  | ||||
| 		// Read one line from connection | ||||
| 		line, _, err := bufio.NewReader(conn).ReadLine() | ||||
| 		if err != nil { | ||||
| 			log.Fatal().Err(err).Msg("Error reading line from connection") | ||||
| 		} | ||||
|  | ||||
| 		var res types.Response | ||||
| 		// Decode line into response | ||||
| 		err = json.Unmarshal(line, &res) | ||||
| 		if err != nil { | ||||
| 			log.Fatal().Err(err).Msg("Error decoding JSON data") | ||||
| 		} | ||||
|  | ||||
| 		if res.Error { | ||||
| 			log.Fatal().Msg(res.Message) | ||||
| 		} | ||||
| 	}, | ||||
| } | ||||
|  | ||||
| func init() { | ||||
| 	rootCmd.AddCommand(notifyCmd) | ||||
| } | ||||
							
								
								
									
										35
									
								
								cmd/itctl/cmd/root.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										35
									
								
								cmd/itctl/cmd/root.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,35 @@ | ||||
| /* | ||||
|  *	itd uses bluetooth low energy to communicate with InfiniTime devices | ||||
|  *	Copyright (C) 2021 Arsen Musayelyan | ||||
|  * | ||||
|  *	This program is free software: you can redistribute it and/or modify | ||||
|  *	it under the terms of the GNU General Public License as published by | ||||
|  *	the Free Software Foundation, either version 3 of the License, or | ||||
|  *	(at your option) any later version. | ||||
|  * | ||||
|  *	This program is distributed in the hope that it will be useful, | ||||
|  *	but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||
|  *	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | ||||
|  *	GNU General Public License for more details. | ||||
|  * | ||||
|  *	You should have received a copy of the GNU General Public License | ||||
|  *	along with this program.  If not, see <http://www.gnu.org/licenses/>. | ||||
|  */ | ||||
|  | ||||
| package cmd | ||||
|  | ||||
| import ( | ||||
| 	"github.com/spf13/cobra" | ||||
| ) | ||||
|  | ||||
| // rootCmd represents the base command when called without any subcommands | ||||
| var rootCmd = &cobra.Command{ | ||||
| 	Use:   "itctl", | ||||
| 	Short: "Control the itd daemon for InfiniTime smartwatches", | ||||
| } | ||||
|  | ||||
| // Execute adds all child commands to the root command and sets flags appropriately. | ||||
| // This is called by main.main(). It only needs to happen once to the rootCmd. | ||||
| func Execute() { | ||||
| 	cobra.CheckErr(rootCmd.Execute()) | ||||
| } | ||||
							
								
								
									
										33
									
								
								cmd/itctl/cmd/set.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										33
									
								
								cmd/itctl/cmd/set.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,33 @@ | ||||
| /* | ||||
|  *	itd uses bluetooth low energy to communicate with InfiniTime devices | ||||
|  *	Copyright (C) 2021 Arsen Musayelyan | ||||
|  * | ||||
|  *	This program is free software: you can redistribute it and/or modify | ||||
|  *	it under the terms of the GNU General Public License as published by | ||||
|  *	the Free Software Foundation, either version 3 of the License, or | ||||
|  *	(at your option) any later version. | ||||
|  * | ||||
|  *	This program is distributed in the hope that it will be useful, | ||||
|  *	but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||
|  *	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | ||||
|  *	GNU General Public License for more details. | ||||
|  * | ||||
|  *	You should have received a copy of the GNU General Public License | ||||
|  *	along with this program.  If not, see <http://www.gnu.org/licenses/>. | ||||
|  */ | ||||
|  | ||||
| package cmd | ||||
|  | ||||
| import ( | ||||
| 	"github.com/spf13/cobra" | ||||
| ) | ||||
|  | ||||
| // setCmd represents the set command | ||||
| var setCmd = &cobra.Command{ | ||||
| 	Use:   "set", | ||||
| 	Short: "Set information on InfiniTime", | ||||
| } | ||||
|  | ||||
| func init() { | ||||
| 	rootCmd.AddCommand(setCmd) | ||||
| } | ||||
							
								
								
									
										79
									
								
								cmd/itctl/cmd/time.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										79
									
								
								cmd/itctl/cmd/time.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,79 @@ | ||||
| /* | ||||
|  *	itd uses bluetooth low energy to communicate with InfiniTime devices | ||||
|  *	Copyright (C) 2021 Arsen Musayelyan | ||||
|  * | ||||
|  *	This program is free software: you can redistribute it and/or modify | ||||
|  *	it under the terms of the GNU General Public License as published by | ||||
|  *	the Free Software Foundation, either version 3 of the License, or | ||||
|  *	(at your option) any later version. | ||||
|  * | ||||
|  *	This program is distributed in the hope that it will be useful, | ||||
|  *	but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||
|  *	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | ||||
|  *	GNU General Public License for more details. | ||||
|  * | ||||
|  *	You should have received a copy of the GNU General Public License | ||||
|  *	along with this program.  If not, see <http://www.gnu.org/licenses/>. | ||||
|  */ | ||||
|  | ||||
| package cmd | ||||
|  | ||||
| import ( | ||||
| 	"bufio" | ||||
| 	"encoding/json" | ||||
| 	"net" | ||||
|  | ||||
| 	"github.com/rs/zerolog/log" | ||||
| 	"github.com/spf13/cobra" | ||||
| 	"go.arsenm.dev/itd/internal/types" | ||||
| ) | ||||
|  | ||||
| // timeCmd represents the time command | ||||
| var timeCmd = &cobra.Command{ | ||||
| 	Use:   `time <ISO8601|"now">`, | ||||
| 	Short: "Set InfiniTime's clock to specified time", | ||||
| 	Run: func(cmd *cobra.Command, args []string) { | ||||
| 		// Ensure required arguments | ||||
| 		if len(args) != 1 { | ||||
| 			cmd.Usage() | ||||
| 			log.Fatal().Msg("Command time requires one argument") | ||||
| 		} | ||||
|  | ||||
| 		// Connect to itd UNIX socket | ||||
| 		conn, err := net.Dial("unix", SockPath) | ||||
| 		if err != nil { | ||||
| 			log.Fatal().Err(err).Msg("Error dialing socket. Is itd running?") | ||||
| 		} | ||||
| 		defer conn.Close() | ||||
|  | ||||
| 		// Encode request into connection | ||||
| 		err = json.NewEncoder(conn).Encode(types.Request{ | ||||
| 			Type: ReqTypeSetTime, | ||||
| 			Data: args[0], | ||||
| 		}) | ||||
| 		if err != nil { | ||||
| 			log.Fatal().Err(err).Msg("Error making request") | ||||
| 		} | ||||
|  | ||||
| 		// Read one line from connetion | ||||
| 		line, _, err := bufio.NewReader(conn).ReadLine() | ||||
| 		if err != nil { | ||||
| 			log.Fatal().Err(err).Msg("Error reading line from connection") | ||||
| 		} | ||||
|  | ||||
| 		var res types.Response | ||||
| 		// Decode line into response | ||||
| 		err = json.Unmarshal(line, &res) | ||||
| 		if err != nil { | ||||
| 			log.Fatal().Err(err).Msg("Error decoding JSON data") | ||||
| 		} | ||||
|  | ||||
| 		if res.Error { | ||||
| 			log.Fatal().Msg(res.Message) | ||||
| 		} | ||||
| 	}, | ||||
| } | ||||
|  | ||||
| func init() { | ||||
| 	setCmd.AddCommand(timeCmd) | ||||
| } | ||||
							
								
								
									
										126
									
								
								cmd/itctl/cmd/upgrade.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										126
									
								
								cmd/itctl/cmd/upgrade.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,126 @@ | ||||
| /* | ||||
|  *	itd uses bluetooth low energy to communicate with InfiniTime devices | ||||
|  *	Copyright (C) 2021 Arsen Musayelyan | ||||
|  * | ||||
|  *	This program is free software: you can redistribute it and/or modify | ||||
|  *	it under the terms of the GNU General Public License as published by | ||||
|  *	the Free Software Foundation, either version 3 of the License, or | ||||
|  *	(at your option) any later version. | ||||
|  * | ||||
|  *	This program is distributed in the hope that it will be useful, | ||||
|  *	but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||
|  *	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | ||||
|  *	GNU General Public License for more details. | ||||
|  * | ||||
|  *	You should have received a copy of the GNU General Public License | ||||
|  *	along with this program.  If not, see <http://www.gnu.org/licenses/>. | ||||
|  */ | ||||
|  | ||||
| package cmd | ||||
|  | ||||
| import ( | ||||
| 	"bufio" | ||||
| 	"encoding/json" | ||||
| 	"net" | ||||
|  | ||||
| 	"github.com/cheggaaa/pb/v3" | ||||
| 	"github.com/mitchellh/mapstructure" | ||||
| 	"github.com/rs/zerolog/log" | ||||
| 	"github.com/spf13/cobra" | ||||
| 	"github.com/spf13/viper" | ||||
| 	"go.arsenm.dev/itd/internal/types" | ||||
| ) | ||||
|  | ||||
| // upgradeCmd represents the upgrade command | ||||
| var upgradeCmd = &cobra.Command{ | ||||
| 	Use:     "upgrade", | ||||
| 	Short:   "Upgrade InfiniTime firmware using files or archive", | ||||
| 	Aliases: []string{"upg"}, | ||||
| 	Run: func(cmd *cobra.Command, args []string) { | ||||
| 		// Connect to itd UNIX socket | ||||
| 		conn, err := net.Dial("unix", SockPath) | ||||
| 		if err != nil { | ||||
| 			log.Fatal().Err(err).Msg("Error dialing socket. Is itd running?") | ||||
| 		} | ||||
| 		defer conn.Close() | ||||
|  | ||||
| 		var data types.ReqDataFwUpgrade | ||||
| 		// Get relevant data struct | ||||
| 		if viper.GetString("archive") != "" { | ||||
| 			// Get archive data struct | ||||
| 			data = types.ReqDataFwUpgrade{ | ||||
| 				Type:  UpgradeTypeArchive, | ||||
| 				Files: []string{viper.GetString("archive")}, | ||||
| 			} | ||||
| 		} else if viper.GetString("initPkt") != "" && viper.GetString("firmware") != "" { | ||||
| 			// Get files data struct | ||||
| 			data = types.ReqDataFwUpgrade{ | ||||
| 				Type:  UpgradeTypeFiles, | ||||
| 				Files: []string{viper.GetString("initPkt"), viper.GetString("firmware")}, | ||||
| 			} | ||||
| 		} else { | ||||
| 			cmd.Usage() | ||||
| 			log.Fatal().Msg("Upgrade command requires either archive or init packet and firmware.") | ||||
| 		} | ||||
|  | ||||
| 		// Encode response into connection | ||||
| 		err = json.NewEncoder(conn).Encode(types.Request{ | ||||
| 			Type: ReqTypeFwUpgrade, | ||||
| 			Data: data, | ||||
| 		}) | ||||
| 		if err != nil { | ||||
| 			log.Fatal().Err(err).Msg("Error making request") | ||||
| 		} | ||||
|  | ||||
| 		// Create progress bar template | ||||
| 		barTmpl := `{{counters . }} B {{bar . "|" "-" (cycle .) " " "|"}} {{percent . }} {{rtime . "%s"}}` | ||||
| 		// Start full bar at 0 total | ||||
| 		bar := pb.ProgressBarTemplate(barTmpl).Start(0) | ||||
| 		// Create new scanner of connection | ||||
| 		scanner := bufio.NewScanner(conn) | ||||
| 		for scanner.Scan() { | ||||
| 			var res types.Response | ||||
| 			// Decode scanned line into response struct | ||||
| 			err = json.Unmarshal(scanner.Bytes(), &res) | ||||
| 			if err != nil { | ||||
| 				log.Fatal().Err(err).Msg("Error decoding JSON response") | ||||
| 			} | ||||
| 			if res.Error { | ||||
| 				log.Fatal().Msg(res.Message) | ||||
| 			} | ||||
| 			var event DFUProgress | ||||
| 			// Decode response data into progress struct | ||||
| 			err = mapstructure.Decode(res.Value, &event) | ||||
| 			if err != nil { | ||||
| 				log.Fatal().Err(err).Msg("Error decoding response data") | ||||
| 			} | ||||
| 			// If transfer finished, break | ||||
| 			if event.Received == event.Total { | ||||
| 				break | ||||
| 			} | ||||
| 			// Set total bytes in progress bar | ||||
| 			bar.SetTotal(event.Total) | ||||
| 			// Set amount of bytes received in progress bar | ||||
| 			bar.SetCurrent(event.Received) | ||||
| 		} | ||||
| 		// Finish progress bar | ||||
| 		bar.Finish() | ||||
| 		if scanner.Err() != nil { | ||||
| 			log.Fatal().Err(scanner.Err()).Msg("Error while scanning output") | ||||
| 		} | ||||
| 	}, | ||||
| } | ||||
|  | ||||
| func init() { | ||||
| 	firmwareCmd.AddCommand(upgradeCmd) | ||||
|  | ||||
| 	// Register flags | ||||
| 	upgradeCmd.Flags().StringP("archive", "a", "", "Path to firmware archive") | ||||
| 	upgradeCmd.Flags().StringP("init-pkt", "i", "", "Path to init packet (.dat file)") | ||||
| 	upgradeCmd.Flags().StringP("firmware", "f", "", "Path to firmware image (.bin file)") | ||||
|  | ||||
| 	// Bind flags to viper keys | ||||
| 	viper.BindPFlag("archive", upgradeCmd.Flags().Lookup("archive")) | ||||
| 	viper.BindPFlag("initPkt", upgradeCmd.Flags().Lookup("init-pkt")) | ||||
| 	viper.BindPFlag("firmware", upgradeCmd.Flags().Lookup("firmware")) | ||||
| } | ||||
							
								
								
									
										77
									
								
								cmd/itctl/cmd/version.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										77
									
								
								cmd/itctl/cmd/version.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,77 @@ | ||||
| /* | ||||
|  *	itd uses bluetooth low energy to communicate with InfiniTime devices | ||||
|  *	Copyright (C) 2021 Arsen Musayelyan | ||||
|  * | ||||
|  *	This program is free software: you can redistribute it and/or modify | ||||
|  *	it under the terms of the GNU General Public License as published by | ||||
|  *	the Free Software Foundation, either version 3 of the License, or | ||||
|  *	(at your option) any later version. | ||||
|  * | ||||
|  *	This program is distributed in the hope that it will be useful, | ||||
|  *	but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||
|  *	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | ||||
|  *	GNU General Public License for more details. | ||||
|  * | ||||
|  *	You should have received a copy of the GNU General Public License | ||||
|  *	along with this program.  If not, see <http://www.gnu.org/licenses/>. | ||||
|  */ | ||||
|  | ||||
| package cmd | ||||
|  | ||||
| import ( | ||||
| 	"bufio" | ||||
| 	"encoding/json" | ||||
| 	"fmt" | ||||
| 	"net" | ||||
|  | ||||
| 	"github.com/rs/zerolog/log" | ||||
| 	"github.com/spf13/cobra" | ||||
| 	"go.arsenm.dev/itd/internal/types" | ||||
| ) | ||||
|  | ||||
| // versionCmd represents the version command | ||||
| var versionCmd = &cobra.Command{ | ||||
| 	Use:     "version", | ||||
| 	Aliases: []string{"ver"}, | ||||
| 	Short:   "Get firmware version of InfiniTime", | ||||
| 	Run: func(cmd *cobra.Command, args []string) { | ||||
| 		// Connect to itd UNIX socket | ||||
| 		conn, err := net.Dial("unix", SockPath) | ||||
| 		if err != nil { | ||||
| 			log.Fatal().Err(err).Msg("Error dialing socket. Is itd running?") | ||||
| 		} | ||||
| 		defer conn.Close() | ||||
|  | ||||
| 		// Encode request into connection | ||||
| 		err = json.NewEncoder(conn).Encode(types.Request{ | ||||
| 			Type: ReqTypeFwVersion, | ||||
| 		}) | ||||
| 		if err != nil { | ||||
| 			log.Fatal().Err(err).Msg("Error making request") | ||||
| 		} | ||||
|  | ||||
| 		// Read one line from connection | ||||
| 		line, _, err := bufio.NewReader(conn).ReadLine() | ||||
| 		if err != nil { | ||||
| 			log.Fatal().Err(err).Msg("Error reading line from connection") | ||||
| 		} | ||||
|  | ||||
| 		var res types.Response | ||||
| 		// Decode line into response | ||||
| 		err = json.Unmarshal(line, &res) | ||||
| 		if err != nil { | ||||
| 			log.Fatal().Err(err).Msg("Error decoding JSON data") | ||||
| 		} | ||||
|  | ||||
| 		if res.Error { | ||||
| 			log.Fatal().Msg(res.Message) | ||||
| 		} | ||||
|  | ||||
| 		// Print returned value | ||||
| 		fmt.Println(res.Value) | ||||
| 	}, | ||||
| } | ||||
|  | ||||
| func init() { | ||||
| 	firmwareCmd.AddCommand(versionCmd) | ||||
| } | ||||
							
								
								
									
										36
									
								
								cmd/itctl/main.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										36
									
								
								cmd/itctl/main.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,36 @@ | ||||
| /* | ||||
|  *	itd uses bluetooth low energy to communicate with InfiniTime devices | ||||
|  *	Copyright (C) 2021 Arsen Musayelyan | ||||
|  * | ||||
|  *	This program is free software: you can redistribute it and/or modify | ||||
|  *	it under the terms of the GNU General Public License as published by | ||||
|  *	the Free Software Foundation, either version 3 of the License, or | ||||
|  *	(at your option) any later version. | ||||
|  * | ||||
|  *	This program is distributed in the hope that it will be useful, | ||||
|  *	but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||
|  *	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | ||||
|  *	GNU General Public License for more details. | ||||
|  * | ||||
|  *	You should have received a copy of the GNU General Public License | ||||
|  *	along with this program.  If not, see <http://www.gnu.org/licenses/>. | ||||
|  */ | ||||
|  | ||||
| package main | ||||
|  | ||||
| import ( | ||||
| 	"os" | ||||
|  | ||||
| 	"go.arsenm.dev/itd/cmd/itctl/cmd" | ||||
|  | ||||
| 	"github.com/rs/zerolog" | ||||
| 	"github.com/rs/zerolog/log" | ||||
| ) | ||||
|  | ||||
| func init() { | ||||
| 	log.Logger = log.Output(zerolog.ConsoleWriter{Out: os.Stderr}) | ||||
| } | ||||
|  | ||||
| func main() { | ||||
| 	cmd.Execute() | ||||
| } | ||||
		Reference in New Issue
	
	Block a user