From 2ff99a284b70068c815cce72917af426fe6ddbc2 Mon Sep 17 00:00:00 2001 From: raul Date: Fri, 29 Mar 2024 09:04:20 +0100 Subject: [PATCH] Adding files to project --- cmd/root.go | 51 +++++++++++++++++++++++++ cmd/server.go | 100 ++++++++++++++++++++++++++++++++++++++++++++++++++ go.mod | 9 +++++ go.sum | 10 +++++ main.go | 11 ++++++ 5 files changed, 181 insertions(+) create mode 100644 cmd/root.go create mode 100644 cmd/server.go create mode 100644 go.mod create mode 100644 go.sum create mode 100644 main.go diff --git a/cmd/root.go b/cmd/root.go new file mode 100644 index 0000000..b856ca3 --- /dev/null +++ b/cmd/root.go @@ -0,0 +1,51 @@ +/* +Copyright © 2024 Raul + +*/ +package cmd + +import ( + "os" + + "github.com/spf13/cobra" +) + + + +// rootCmd represents the base command when called without any subcommands +var rootCmd = &cobra.Command{ + Use: "mini-chat", + Short: "A brief description of your application", + Long: `A longer description that spans multiple lines and likely contains +examples and usage of using your application. For example: + +Cobra is a CLI library for Go that empowers applications. +This application is a tool to generate the needed files +to quickly create a Cobra application.`, + // Uncomment the following line if your bare application + // has an action associated with it: + // Run: func(cmd *cobra.Command, args []string) { }, +} + +// 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() { + err := rootCmd.Execute() + if err != nil { + os.Exit(1) + } +} + +func init() { + // Here you will define your flags and configuration settings. + // Cobra supports persistent flags, which, if defined here, + // will be global for your application. + + // rootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", "config file (default is $HOME/.mini-chat.yaml)") + + // Cobra also supports local flags, which will only run + // when this action is called directly. + rootCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle") +} + + diff --git a/cmd/server.go b/cmd/server.go new file mode 100644 index 0000000..e26e971 --- /dev/null +++ b/cmd/server.go @@ -0,0 +1,100 @@ +/* +Copyright © 2024 Raul +*/ +package cmd + +import ( + "bufio" + "fmt" + "net" + "strings" + "time" + + //"os" + + "github.com/spf13/cobra" +) + +// serverCmd represents the server command +var serverCmd = &cobra.Command{ + Use: "server", + Short: "A brief description of your command", + Long: `A longer description that spans multiple lines and likely contains examples +and usage of using your command. For example: + +Cobra is a CLI library for Go that empowers applications. +This application is a tool to generate the needed files +to quickly create a Cobra application.`, + Run: func(cmd *cobra.Command, args []string) { + server(cmd) + }, +} + +func init() { + rootCmd.AddCommand(serverCmd) + + // Here you will define your flags and configuration settings. + + // Cobra supports Persistent Flags which will work for this command + // and all subcommands, e.g.: + // serverCmd.PersistentFlags().String("foo", "", "A help for foo") + + // Cobra supports local flags which will only run when this command + // is called directly, e.g.: + // serverCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle") + serverCmd.Flags().String("port", "1302", "Port to listen on") +} + +func server(cmd *cobra.Command) { + var lport string + port, _ := cmd.Flags().GetString("port") + if port != "" { + lport = port + } else { + lport = "1302" + } + + ln, err := net.Listen("tcp", ":"+lport) + cobra.CheckErr(err) + fmt.Printf("Listening on port %v...\n", lport) + for { + conn, err := ln.Accept() + cobra.CheckErr(err) + go handleConn(conn) + } + +} + +type chatter struct { + Username string + IP string +} + +func handleConn(conn net.Conn) { + + defer conn.Close() + IP := getIP(conn) + t := time.Now() + date := fmt.Sprintf("%d-%02d-%02dT%02d:%02d:%02d", t.Year(), t.Month(), t.Day(), t.Hour(), t.Minute(), t.Second()) + final_message := fmt.Sprintf("[%v] Received connection from %v!\n", date, IP) + fmt.Print(final_message) + conn.Write([]byte("What's your name?\nName: ")) + name, err := bufio.NewReader(conn).ReadString('\n') + cobra.CheckErr(err) + var NewUser = new(chatter) + NewUser.Username = strings.TrimRight(name, "\n") + NewUser.IP = IP + for { + message, err := bufio.NewReader(conn).ReadString('\n') + cobra.CheckErr(err) + fmt.Print("{" + NewUser.IP + "} " + NewUser.Username + ": " + message) + conn.Write([]byte(message)) + } +} + +func getIP(conn net.Conn) (IP string) { + if addr, ok := conn.RemoteAddr().(*net.TCPAddr); ok { + IP = fmt.Sprintf("%v", addr) + } + return IP +} diff --git a/go.mod b/go.mod new file mode 100644 index 0000000..cebe74b --- /dev/null +++ b/go.mod @@ -0,0 +1,9 @@ +module mini-chat + +go 1.22.1 + +require ( + github.com/inconshreveable/mousetrap v1.1.0 // indirect + github.com/spf13/cobra v1.8.0 // indirect + github.com/spf13/pflag v1.0.5 // indirect +) diff --git a/go.sum b/go.sum new file mode 100644 index 0000000..d0e8c2c --- /dev/null +++ b/go.sum @@ -0,0 +1,10 @@ +github.com/cpuguy83/go-md2man/v2 v2.0.3/go.mod h1:tgQtvFlXSQOSOSIRvRPT7W67SCa46tRHOmNcaadrF8o= +github.com/inconshreveable/mousetrap v1.1.0 h1:wN+x4NVGpMsO7ErUn/mUI3vEoE6Jt13X2s0bqwp9tc8= +github.com/inconshreveable/mousetrap v1.1.0/go.mod h1:vpF70FUmC8bwa3OWnCshd2FqLfsEA9PFc4w1p2J65bw= +github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM= +github.com/spf13/cobra v1.8.0 h1:7aJaZx1B85qltLMc546zn58BxxfZdR/W22ej9CFoEf0= +github.com/spf13/cobra v1.8.0/go.mod h1:WXLWApfZ71AjXPya3WOlMsY9yMs7YeiHhFVlvLyhcho= +github.com/spf13/pflag v1.0.5 h1:iy+VFUOCP1a+8yFto/drg2CJ5u0yRoB7fZw3DKv/JXA= +github.com/spf13/pflag v1.0.5/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg= +gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= diff --git a/main.go b/main.go new file mode 100644 index 0000000..f109379 --- /dev/null +++ b/main.go @@ -0,0 +1,11 @@ +/* +Copyright © 2024 Raul + +*/ +package main + +import "mini-chat/cmd" + +func main() { + cmd.Execute() +}