summaryrefslogtreecommitdiffstats
path: root/krebs/3modules/go.nix
diff options
context:
space:
mode:
authorlassulus <lassulus@lassul.us>2021-04-25 17:34:23 +0200
committerlassulus <lassulus@lassul.us>2021-04-25 17:34:23 +0200
commit9ae85f32e3c23fdd535ac3e905f1891ecba2923d (patch)
treea878a4408e4af6deaef5d052d5432ef251e709f0 /krebs/3modules/go.nix
parente41239db3caa26405ed61c394dd58d31521f80b7 (diff)
go: store urls in redis
Diffstat (limited to 'krebs/3modules/go.nix')
-rw-r--r--krebs/3modules/go.nix48
1 files changed, 14 insertions, 34 deletions
diff --git a/krebs/3modules/go.nix b/krebs/3modules/go.nix
index 4df73509c..5686fea3f 100644
--- a/krebs/3modules/go.nix
+++ b/krebs/3modules/go.nix
@@ -20,61 +20,41 @@ let
};
imp = {
+ services.redis = {
+ enable = true;
+ };
+
krebs.htgen.go = {
port = cfg.port;
script = ''. ${pkgs.writeDash "go" ''
- find_item() {
- if test ''${#1} -ge 7; then
- set -- "$(find "$STATEDIR/items" -mindepth 1 -maxdepth 1 \
- -regex "$STATEDIR/items/$1[0-9A-Za-z]*$")"
- if test -n "$1" && test $(echo "$1" | wc -l) = 1; then
- echo "$1"
- return 0
- fi
- fi
- return 1
- }
-
- STATEDIR=$HOME
- mkdir -p "$STATEDIR/items"
+ set -x
case "$Method $Request_URI" in
"GET /"*)
- if item=$(find_item "''${Request_URI#/}"); then
- uri=$(cat "$item")
+ if item=$(${pkgs.redis}/bin/redis-cli --raw get "''${Request_URI#/}"); then
printf 'HTTP/1.1 302 Found\r\n'
printf 'Content-Type: text/plain\r\n'
printf 'Connection: closed\r\n'
- printf 'Location: %s\r\n' "$uri"
+ printf 'Location: %s\r\n' "$item"
printf '\r\n'
exit
fi
;;
"POST /")
- uri=$(mktemp -t htgen.$$.content.XXXXXXXX)
- trap 'rm $uri >&2' EXIT
-
- head -c "$req_content_length" \
+ uri=$(head -c "$req_content_length" \
| sed 's/+/ /g;s/%\(..\)/\\x\1/g;' \
| xargs -0 echo -e \
| tee /tmp/tee.log \
| ${pkgs.urix}/bin/urix \
| head -1 \
- > "$uri"
- sha256=$(sha256sum -b "$uri" | cut -d\ -f1)
- base32=$(${pkgs.nixStable}/bin/nix-hash --to-base32 --type sha256 "$sha256")
- item="$STATEDIR/items/$base32"
- ref="http://$req_host/$base32"
+ )
- if ! test -e "$item"; then
- mkdir -v -p "$STATEDIR/items" >&2
- cp -v "$uri" "$item" >&2
- fi
+ sha256=$(echo "$uri" | sha256sum -b | cut -d\ -f1)
+ base32=$(${pkgs.nixStable}/bin/nix-hash --to-base32 --type sha256 "$sha256")
+ base32short=$(echo "$base32" | cut -b-5)
+ ${pkgs.redis}/bin/redis-cli set "$base32short" "$uri" >/dev/null
- base32short=$(echo "$base32" | cut -b-7)
- if item=$(find_item "$base32short"); then
- ref="http://$req_host/$base32short"
- fi
+ ref="http://$req_host/$base32short"
printf 'HTTP/1.1 200 OK\r\n'
printf 'Content-Type: text/plain; charset=UTF-8\r\n'