Andreas/gotty
1
0
Fork 0
mirror of https://github.com/yudai/gotty.git synced 2026-01-04 16:48:50 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

add api-prefix flag for nginx path redirect

Browse source
This commit is contained in:
ZhaoYuh 2022-05-18 16:13:24 +08:00
parent a080c85cbc
commit c5ef9697fd
3 changed files with 8 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
server/options.go
View file

@ -10,6 +10,7 @@ type Options struct {
PermitWrite bool `hcl:"permit_write" flagName:"permit-write" flagSName:"w" flagDescribe:"Permit clients to write to the TTY (BE CAREFUL)" default:"false"`
EnableBasicAuth bool `hcl:"enable_basic_auth" default:"false"`
Credential string `hcl:"credential" flagName:"credential" flagSName:"c" flagDescribe:"Credential for Basic Authentication (ex: user:pass, default disabled)" default:""`
ApiPrefix string `hcl:"api_prefix" flagName:"api-prefix" flagDescribe:"Add a string to the URL" default:""`
EnableRandomUrl bool `hcl:"enable_random_url" flagName:"random-url" flagSName:"r" flagDescribe:"Add a random string to the URL" default:"false"`
RandomUrlLength int `hcl:"random_url_length" flagName:"random-url-length" flagDescribe:"Random URL length" default:"8"`
EnableTLS bool `hcl:"enable_tls" flagName:"tls" flagSName:"t" flagDescribe:"Enable TLS/SSL" default:"false"`

5
server/server.go
View file

@ -4,12 +4,14 @@ import (
"context"
"crypto/tls"
"crypto/x509"
"fmt"
"html/template"
"io/ioutil"
"log"
"net"
"net/http"
"regexp"
"strings"
noesctmpl "text/template"
"time"
@ -96,6 +98,9 @@ func (server *Server) Run(ctx context.Context, options ...RunOption) error {
counter := newCounter(time.Duration(server.options.Timeout) * time.Second)
path := "/"
if strings.TrimSpace(server.options.ApiPrefix) != "" {
path = fmt.Sprintf(`/%s/`, strings.TrimSpace(server.options.ApiPrefix))
}
if server.options.EnableRandomUrl {
path = "/" + randomstring.Generate(server.options.RandomUrlLength) + "/"
}