pass alt redirect and alt condition and alt redirect
This commit is contained in:
parent
1f2b1dd8e5
commit
6d386bf591
@ -9,3 +9,4 @@ services:
|
|||||||
redis:
|
redis:
|
||||||
image: redis:7
|
image: redis:7
|
||||||
volumes: [ "./redis:/data" ]
|
volumes: [ "./redis:/data" ]
|
||||||
|
ports: [ "6379:6379" ]
|
||||||
|
31
main.go
31
main.go
@ -72,17 +72,27 @@ func indexHandler(w http.ResponseWriter, r *http.Request) {
|
|||||||
|
|
||||||
command := r.PostForm.Get("Command")
|
command := r.PostForm.Get("Command")
|
||||||
shortlink := r.PostForm.Get("Shortlink")
|
shortlink := r.PostForm.Get("Shortlink")
|
||||||
value := r.PostForm.Get("Value")
|
redirect := r.PostForm.Get("Redirect")
|
||||||
fmt.Printf("command: %v, shortlink: %v, value: %v\n", command, shortlink, value)
|
alt_redirect := r.PostForm.Get("AltRedirect")
|
||||||
fmt.Println(shortlink)
|
alt_condition := r.PostForm.Get("AltCondition")
|
||||||
fmt.Println(value)
|
fmt.Printf("command: %v, shortlink: %v, redirect: %v, alt_redirect: %v, alt_condition: %v\n",
|
||||||
|
command,
|
||||||
|
shortlink,
|
||||||
|
redirect,
|
||||||
|
alt_redirect,
|
||||||
|
alt_condition,
|
||||||
|
)
|
||||||
|
|
||||||
|
formstring := command+":"+shortlink+":"+redirect+":"+alt_condition+":"+alt_redirect
|
||||||
|
|
||||||
|
fmt.Println("formstring: " + formstring)
|
||||||
|
|
||||||
signature := r.Header.Get("Signature")
|
signature := r.Header.Get("Signature")
|
||||||
calculatedSignature := fmt.Sprintf(
|
calculatedSignature := fmt.Sprintf(
|
||||||
"SUS-SIGNATURE-%v",
|
"SUS-SIGNATURE-%v",
|
||||||
getSha256HMACSignature(
|
getSha256HMACSignature(
|
||||||
[]byte(SECRET),
|
[]byte(SECRET),
|
||||||
command+":"+shortlink+":"+value,
|
command+":"+shortlink+":"+redirect+":"+alt_condition+":"+alt_redirect,
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -100,7 +110,7 @@ func indexHandler(w http.ResponseWriter, r *http.Request) {
|
|||||||
ctx := context.Background()
|
ctx := context.Background()
|
||||||
_, err := client.Get(ctx, shortlink).Result()
|
_, err := client.Get(ctx, shortlink).Result()
|
||||||
if err == redis.Nil {
|
if err == redis.Nil {
|
||||||
err = client.Set(ctx, shortlink, value, 0).Err()
|
err = client.Set(ctx, shortlink, shortlink, 0).Err()
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
fmt.Println(err)
|
fmt.Println(err)
|
||||||
@ -128,11 +138,6 @@ func indexHandler(w http.ResponseWriter, r *http.Request) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if command == "delete" {
|
if command == "delete" {
|
||||||
if value != "confirm" {
|
|
||||||
w.WriteHeader(400)
|
|
||||||
w.Write([]byte("400 Bad Request"))
|
|
||||||
}
|
|
||||||
|
|
||||||
ctx := context.Background()
|
ctx := context.Background()
|
||||||
if err := client.Del(ctx, shortlink).Err(); err != nil {
|
if err := client.Del(ctx, shortlink).Err(); err != nil {
|
||||||
w.WriteHeader(500)
|
w.WriteHeader(500)
|
||||||
@ -152,13 +157,13 @@ func indexHandler(w http.ResponseWriter, r *http.Request) {
|
|||||||
|
|
||||||
resp := ""
|
resp := ""
|
||||||
for _, key := range keys {
|
for _, key := range keys {
|
||||||
value, err = client.Get(ctx, key).Result()
|
shortlink, err = client.Get(ctx, key).Result()
|
||||||
if err == redis.Nil {
|
if err == redis.Nil {
|
||||||
w.WriteHeader(500)
|
w.WriteHeader(500)
|
||||||
w.Write([]byte("500 Internal Server Error"))
|
w.Write([]byte("500 Internal Server Error"))
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
resp += key + ":" + value + "\n"
|
resp += key + ":" + shortlink + "\n"
|
||||||
}
|
}
|
||||||
|
|
||||||
w.WriteHeader(200)
|
w.WriteHeader(200)
|
||||||
|
16
susmng.py
16
susmng.py
@ -18,8 +18,10 @@ def get_args():
|
|||||||
parser.add_argument('command')
|
parser.add_argument('command')
|
||||||
parser.add_argument('-s', '--server', default="")
|
parser.add_argument('-s', '--server', default="")
|
||||||
parser.add_argument('-l', '--shortlink', default="")
|
parser.add_argument('-l', '--shortlink', default="")
|
||||||
parser.add_argument('-v', '--value', default="")
|
parser.add_argument('-r', '--redirect', default="")
|
||||||
parser.add_argument('-c', '--config', type=pathlib.Path, default=pathlib.Path(os.path.expanduser('~/.config/susmng/config.json')))
|
parser.add_argument('-a', '--alt-redirect', default="")
|
||||||
|
parser.add_argument('-c', '--alt-condition', default="")
|
||||||
|
parser.add_argument('--config', type=pathlib.Path, default=pathlib.Path(os.path.expanduser('~/.config/susmng/config.json')))
|
||||||
parser.add_argument('-H', '--http', action='store_true')
|
parser.add_argument('-H', '--http', action='store_true')
|
||||||
return parser.parse_args()
|
return parser.parse_args()
|
||||||
|
|
||||||
@ -48,19 +50,21 @@ def main(args):
|
|||||||
|
|
||||||
secret = config['secrets'][server]
|
secret = config['secrets'][server]
|
||||||
|
|
||||||
if args.command == "delete" and args.value != "confirm":
|
formstring = args.command+":"+args.shortlink+":"+args.redirect+":"+args.alt_condition+":"+args.alt_redirect
|
||||||
print("--value not set to 'confirm'... delete operation may fail")
|
print(f"{formstring=}")
|
||||||
|
|
||||||
r = requests.post(f"{'http' if args.http else 'https'}://{server}",
|
r = requests.post(f"{'http' if args.http else 'https'}://{server}",
|
||||||
data = {
|
data = {
|
||||||
'Command': args.command,
|
'Command': args.command,
|
||||||
'Shortlink': args.shortlink,
|
'Shortlink': args.shortlink,
|
||||||
'Value': args.value,
|
'Redirect': args.redirect,
|
||||||
|
'AltCondition': args.alt_condition,
|
||||||
|
'AltRedirect': args.alt_redirect,
|
||||||
},
|
},
|
||||||
headers = {
|
headers = {
|
||||||
'Signature': 'SUS-SIGNATURE-' + hmac.new(
|
'Signature': 'SUS-SIGNATURE-' + hmac.new(
|
||||||
secret.encode("UTF-8"),
|
secret.encode("UTF-8"),
|
||||||
(args.command+":"+args.shortlink+":"+args.value).encode("UTF-8"),
|
formstring.encode("UTF-8"),
|
||||||
hashlib.sha256
|
hashlib.sha256
|
||||||
).hexdigest()
|
).hexdigest()
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user