116 lines
2.8 KiB
Python
116 lines
2.8 KiB
Python
# t0url - url.t0.vc
|
|
# MIT License
|
|
|
|
import random
|
|
import shelve
|
|
import string
|
|
import validators
|
|
|
|
from flask import abort, Flask, request, redirect
|
|
from werkzeug.exceptions import HTTPException
|
|
|
|
DB = 'data/t0url'
|
|
PORT = 5004
|
|
DOMAIN = 'url.t0.vc'
|
|
URL = 'https://' + DOMAIN
|
|
POST = 'url'
|
|
|
|
def help():
|
|
form = (
|
|
'<form action="{0}" method="POST" accept-charset="UTF-8">'
|
|
'<input name="web" type="hidden" value="true">'
|
|
'<input name="name" style="border: none;">'
|
|
'<br>URL: <input name="{1}">'
|
|
'<br>CAPTCHA: Who owns this site? <input name="captcha">'
|
|
'<br><button type="submit">Submit</button></form>'.format(URL, POST)
|
|
)
|
|
return """
|
|
<pre>
|
|
url.t0.vc
|
|
NAME
|
|
t0url: command line short URL.
|
|
|
|
USAGE
|
|
<command> | curl -F '{0}=<-' {1}
|
|
or upload from the web:
|
|
|
|
{2}
|
|
|
|
EXAMPLES
|
|
~$ cat yourfile | curl -F '{0}=<-' {1}
|
|
{1}/VFAR
|
|
~$ firefox {1}/VFAR
|
|
|
|
Add this to your .bashrc:
|
|
|
|
alias {0}="curl -F '{0}=<-' {1}"
|
|
|
|
Now you can pipe a URL directly into {0}!
|
|
|
|
SOURCE CODE
|
|
https://txt.t0.vc/RWFA
|
|
https://github.com/tannercollin/t0url
|
|
|
|
SEE ALSO
|
|
https://txt.t0.vc
|
|
</pre>""".format(POST, URL, form)
|
|
|
|
def new_id():
|
|
return ''.join(random.choice(string.ascii_uppercase) for _ in range(4))
|
|
|
|
flask_app = Flask(__name__)
|
|
|
|
@flask_app.route('/', methods=['GET'])
|
|
def index():
|
|
return '<html><body>{}</body></html>'.format(help())
|
|
|
|
@flask_app.route('/', methods=['POST'])
|
|
def new():
|
|
try:
|
|
is_web = 'web' in request.form
|
|
|
|
name = request.form.get('name', None)
|
|
if name:
|
|
# TODO: fail if name field is filled out
|
|
print('Name filled out:', name)
|
|
|
|
captcha = request.form.get('captcha', None)
|
|
if is_web and 'tanner' not in captcha.lower() and 'collin' not in captcha.lower():
|
|
print('Captcha failed:', captcha)
|
|
return redirect('https://txt.t0.vc/LPPZ')
|
|
else:
|
|
print('Captcha passed:', captcha)
|
|
|
|
with shelve.open(DB) as db:
|
|
nid = new_id()
|
|
while nid in db:
|
|
nid = new_id()
|
|
|
|
url = request.form['url'].strip()
|
|
|
|
if not url: abort(400, 'Error: URL missing')
|
|
if DOMAIN in url: abort(400, 'Error: use a different URL')
|
|
if not url.startswith('http'):
|
|
url = 'http://' + url
|
|
if not validators.url(url): abort(400, 'Error: invalid URL')
|
|
|
|
db[nid] = url
|
|
|
|
print('Adding url {}:\n{}'.format(nid, url))
|
|
|
|
return '{}/{}'.format(URL, nid) + '\n'
|
|
except HTTPException:
|
|
raise
|
|
except:
|
|
abort(400, 'Error: unknown problem')
|
|
|
|
@flask_app.route('/<nid>', methods=['GET'])
|
|
def get(nid):
|
|
try:
|
|
with shelve.open(DB) as db:
|
|
return redirect(db[nid])
|
|
except:
|
|
abort(404)
|
|
|
|
flask_app.run(port=PORT)
|