Add Cobra and start codebase refactoring

This commit is contained in:
raul 2024-04-22 08:05:34 +02:00
parent 6a75e7cd41
commit 3bfe744b8d
6 changed files with 162 additions and 28 deletions

0
chat-tests/LICENSE Normal file
View File

47
chat-tests/cmd/root.go Normal file
View File

@ -0,0 +1,47 @@
/*
Copyright © 2024 Raul <raul@bulgariu.xyz>
*/
package cmd
import (
"os"
"github.com/spf13/cobra"
)
// rootCmd represents the base command when called without any subcommands
var rootCmd = &cobra.Command{
Use: "chat-tests",
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/.chat-tests.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")
}

93
chat-tests/cmd/server.go Normal file
View File

@ -0,0 +1,93 @@
/*
Copyright © 2024 Raul <raul@bulgariu.xyz>
*/
package cmd
import (
"fmt"
"github.com/spf13/cobra"
"log"
"net"
)
// 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) {
setParameters(cmd)
Server()
},
}
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")
serverCmd.PersistentFlags().String("port", "1302", "port to use for listening")
// 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")
}
var (
port string = "1302"
)
type Updater interface {
UpdateUser()
}
type User struct {
Username string
IP string
}
func setParameters(cmd *cobra.Command) {
}
func (u User) UpdateUser(usr string, ip string) User {
newU := new(User)
newU.Username = usr
newU.IP = ip
return *newU
}
func Server() {
ln, err := net.Listen("tcp", ":"+port)
checkErr(err)
fmt.Printf("Listening on port %v...\n", port)
for {
conn, err := ln.Accept()
checkErr(err)
go handleConn(conn)
}
}
func getUserInput() {
}
func handleConn(conn net.Conn) {
fmt.Println("Received connection")
fmt.Fprintln(conn, "Hello buddy!")
}
func checkErr(err error) {
if err != nil {
log.Fatalf("Error: %v\n", err)
}
}

View File

@ -1,3 +1,9 @@
module chat-tests module chat-tests
go 1.22.2 go 1.22.2
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
)

10
chat-tests/go.sum Normal file
View File

@ -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=

View File

@ -1,33 +1,11 @@
/*
Copyright © 2024 Raul <raul@bulgariu.xyz>
*/
package main package main
import ( import "chat-tests/cmd"
"fmt"
"log"
"net"
)
const (
port string = "1302"
)
func main() { func main() {
ln, err := net.Listen("tcp", ":"+port) cmd.Execute()
checkErr(err)
fmt.Printf("Listening on port %v...\n", port)
for {
conn, err := ln.Accept()
checkErr(err)
go handleFunc(conn)
}
}
func handleFunc(conn net.Conn) {
fmt.Println("Received connection")
fmt.Fprintln(conn, "Hello buddy!")
}
func checkErr(err error) {
if err != nil {
log.Fatalf("Error: %v\n", err)
}
} }