cmd/tofuproxy/main.go | 3 +++ cmd/warc-extract/main.go | 2 ++ diff --git a/cmd/tofuproxy/main.go b/cmd/tofuproxy/main.go index fa1a38f2bf69e1b35e10f292b3f02a6c387742e3f519f851162fe7c372601cde..287a0f6ce01a5e759384ce2a4e34c0d6ce5101b3fc4beb9d0ec5c3050917406e 100644 --- a/cmd/tofuproxy/main.go +++ b/cmd/tofuproxy/main.go @@ -29,6 +29,7 @@ "go.stargrave.org/tofuproxy" "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 @@ dnsSrv := flag.String("dns", "[::1]:53", "DNS server") 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 @@ tofuproxy.CACert = caCert tofuproxy.CAPrv = caPrv tofuproxy.X509Algo = *ai rounds.WARCOnly = *warcOnly + warc.UnZSTDPath = *unzstdPath ln, err := net.Listen("tcp", *bind) if err != nil { diff --git a/cmd/warc-extract/main.go b/cmd/warc-extract/main.go index 94a70316c1fb91a955173d49f1b928343d83e90d4c07b9efd64b46cbf11e33c0..a94e42e4f871beb47338b7eac78832887488621f73beace46f0fbb3335859fb9 100644 --- a/cmd/warc-extract/main.go +++ b/cmd/warc-extract/main.go @@ -36,8 +36,10 @@ uri := flag.String("uri", "", "URI to extract, if specified") hdr := flag.Bool("hdr", false, "Also extract WARC's header") idx := flag.Bool("idx", false, "Save WARC indices") recompress := flag.Bool("for-enzstd", false, "Output for enzstd utility") + unzstdPath := flag.String("unzstd", "cmd/zstd/unzstd", "Path to unzstd utility") flag.Parse() log.SetFlags(log.Lshortfile) + warc.UnZSTDPath = *unzstdPath if *recompress { var hdr bytes.Buffer