35 lines
886 B
Go
35 lines
886 B
Go
package clientcmd
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"net"
|
|
"os"
|
|
)
|
|
|
|
// GetServerInfo queries the supplied server name for information about the server
|
|
func GetServerInfo(conn *net.Conn) error {
|
|
for { // Server should send 'Connected\n' to client after connect, then client can ask for server details
|
|
message, err := bufio.NewReader(*conn).ReadString('\n')
|
|
if err != nil {
|
|
return fmt.Errorf("error reading from connection: %s", err)
|
|
}
|
|
fmt.Printf("Message from server: %s", message)
|
|
switch message {
|
|
case "Connected\n":
|
|
fmt.Println("Server has acknowledged connection, asking for server details")
|
|
bytesSent, err := fmt.Fprintf(*conn, "Details\n")
|
|
if err != nil {
|
|
fmt.Println("Error sending message to server! ", err)
|
|
os.Exit(0)
|
|
}
|
|
if bytesSent > 0 {
|
|
fmt.Println("Message Sent")
|
|
}
|
|
default:
|
|
fmt.Println("Details: ", message)
|
|
os.Exit(0)
|
|
}
|
|
}
|
|
}
|