goreplay-cli package (#1148)

change package from `main -> goreplay`
this will allow importing `goreplay` as a package

Makefile

@@ -31,19 +31,19 @@ vendor:
 	go mod vendor
 
 release-bin-linux-amd64: vendor
-	docker run --platform linux/amd64 --rm -v `pwd`:$(SOURCE_PATH) -t --env GOOS=linux --env GOARCH=amd64 -i $(CONTAINER_AMD) go build -mod=vendor -o $(BIN_NAME) -tags netgo $(LDFLAGS)
+	docker run --platform linux/amd64 --rm -v `pwd`:$(SOURCE_PATH) -t --env GOOS=linux --env GOARCH=amd64 -i $(CONTAINER_AMD) go build -mod=vendor -o $(BIN_NAME) -tags netgo $(LDFLAGS) ./cmd/gor/
 
 release-bin-linux-arm64: vendor
-	docker run --platform linux/arm64 --rm -v `pwd`:$(SOURCE_PATH) -t --env GOOS=linux --env GOARCH=arm64 -i $(CONTAINER_ARM) go build -mod=vendor -o $(BIN_NAME) -tags netgo $(LDFLAGS)
+	docker run --platform linux/arm64 --rm -v `pwd`:$(SOURCE_PATH) -t --env GOOS=linux --env GOARCH=arm64 -i $(CONTAINER_ARM) go build -mod=vendor -o $(BIN_NAME) -tags netgo $(LDFLAGS) ./cmd/gor/
 
 release-bin-mac-amd64: vendor
-	GOOS=darwin go build -mod=vendor -o $(BIN_NAME) $(MAC_LDFLAGS)
+	GOOS=darwin go build -mod=vendor -o $(BIN_NAME) $(MAC_LDFLAGS) ./cmd/gor/
 
 release-bin-mac-arm64: vendor
 	GOOS=darwin GOARCH=arm64 CGO_ENABLED=1 go build -mod=vendor -o $(BIN_NAME) $(MAC_LDFLAGS)
 
 release-bin-windows: vendor
-	docker run -it --rm -v `pwd`:$(SOURCE_PATH) -w $(SOURCE_PATH) -e CGO_ENABLED=1 docker.elastic.co/beats-dev/golang-crossbuild:1.19.2-main --build-cmd "make VERSION=$(VERSION) build" -p "windows/amd64"
+	docker run -it --rm -v `pwd`:$(SOURCE_PATH) -w $(SOURCE_PATH) -e CGO_ENABLED=1 docker.elastic.co/beats-dev/golang-crossbuild:1.19.2-main --build-cmd "make VERSION=$(VERSION) build" -p "windows/amd64" ./cmd/gor/
 	mv $(BIN_NAME) "$(BIN_NAME).exe"
 
 release-linux-amd64: dist release-bin-linux-amd64

gor.go → cmd/gor/gor.go

@@ -6,6 +6,7 @@ import (
 	"expvar"
 	"flag"
 	"fmt"
+	"github.com/buger/goreplay"
 	"log"
 	"net/http"
 	"net/http/httputil"
@@ -71,23 +72,23 @@ func main() {
 	}
 
 	args := os.Args[1:]
-	var plugins *InOutPlugins
+	var plugins *goreplay.InOutPlugins
 	if len(args) > 0 && args[0] == "file-server" {
 		if len(args) != 2 {
 			log.Fatal("You should specify port and IP (optional) for the file server. Example: `gor file-server :80`")
 		}
 		dir, _ := os.Getwd()
 
-		Debug(0, "Started example file server for current directory on address ", args[1])
+		goreplay.Debug(0, "Started example file server for current directory on address ", args[1])
 
 		log.Fatal(http.ListenAndServe(args[1], loggingMiddleware(args[1], http.FileServer(http.Dir(dir)))))
 	} else {
 		flag.Parse()
-		checkSettings()
-		plugins = NewPlugins()
+		goreplay.CheckSettings()
+		plugins = goreplay.NewPlugins()
 	}
 
-	log.Printf("[PPID %d and PID %d] Version:%s\n", os.Getppid(), os.Getpid(), VERSION)
+	log.Printf("[PPID %d and PID %d] Version:%s\n", os.Getppid(), os.Getpid(), goreplay.VERSION)
 
 	if len(plugins.Inputs) == 0 || len(plugins.Outputs) == 0 {
 		log.Fatal("Required at least 1 input and 1 output")
@@ -101,20 +102,20 @@ func main() {
 		profileCPU(*cpuprofile)
 	}
 
-	if Settings.Pprof != "" {
+	if goreplay.Settings.Pprof != "" {
 		go func() {
-			log.Println(http.ListenAndServe(Settings.Pprof, nil))
+			log.Println(http.ListenAndServe(goreplay.Settings.Pprof, nil))
 		}()
 	}
 
 	closeCh := make(chan int)
-	emitter := NewEmitter()
-	go emitter.Start(plugins, Settings.Middleware)
-	if Settings.ExitAfter > 0 {
-		log.Printf("Running gor for a duration of %s\n", Settings.ExitAfter)
+	emitter := goreplay.NewEmitter()
+	go emitter.Start(plugins, goreplay.Settings.Middleware)
+	if goreplay.Settings.ExitAfter > 0 {
+		log.Printf("Running gor for a duration of %s\n", goreplay.Settings.ExitAfter)
 
-		time.AfterFunc(Settings.ExitAfter, func() {
-			log.Printf("gor run timeout %s\n", Settings.ExitAfter)
+		time.AfterFunc(goreplay.Settings.ExitAfter, func() {
+			log.Printf("gor run timeout %s\n", goreplay.Settings.ExitAfter)
 			close(closeCh)
 		})
 	}

elasticsearch.go

@@ -1,14 +1,13 @@
-package main
+package goreplay
 
 import (
 	"encoding/json"
+	"github.com/buger/goreplay/proto"
 	"log"
 	"net/url"
 	"strings"
 	"time"
 
-	"github.com/buger/goreplay/proto"
-
 	elastigo "github.com/mattbaird/elastigo/lib"
 )
 

elasticsearch_test.go

@@ -1,4 +1,4 @@
-package main
+package goreplay
 
 import (
 	"testing"

emitter.go

@@ -1,13 +1,13 @@
-package main
+package goreplay
 
 import (
 	"fmt"
+	"github.com/buger/goreplay/internal/byteutils"
 	"hash/fnv"
 	"io"
 	"log"
 	"sync"
 
-	"github.com/buger/goreplay/byteutils"
 	"github.com/coocood/freecache"
 )
 

emitter_test.go

@@ -1,4 +1,4 @@
-package main
+package goreplay
 
 import (
 	"fmt"

examples/middleware/token_modifier.go

@@ -24,9 +24,8 @@ import (
 	"bytes"
 	"encoding/hex"
 	"fmt"
-	"os"
-
 	"github.com/buger/goreplay/proto"
+	"os"
 )
 
 // requestID -> originalToken

gor_stat.go

@@ -1,4 +1,4 @@
-package main
+package goreplay
 
 import (
 	"runtime"

http_modifier.go

@@ -1,12 +1,11 @@
-package main
+package goreplay
 
 import (
 	"bytes"
 	"encoding/base64"
+	"github.com/buger/goreplay/proto"
 	"hash/fnv"
 	"strings"
-
-	"github.com/buger/goreplay/proto"
 )
 
 type HTTPModifier struct {

http_modifier_settings.go

@@ -1,4 +1,4 @@
-package main
+package goreplay
 
 import (
 	"errors"
@@ -24,9 +24,7 @@ type HTTPModifierConfig struct {
 	Methods                HTTPMethods                `json:"http-allow-method"`
 }
 
-//
 // Handling of --http-allow-header, --http-disallow-header options
-//
 type headerFilter struct {
 	name   []byte
 	regexp *regexp.Regexp
@@ -56,9 +54,7 @@ func (h *HTTPHeaderFilters) Set(value string) error {
 	return nil
 }
 
-//
 // Handling of --http-basic-auth-filter option
-//
 type basicAuthFilter struct {
 	regexp *regexp.Regexp
 }
@@ -82,9 +78,7 @@ func (h *HTTPHeaderBasicAuthFilters) Set(value string) error {
 	return nil
 }
 
-//
 // Handling of --http-allow-header-hash and --http-allow-param-hash options
-//
 type hashFilter struct {
 	name    []byte
 	percent uint32
@@ -129,9 +123,7 @@ func (h *HTTPHashFilters) Set(value string) error {
 	return nil
 }
 
-//
 // Handling of --http-set-header option
-//
 type httpHeader struct {
 	Name  string
 	Value string
@@ -160,9 +152,7 @@ func (h *HTTPHeaders) Set(value string) error {
 	return nil
 }
 
-//
 // Handling of --http-set-param option
-//
 type httpParam struct {
 	Name  []byte
 	Value []byte
@@ -208,9 +198,7 @@ func (h *HTTPMethods) Set(value string) error {
 	return nil
 }
 
-//
 // Handling of --http-rewrite-url option
-//
 type urlRewrite struct {
 	src    *regexp.Regexp
 	target []byte
@@ -237,9 +225,7 @@ func (r *URLRewriteMap) Set(value string) error {
 	return nil
 }
 
-//
 // Handling of --http-rewrite-header option
-//
 type headerRewrite struct {
 	header []byte
 	src    *regexp.Regexp
@@ -275,9 +261,7 @@ func (r *HeaderRewriteMap) Set(value string) error {
 	return nil
 }
 
-//
 // Handling of --http-allow-url option
-//
 type urlRegexp struct {
 	regexp *regexp.Regexp
 }

http_modifier_settings_test.go

@@ -1,4 +1,4 @@
-package main
+package goreplay
 
 import (
 	"testing"

http_modifier_test.go

@@ -1,10 +1,9 @@
-package main
+package goreplay
 
 import (
 	"bytes"
-	"testing"
-
 	"github.com/buger/goreplay/proto"
+	"testing"
 )
 
 func TestHTTPModifierWithoutConfig(t *testing.T) {

http_prettifier.go

@@ -1,14 +1,13 @@
-package main
+package goreplay
 
 import (
 	"bytes"
 	"compress/gzip"
 	"fmt"
+	"github.com/buger/goreplay/proto"
 	"io/ioutil"
 	"net/http/httputil"
 	"strconv"
-
-	"github.com/buger/goreplay/proto"
 )
 
 func prettifyHTTP(p []byte) []byte {

http_prettifier_test.go

@@ -1,12 +1,11 @@
-package main
+package goreplay
 
 import (
 	"bytes"
 	"compress/gzip"
+	"github.com/buger/goreplay/proto"
 	"strconv"
 	"testing"
-
-	"github.com/buger/goreplay/proto"
 )
 
 func TestHTTPPrettifierGzip(t *testing.T) {

input_dummy.go

@@ -1,4 +1,4 @@
-package main
+package goreplay
 
 import (
 	"time"

input_file.go

@@ -1,4 +1,4 @@
-package main
+package goreplay
 
 import (
 	"bufio"

input_file_test.go

@@ -1,4 +1,4 @@
-package main
+package goreplay
 
 import (
 	"bytes"

input_http.go

@@ -1,4 +1,4 @@
-package main
+package goreplay
 
 import (
 	"log"

input_http_test.go

@@ -1,4 +1,4 @@
-package main
+package goreplay
 
 import (
 	"bytes"

input_kafka.go

@@ -1,4 +1,4 @@
-package main
+package goreplay
 
 import (
 	"encoding/json"

input_kafka_test.go

@@ -1,4 +1,4 @@
-package main
+package goreplay
 
 import (
 	"testing"

input_raw.go

@@ -1,17 +1,16 @@
-package main
+package goreplay
 
 import (
 	"context"
 	"fmt"
+	"github.com/buger/goreplay/internal/capture"
+	"github.com/buger/goreplay/internal/tcp"
+	"github.com/buger/goreplay/proto"
 	"log"
 	"net"
 	"strconv"
 	"strings"
 	"sync"
-
-	"github.com/buger/goreplay/capture"
-	"github.com/buger/goreplay/proto"
-	"github.com/buger/goreplay/tcp"
 )
 
 // RAWInputConfig represents configuration that can be applied on raw input

input_raw_test.go

@@ -1,7 +1,10 @@
-package main
+package goreplay
 
 import (
 	"bytes"
+	"github.com/buger/goreplay/internal/capture"
+	"github.com/buger/goreplay/internal/tcp"
+	"github.com/buger/goreplay/proto"
 	"io/ioutil"
 	"net"
 	"net/http"
@@ -13,10 +16,6 @@ import (
 	"sync/atomic"
 	"testing"
 	"time"
-
-	"github.com/buger/goreplay/capture"
-	"github.com/buger/goreplay/proto"
-	"github.com/buger/goreplay/tcp"
 )
 
 const testRawExpire = time.Millisecond * 200

input_tcp.go

@@ -1,4 +1,4 @@
-package main
+package goreplay
 
 import (
 	"bufio"