#!/bin/bash
# Copyright (C) 2009-2011 Andy Spencer
#
# This program is free software: you can redistribute it and/or modify it under
# the terms of the GNU Affero General Public License as published by the Free
# Software Foundation, either version 3 of the License, or (at your option) any
# later version.
#
# This program is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
# FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for more
# details.
# Remove url codings from stdin
function get_modeline {
echo "$QUERY_STRING" |
sed -e 's/%\([0-9A-F][0-9A-F]\)/\\\\\x\1/g; s/[,&]/ /g' |
xargs echo -e
}
# Extract an uploaded file from standard input
# $1 is the name of the input to extract
function cut_file {
bnd="${CONTENT_TYPE/*boundary\=/}"
awk -v "want=$1" -v "bnd=$bnd" '
BEGIN { RS="\r\n" }
# reset based on boundaries
$0 == "--"bnd"" { st=1; next; }
$0 == "--"bnd"--" { st=0; next; }
$0 == "--"bnd"--\r" { st=0; next; }
# search for wanted file
st == 1 && $0 ~ "^Content-Disposition:.* name=\""want"\"" { st=2; next; }
st == 1 && $0 == "" { st=9; next; }
# wait for newline, then start printing
st == 2 && $0 == "" { st=3; next; }
st == 3 { print $0 }
' | head -c $((128*1024)) # Limit size to 128K
}
# Print out a generic header
function header {
echo "Content-Type: $1; charset=UTF-8"
echo
}
# Print plain message and exit
function message {
while [ "$1" == '-h' ]; do
shift; echo $1; shift
done
header text/plain
echo "$*"
exit
}
# List previous pastes
function do_cmd {
header text/plain
case "$1" in
head)
for i in $(ls -t db/*); do
basename $i
basename $i | sed 's/./-/g'
sed '1,/^$/d; /^\s*$/d' $i | sed -n '1,5s/\(.\{0,60\}\).*/\1/p'
echo
done
;;
ls)
ls -t db | column
;;
esac
}
# Format a file for viewing
function do_print {
if [ -f "./$1" ]; then
input="$1"
elif [ -f "db/$1" ]; then
input="db/$1"
trim='1,/^$/d' # sed command to remove cruft
else
message -h 'Status: 404 Not Found' \
"File '$1' not found"
fi
if [[ "$HTTP_ACCEPT" == *'html'* &&
"$QUERY_STRING" != 'raw'* ]]; then
# Create a temp file with the provided modeline
output="$(mktemp)"
tmp="$(mktemp)"
sed "\$avim: $(get_modeline)" "$input" > "$tmp"
# - I have some plugins in ~/.vim
# - Run ex in screen to trick it into thinking that it
# has a real terminal, note that we also have to set
# term=xterm-256color in vimrc
HOME=/home/andy \
screen -D -m ex -nXZ -i NONE -u vimrc \
'+sil! set fde= fdt= fex= inde= inex= key= pa= pexpr=' \
'+sil! set iconstring= ruf= stl= tal=' \
"+sil! set titlestring=$1\ -\ vpaste.net" \
'+sil! set noml' \
'+sil! $d|'$trim \
'+sil! %s/\r//g' \
'+sil! TOhtml' \
"+sav! $output" \
'+qall!' \
"$tmp"
header text/html
cat "$output"
rm "$tmp" "$output"
else
header text/plain
sed "$trim" "$input"
fi
}
# Upload handler
function do_upload {
body=$(cat -)
spam=$(echo -n "$body" | cut_file "ignoreme")
text=$(echo -n "$body" | cut_file "(text|x)")
[ ! -z "$spam" ] && message "Spam check.."
[ -z "$text" ] && message "No text pasted"
# Format and save message
output="$(mktemp db/XXXXX)"
cat >"$output" <<-EOF
vim: $(get_modeline)
Date: $(date -R)
From: $REMOTE_ADDR
$text
EOF
# Redirect user
uri="$SCRIPT_URI$(basename "$output")"
message -h 'Status: 302 Found' \
-h "Location: $uri" \
"$uri"
}
# Default index page
function do_help {
filetypes=$(
ls /usr/share/vim/vim*/syntax/ /home/andy/.vim/syntax/ |
sed -n '/^\(syntax\|manual\|synload\|2html\|colortest\|hitest\).vim$/d; s/.vim$//p' |
sort | uniq
)
uploads=$(ls -t db | head -n 5)
header text/html
cat <<-EOF
vpaste.net - Vim based pastebin
NAME
vpaste: Vim based pastebin
SYNOPSIS
vpaste file [option=value,..]
<command> | vpaste [option=value,..]
<command> | curl -F 'text=<-' $SCRIPT_URI[?option=value,..]
:map vp :exec "w !vpaste ft=".&ft<CR>
:vmap vp <ESC>:exec "'<,'>w !vpaste ft=".&ft<CR>
DESCRIPTION
Add ?[option[=value],..] to make your text a rainbow.
Options specified when uploading are used as defaults.
OPTIONS
- ft, filetype={filetype}
- A filetype to use for highlighting, see above menu for supported types
- bg, background={light|dark}
- Background color to use for the page
- et, expandtab
- Expand tabs to spaces
- ts, tabstop=[N]
- Number of spaces to use for tabs when et is set
- ...
- See :help modeline for more information
BUGS
- Using strange filetypes (ft=2html) may result in strange output.
- Other?
SOURCE
LATEST UPLOADS
$(for upload in ${uploads[@]}; do
echo -n "- "
echo -n "$upload "
echo -n "text "
echo -n "rainbow"
echo "
"
done)
list all
sample all
EOF
}
# Main
pathinfo="${REQUEST_URI/*\/}"
pathinfo="${pathinfo/\?*}"
if [ "$pathinfo" = ls ]; then
do_cmd ls
elif [ "$pathinfo" = head ]; then
do_cmd head
elif [ "$pathinfo" ]; then
do_print "$pathinfo"
elif [ "$CONTENT_TYPE" ]; then
do_upload
else
do_help
fi