main.go

package main

import (
	"io/ioutil"
	"log"
	"os"
	"strings"

	"github.com/Omegaice/go-xmodem/ymodem"
	"github.com/spf13/cobra"
	"github.com/tarm/serial"
)

var Port, Message, Wait string

func main() {
	log.SetFlags(log.LstdFlags | log.Lshortfile)

	var cmdSend = &cobra.Command{
		Use:   "send [port]",
		Short: "Send file",
		Long:  ``,
		Run: func(cmd *cobra.Command, args []string) {
			// Open connection
			connection, err := serial.OpenPort(&serial.Config{Name: Port, Baud: 115200})
			if err != nil {
				log.Fatalln(err)
			}

			// Send initial message
			if len(Message) > 0 {
				if _, err := connection.Write([]byte(Message + "\r\n")); err != nil {
					log.Println(err)
				}
			}

			// Wait for message
			if len(Wait) > 0 {
				var result string
				buffer := make([]byte, 64)
				for {
					n, err := connection.Read(buffer)
					if err != nil {
						log.Fatalln(err)
					}
					if n == 0 {
						break
					}
					result += string(buffer[0:n])
					if strings.Contains(result, Wait) {
						break
					}
				}
			}

			// Open file
			fIn, err := os.Open(args[0])
			if err != nil {
				log.Fatalln(err)
			}

			data, err := ioutil.ReadAll(fIn)
			if err != nil {
				log.Fatalln(err)
			}

			fIn.Close()

			// Send file
			if err := ymodem.ModemSend(connection, data, args[0]); err != nil {
				log.Fatalln(err)
			}

			log.Println(args[0], "sent successful")
		},
	}
	cmdSend.Flags().StringVarP(&Port, "port", "p", "", "serial port to connect to")
	cmdSend.Flags().StringVarP(&Message, "message", "m", "", "message to initiate data transfer")
	cmdSend.Flags().StringVarP(&Wait, "wait", "w", "", "message to wait before initiating data transfer")

	var cmdReceive = &cobra.Command{
		Use:   "receive [port]",
		Short: "Receive file",
		Long:  ``,
		Run: func(cmd *cobra.Command, args []string) {
			// Open connection
			connection, err := serial.OpenPort(&serial.Config{Name: Port, Baud: 115200})
			if err != nil {
				log.Fatalln(err)
			}

			// Send initial message
			if len(Message) > 0 {
				if _, err := connection.Write([]byte(Message + "\r\n")); err != nil {
					log.Println(err)
				}
			}

			// Wait for message
			if len(Wait) > 0 {
				var result string
				buffer := make([]byte, 64)
				for {
					n, err := connection.Read(buffer)
					if err != nil {
						log.Fatalln(err)
					}
					if n == 0 {
						break
					}
					result += string(buffer[0:n])
					if strings.Contains(result, Wait) {
						break
					}
				}
			}

			// Receive file
			filename, data, err := ymodem.ModemReceive(connection)
			if err != nil {
				log.Fatalln(err)
			}

			// Write file
			fOut, err := os.Create(filename)
			if err != nil {
				log.Fatalln(err)
			}
			fOut.Write(data)
			fOut.Close()

			log.Println(filename, "write successful")
		},
	}
	cmdReceive.Flags().StringVarP(&Port, "port", "p", "", "serial port to connect to")
	cmdReceive.Flags().StringVarP(&Message, "message", "m", "", "message to initiate data transfer")
	cmdReceive.Flags().StringVarP(&Wait, "wait", "w", "", "message to wait before initiating data transfer")

	var Root = &cobra.Command{
		Use:   "go-xmodem",
		Short: "",
		Long:  ``,
		Run:   func(cmd *cobra.Command, args []string) {},
	}

	Root.AddCommand(cmdSend)
	Root.AddCommand(cmdReceive)
	Root.Execute()
}