Pileus Git - vpaste/blob - index.cgi

   1 #!/bin/bash
   2
   3 # Copyright (C) 2009-2013 Andy Spencer
   4 #
   5 # This program is free software: you can redistribute it and/or modify it under
   6 # the terms of the GNU Affero General Public License as published by the Free
   7 # Software Foundation, either version 3 of the License, or (at your option) any
   8 # later version.
   9 #
  10 # This program is distributed in the hope that it will be useful, but WITHOUT
  11 # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
  12 # FOR A PARTICULAR PURPOSE.  See the GNU Affero General Public License for more
  13 # details.
  14
  15 # Remove url codings from stdin
  16 function get_modeline {
  17         echo "$QUERY_STRING" |
  18         sed -e 's/%\([0-9A-F][0-9A-F]\)/\\\\\x\1/g; s/[,&?]/ /g' |
  19         xargs echo -e
  20 }
  21 function get_param {
  22         get_modeline | awk -v "key=$1" 'BEGIN{RS=" "; FS="="}; $1 ~ key {print $2}'
  23 }
  24
  25 # Extract an uploaded file from standard input
  26 #   $1 is the name of the input to extract
  27 function cut_file {
  28         bnd="${CONTENT_TYPE/*boundary\=/}"
  29         awk -v "want=$1" -v "bnd=$bnd" '
  30                 BEGIN { RS="\r\n" }
  31
  32                 # reset based on boundaries
  33                 $0 == "--"bnd""     { st=1; next; }
  34                 $0 == "--"bnd"--"   { st=0; next; }
  35                 $0 == "--"bnd"--\r" { st=0; next; }
  36
  37                 # search for wanted file
  38                 st == 1 && $0 ~  "^Content-Disposition:.* name=\""want"\"" { st=2; next; }
  39                 st == 1 && $0 == "" { st=9; next; }
  40
  41                 # wait for newline, then start printing
  42                 st == 2 && $0 == "" { st=3; next; }
  43                 st == 3             { print $0    }
  44         ' | head -c $((128*1024)) # Limit size to 128K
  45 }
  46
  47 # Respond to a request
  48 function respond {
  49         local allow ctype heads files texts gzip h f
  50         while [ "$1" ]; do
  51                 case $1 in
  52                         -y) allow="true"; ;;
  53                         -c) ctype="$2"; shift ;;
  54                         -h) heads=("${heads[@]}" "$2"); shift ;;
  55                         -f) files=("${files[@]}" "$2"); shift ;;
  56                         *)  texts=("${texts[@]}" "$1"); ;;
  57                 esac
  58                 shift
  59         done
  60
  61         # Check if the browser supports gzip
  62         if [[ "$HTTP_ACCEPT_ENCODING" == *'gzip'* ]]; then
  63                 gzip=true
  64         fi
  65
  66         # Output header
  67         if [ "$ctype" ]; then
  68                 echo "Content-Type: $ctype; charset=UTF-8"
  69         else
  70                 echo "Content-Type: text/plain; charset=UTF-8"
  71         fi
  72         if [ "$gzip" ]; then
  73                 echo "Content-Encoding: gzip"
  74         fi
  75         if [ "$allow" ]; then
  76                 echo "Access-Control-Allow-Origin: *"
  77                 echo "Access-Control-Allow-Headers: Content-Type"
  78         fi
  79         for h in "${heads[@]}"; do
  80                 echo "$h"
  81         done
  82         echo
  83
  84         # Output text messages
  85         if [ "$texts" ]; then
  86                 if [ "$gzip" ]; then
  87                         echo "${texts[@]}" | gzip
  88                 else
  89                         echo "${texts[@]}"
  90                 fi
  91                 exit
  92         fi
  93
  94         # Output body files
  95         if [ "$files" ]; then
  96                 for f in "${files[@]}"; do
  97                         if   [[   "$gzip" && "$f" != *'.gz' ]]; then
  98                                 gzip < "$f"
  99                         elif [[ ! "$gzip" && "$f" == *'.gz' ]]; then
 100                                 zcat < "$f"
 101                         else
 102                                 cat "$f"
 103                         fi
 104                 done
 105                 exit
 106         fi
 107
 108         # Gzip remaining stream
 109         if [ "$gzip" ]; then
 110                 exec 1> >(gzip)
 111         fi
 112 }
 113
 114 # Format and output a file
 115 function format {
 116         # Create a temp file with the provided modeline
 117         tmp="$(mktemp)"
 118         sed "\$avim: $(get_modeline)" "$1" > "$tmp"
 119
 120         # Determine cache name
 121         md5="$(cat index.cgi vimrc "$tmp" /usr/bin/ex | md5sum -b)"
 122         out="cache/${md5% *}.htm"
 123         zip="$out.gz"
 124
 125         # Cache the file, if needed
 126         if [ ! -f "$zip" ]; then
 127                 # - I have some plugins in ~/.vim
 128                 # - Run ex in pty to trick it into thinking that it
 129                 #   has a real terminal, note that we also have to set
 130                 #   term=xterm-256color in vimrc
 131                 HOME=/home/andy \
 132                 /home/vpaste/bin/pty \
 133                 /usr/bin/ex -nXZ -i NONE -u vimrc \
 134                         '+sil! set fde= fdt= fex= inde= inex= key= pa= pexpr=' \
 135                         '+sil! set iconstring= ruf= stl= tal=' \
 136                         "+sil! set titlestring=$1\ -\ vpaste.net" \
 137                         '+sil! set noml' \
 138                         '+sil! $d|'$2    \
 139                         '+sil! %s/\r//g' \
 140                         '+sil! TOhtml'   \
 141                         "+sav! $out"     \
 142                         '+qall!'         \
 143                         "$tmp" >/dev/null 2>&1
 144                 gzip "$out"
 145         fi
 146         rm "$tmp"
 147
 148         # Output the file
 149         respond -y -c "text/html" -f "$zip"
 150 }
 151
 152 # List previous pastes
 153 function do_cmd {
 154         respond
 155         case "$1" in
 156         ls)
 157                 ls -t db | column
 158                 ;;
 159         head)
 160                 awk -v 'rows=4' -v 'cols=60' '
 161                         FNR==1      { gsub(/.*\//, "", FILENAME);
 162                                       print FILENAME
 163                                       print "-----" }
 164                         FNR==1,/^$/ { next }
 165                         /\S/        { i++; printf "%."cols"s\n", $0 }
 166                         i>=rows     { nextfile  }
 167                         ENDFILE     { i=0; print ""  }
 168                 ' $(ls -t db/*)
 169                 ;;
 170         stat)
 171                 ls -l --time-style='+%Y %m' db |
 172                 awk -v 'hdr=Jan  Feb  Mar  Apr  May  Jun  Jul  Aug  Sep  Oct  Nov  Dec' '
 173                         BEGIN { printf "%64s\n", hdr }
 174                         NR>1  { cnt[$6+0][$7+0]++ }
 175                         END   { for (y in cnt) {
 176                                   printf "%4d", y
 177                                   for (m=1; m<=12; m++)
 178                                     printf "%5s", cnt[y][m]
 179                                   printf "\n" } }'
 180                 ;;
 181         esac
 182 }
 183
 184 # Format a file for viewing
 185 function do_print {
 186         if [ -f "./$1" ]; then
 187                 input="$1"
 188         elif [ -f "db/$1" ]; then
 189                 input="db/$1"
 190                 trim='1,/^$/d' # sed command to remove cruft
 191         else
 192                 respond -h 'Status: 404 Not Found' \
 193                         "File '$1' not found"
 194         fi
 195
 196         # Check for javascript
 197         if [[ "$input" == 'embed.js' &&
 198               "$HTTP_ACCEPT" != *'html'* ]]; then
 199                 respond -c text/javascript -f "$input"
 200         fi
 201
 202         # Check for raw paste
 203         if [[ "$QUERY_STRING" == 'raw'* ||
 204               "$REQUEST_URI"  != *'?'* &&
 205               ( "$input"       != 'db/'* ||
 206                 "$HTTP_ACCEPT" != *'html'* ) ]]; then
 207                 respond
 208                 sed "$trim" "$input"
 209                 exit
 210         fi
 211
 212         # Output the file
 213         format "$input" "$trim"
 214 }
 215
 216 # Format a file for viewing
 217 function do_view {
 218         format -
 219 }
 220
 221 # Upload handler
 222 function do_upload {
 223         body=$(cat -)
 224         spam=$(echo -n "$body" | cut_file "ignoreme")
 225         text=$(echo -n "$body" | cut_file "(text|x)")
 226         bans=$(echo -n "$REMOTE_ADDR" | grep -f blacklist)
 227         [ ! -z "$spam" ] && respond "Spam check.."
 228         [ ! -z "$bans" ] && respond "You have been banned"
 229         [   -z "$text" ] && respond "No text pasted"
 230
 231         # Format and save message
 232         output="$(mktemp db/XXXXX)"
 233         cat >"$output" <<-EOF
 234                 vim: $(get_modeline)
 235                 Date: $(date -R)
 236                 From: $REMOTE_ADDR
 237                 User-Agent: $HTTP_USER_AGENT
 238
 239                 $text
 240         EOF
 241
 242         # Redirect user
 243         uri="$url$(basename "$output")"
 244         respond -h 'Status: 302 Found' \
 245                 -h "Location: $uri"    \
 246                 "$uri"
 247 }
 248
 249 # Default index page
 250 function do_help {
 251         filetypes=$(
 252                 ls /usr/share/vim/vim*/syntax/ /home/andy/.vim/syntax/ |
 253                 sed -n '/^\(syntax\|manual\|synload\|2html\|colortest\|hitest\).vim$/d; s/.vim$//p' |
 254                 sort | uniq
 255         )
 256         uploads=$(ls -t db 2>/dev/null | head -n 5)
 257         filetype=$(get_param '^(ft|filet(y(pe?)?)?)$')
 258         vpaste='<a href="vpaste?ft=sh">vpaste</a>'
 259         repo='https://lug.rose-hulman.edu/svn/misc/trunk/htdocs/vpaste/'
 260
 261         respond -c text/html
 262         cat <<-EOF
 263         <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN"
 264           "http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd">
 265         <html xmlns="http://www.w3.org/1999/xhtml" lang="en" xml:lang="en">
 266                 <head>
 267                         <title>vpaste.net - Vim based pastebin</title>
 268                         <meta http-equiv="Content-type" content="text/html;charset=UTF-8" />
 269                         <meta name="description" content="vpaste: Vim based pastebin" />
 270                         <meta name="keywords" content="vpaste,paste,pastebin,vim" />
 271                         <meta name="google-site-verification" content="OvHF73zD7osJ1VSq9rJxnMFlja36944ud6CiP_iXQnI" />
 272                         <style type="text/css">
 273                                 *          { margin: 0;
 274                                              padding: 0; }
 275                                 body       { margin: 4em 8em 4em 8em;
 276                                              font-family: sans-serif; }
 277                                 input      { padding: 2px 6px 3px 6px; }
 278                                 /* Items */
 279                                 textarea   { width: 100%;
 280                                              margin-bottom: 0.5em; }
 281                                 .buttons   { float: left; }
 282                                 .links     { float: right; }
 283                                 .links *   { text-decoration: none;
 284                                              margin-left: 0.5em; }
 285                                 .box       { display: none;
 286                                              clear: both;
 287                                              margin-top: 2.7em;
 288                                              border-top: solid 1px #888; }
 289                                 /* box contents */
 290                                 h1         { margin-top: 1.0em;
 291                                              font-size: larger; }
 292                                 ul,dd,dl,p { margin: 0 0 0 2em; }
 293                                 dt         { font-weight: bold;
 294                                              padding: 0.5em 0 0 0; }
 295                                 span       { font-family: monospace; }
 296                                 .cmds dd   { font-family: monospace; }
 297                         </style>
 298                         <script type="text/javascript">
 299                                 //<![CDATA[
 300                                 function show(id) {
 301                                         var boxes = document.getElementsByClassName('box')
 302                                         for (var i = 0; i < boxes.length; i++) {
 303                                                 var box = boxes[i]
 304                                                 if (box.id == id && box.style.display != 'block')
 305                                                         box.style.display = 'block'
 306                                                 else
 307                                                         box.style.display = "none"
 308                                         }
 309                                 }
 310                                 function autoshow() {
 311                                         var id  = document.location.toString().replace(/.*#/, '')
 312                                         var box = document.getElementById(id)
 313                                         if (box) box.style.display = "block"
 314                                 }
 315                                 //]]>
 316                         </script>
 317                 </head>
 318
 319                 <body onload="autoshow()">
 320                         <form id="form" method="post" action="" enctype="multipart/form-data">
 321                                 <div>
 322                                         <input style="display:none" type="text" name="ignoreme" value="" />
 323                                         <textarea name="text" cols="80" rows="25"></textarea>
 324                                 </div>
 325                                 <div class="buttons">
 326                                         <select onchange="document.getElementById('form').action =
 327                                                           document.location + '?ft=' + this.value;">
 328                                                 <option value="" disabled="disabled">Filetype</option>
 329                                                 <option value="">None</option>
 330                                                 $(for ft in $filetypes; do
 331                                                         echo "<option$(
 332                                                         [ "$ft" = "$filetype" ] &&
 333                                                                 echo ' selected="selected"'
 334                                                         )>$ft</option>"
 335                                                 done)
 336                                         </select>
 337                                         <input type="submit" value="Paste" />
 338                                 </div>
 339                                 <div class="links">
 340                                         <a href="">vpaste</a> <span>-</span>
 341                                         <a href="#usage"   onclick="show('usage'  )">Usage</a>
 342                                         <a href="#devel"   onclick="show('devel'  )">Development</a>
 343                                         <a href="#uploads" onclick="show('uploads')">Uploads</a>
 344                                 </div>
 345                         </form>
 346
 347                         <div class="box" id="usage">
 348                                 <h1>Pasting</h1>
 349                                 <dl class="cmds">
 350                                         <dt>From a shell</dt>
 351                                         <dd> $vpaste file [option=value,..]</dd>
 352                                         <dd> &lt;command&gt; | $vpaste [option=value,..]</dd>
 353
 354                                         <dt>From Vim</dt>
 355                                         <dd> :map vp :exec "w !vpaste ft=".&amp;ft&lt;CR&gt;</dd>
 356                                         <dd> :vmap vp &lt;ESC&gt;:exec "'&lt;,'&gt;w !vpaste ft=".&amp;ft&lt;CR&gt;</dd>
 357
 358                                         <dt>With curl</dt>
 359                                         <dd> &lt;command&gt; | curl -F 'text=&lt;-' $url[?option=value,..]</dd>
 360                                 </dl>
 361
 362                                 <h1>Options</h1>
 363                                 <p>Add <b>?option[=value],..</b> to make your text a rainbow.</p>
 364                                 <p>Options specified when uploading are saved as defaults.</p>
 365
 366                                 <dl>
 367                                         <dt>bg, background={light|dark}</dt>
 368                                         <dd>Background color to use for the page</dd>
 369                                         <dt>et, expandtab</dt>
 370                                         <dd>Expand tabs to spaces</dd>
 371                                         <dt>fdm, foldmethod=(syntax|indent)</dt>
 372                                         <dd>Turn on dynamic code folding</dd>
 373                                         <dt>ft, filetype={filetype}</dt>
 374                                         <dd>A filetype to use for highlighting, see above menu for supported types</dd>
 375                                         <dt>nu, number</dt>
 376                                         <dd>Add line numbers</dd>
 377                                         <dt>ts, tabstop=[N]</dt>
 378                                         <dd>Number of spaces to use for tabs when <b>et</b> is set</dd>
 379                                         <dt>...</dt>
 380                                         <dd>See :help modeline for more information</dd>
 381                                 </dl>
 382                         </div>
 383
 384                         <div class="box" id="devel">
 385                                 <h1>License</h1>
 386                                 <p>Copyright © 2009-2013
 387                                    Andy Spencer &lt;andy753421@gmail.com&gt;</p>
 388                                 <p>See individual files for licenses</p>
 389
 390                                 <h1>Source code</h1>
 391                                 <dl>
 392                                         <dt>Client</dt>
 393                                         <dd><a href="vpaste?ft=sh">vpaste</a>
 394                                             <a href="embed.js?ft=javascript">embed.js</a></dd>
 395                                         <dt>Server</dt>
 396                                         <dd><a href="index.cgi?ft=sh">index.cgi</a>
 397                                             <a href="vimrc?ft=vim">vimrc</a>
 398                                             <a href="htaccess?ft=apache">htaccess</a>
 399                                             <a href="robots.txt?ft=robots">robots.txt</a>
 400                                             <a href="sitemap.xml?ft=xml">sitemap.xml</a>
 401                                             <a href="blacklist?raw">blacklist</a></dd>
 402                                         <dt>Patches</dt>
 403                                         <dd><a href="2html.patch?ft=diff">2html.patch</a></dd>
 404                                         <dt>Subversion</dt>
 405                                         <dd><a href="$repo">$repo</a></dd>
 406                                 </dl>
 407
 408                                 <h1>Bugs</h1>
 409                                 <ul>
 410                                         <li>Using strange filetypes (ft=2html) may result in strange output.</li>
 411                                         <li><a href="mailto:andy753421@gmail.com?subject=vpaste bug">Other?</a></li>
 412                                 </ul>
 413                         </div>
 414
 415                         <div class="box" id="uploads">
 416                                 <h1>Recent Uploads</h1>
 417                                 <ul>$(for upload in ${uploads[@]}; do
 418                                     echo -n "<li>"
 419                                     echo -n "<span>$upload</span> "
 420                                     echo -n "<a href='$upload?raw'>text</a> "
 421                                     echo -n "<a href='$upload'>rainbow</a>"
 422                                     echo "</li>"
 423                                 done)
 424                                 </ul>
 425                                 <p><a href="ls">list all</a></p>
 426                                 <p><a href="head">sample all</a></p>
 427                                 <p><a href="stat">statistics</a></p>
 428                         </div>
 429                 </body>
 430         </html>
 431         EOF
 432 }
 433
 434 # Main
 435 PATH=/bin:/usr/bin
 436 url="http://$HTTP_HOST${REQUEST_URI/\?*}"
 437 pathinfo="${REQUEST_URI/*\/}"
 438 pathinfo="${pathinfo/\?*}"
 439
 440 if [ "$pathinfo" = ls ]; then
 441         do_cmd ls
 442 elif [ "$pathinfo" = head ]; then
 443         do_cmd head
 444 elif [ "$pathinfo" = stat ]; then
 445         do_cmd stat
 446 elif [ "$pathinfo" = view ]; then
 447         do_view
 448 elif [ "$pathinfo" ]; then
 449         do_print "$pathinfo"
 450 elif [ "$CONTENT_TYPE" ]; then
 451         do_upload
 452 else
 453         do_help
 454 fi