No need to hardcode UnZSTDPath during compilation

[tofuproxy.git] / cmd / tofuproxy / main.go

diff --git a/cmd/tofuproxy/main.go b/cmd/tofuproxy/main.go
index babc8b7e34c34f3cfbe76d7ad9601e4d4edce9d6..ddfa4771d74b910cd82e38855300311b866a7f3c 100644 (file)
--- a/cmd/tofuproxy/main.go
+++ b/cmd/tofuproxy/main.go
@@ -29,6 +29,7 @@ import (
        "go.stargrave.org/tofuproxy/fifos"
        "go.stargrave.org/tofuproxy/rounds"
        ttls "go.stargrave.org/tofuproxy/tls"
+       "go.stargrave.org/tofuproxy/warc"
 )
 
 func main() {
@@ -42,6 +43,7 @@ func main() {
        fifosDir := flag.String("fifos", "fifos", "Directory with FIFOs")
        notai := flag.Bool("notai", false, "Do not prepend TAI64N to logs")
        warcOnly := flag.Bool("warc-only", false, "Server only WARC URIs")
+       unzstdPath := flag.String("unzstd", "cmd/zstd/unzstd", "Path to unzstd utility")
        flag.Parse()
        log.SetFlags(log.Lshortfile)
 
@@ -64,6 +66,7 @@ func main() {
        tofuproxy.CAPrv = caPrv
        tofuproxy.X509Algo = *ai
        rounds.WARCOnly = *warcOnly
+       warc.UnZSTDPath = *unzstdPath
 
        ln, err := net.Listen("tcp", *bind)
        if err != nil {