Golang rewrite

packages/client/hut.txt → ascii/hut.txt

@@ -1,3 +1,4 @@
+
                            (   )
                           (    )
                            (    )

cmd/streamhut/main.go

@@ -0,0 +1,51 @@
+package main
+
+import (
+	"fmt"
+	"os"
+	"os/signal"
+	"syscall"
+
+	"github.com/streamhut/streamhut/pkg/db"
+	"github.com/streamhut/streamhut/pkg/db/sqlite3db"
+	"github.com/streamhut/streamhut/pkg/httpserver"
+	"github.com/streamhut/streamhut/pkg/tcpserver"
+	"github.com/streamhut/streamhut/pkg/wsserver"
+)
+
+func main() {
+	dbtype := "sqlite3"
+	var db db.DB
+	if dbtype == "sqlite3" {
+		db = sqlite3db.NewDB(&sqlite3db.Config{
+			DBPath: "./data/sqlite3.db",
+		})
+	}
+	ws := wsserver.NewWS(&wsserver.Config{
+		DB: db,
+	})
+	tcpServer := tcpserver.NewServer(&tcpserver.Config{
+		WS:   ws,
+		Port: 1337,
+		DB:   db,
+	})
+
+	go tcpServer.Start()
+
+	server := httpserver.NewServer(&httpserver.Config{
+		Port: 8080,
+		WS:   ws,
+	})
+
+	var gracefulStop = make(chan os.Signal)
+	signal.Notify(gracefulStop, syscall.SIGTERM)
+	signal.Notify(gracefulStop, syscall.SIGINT)
+	go func() {
+		sig := <-gracefulStop
+		fmt.Printf("caught sig: %+v\nshutting down...", sig)
+		db.Close()
+		os.Exit(0)
+	}()
+
+	server.Start()
+}

common/byteutil/byteutil.go

@@ -0,0 +1,19 @@
+package byteutil
+
+var arrSize = 100
+
+// BufferWithMime ...
+func BufferWithMime(buf []byte, mime string) []byte {
+	m := make([]byte, arrSize)
+	copy(m, []byte(mime))
+	buf = append(m, buf...)
+	return buf
+}
+
+// DecoupleBufferWithMime ...
+func DecoupleBufferWithMime(buf []byte) ([]byte, string) {
+	mime := buf[:arrSize]
+	b := buf[arrSize:]
+
+	return b, string(mime)
+}

common/byteutil/byteutil_test.go

@@ -0,0 +1,24 @@
+package byteutil
+
+import (
+	"fmt"
+	"testing"
+)
+
+func TestBufferWithMime(t *testing.T) {
+	b := []byte("hello")
+	m := "text/plain"
+	bm := BufferWithMime(b, m)
+
+	fmt.Println(string(bm))
+}
+
+func TestDecoupleBufferWithMime(t *testing.T) {
+	b := []byte("hello")
+	m := "text/plain"
+	bm := BufferWithMime(b, m)
+	buf, mime := DecoupleBufferWithMime(bm)
+
+	fmt.Println(string(buf))
+	fmt.Println(mime)
+}

common/stringutil/stringutil.go

@@ -0,0 +1,22 @@
+package stringutil
+
+import (
+	"math/rand"
+	"time"
+)
+
+//var letterRunes = []rune("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ")
+var letterRunes = []rune("abcdefghijklmnopqrstuvwxyz")
+
+// RandStringRunes ...
+func RandStringRunes(n int) string {
+	b := make([]rune, n)
+	for i := range b {
+		b[i] = letterRunes[rand.Intn(len(letterRunes))]
+	}
+	return string(b)
+}
+
+func init() {
+	rand.Seed(time.Now().UnixNano())
+}

common/stringutil/stringutil_test.go

@@ -0,0 +1,11 @@
+package stringutil
+
+import (
+	"fmt"
+	"testing"
+)
+
+func TestRandStringRunes(t *testing.T) {
+	str := RandStringRunes(5)
+	fmt.Println(str)
+}

common/util/util.go

@@ -0,0 +1,11 @@
+package util
+
+// ValidChannelName ...
+func ValidChannelName(channel string) bool {
+	return true
+}
+
+// NormalizeChannelName ...
+func NormalizeChannelName(channel string) string {
+	return channel
+}