Iain Learmonth
d54fae7423
link to onion services page from origin page link to previews of distribution lists from lists
127 lines
5.1 KiB
Python
127 lines
5.1 KiB
Python
from datetime import datetime
|
|
|
|
from flask import flash, redirect, url_for, render_template, Response, Blueprint
|
|
from flask_wtf import FlaskForm
|
|
from sqlalchemy import exc
|
|
from wtforms import StringField, SelectField, SubmitField, BooleanField
|
|
from wtforms.validators import DataRequired
|
|
|
|
from app.extensions import db
|
|
from app.models.base import Group
|
|
from app.models.mirrors import Origin
|
|
from app.portal.util import response_404, view_lifecycle
|
|
|
|
bp = Blueprint("origin", __name__)
|
|
|
|
|
|
class NewOriginForm(FlaskForm):
|
|
domain_name = StringField('Domain Name', validators=[DataRequired()])
|
|
description = StringField('Description', validators=[DataRequired()])
|
|
group = SelectField('Group', validators=[DataRequired()])
|
|
auto_rotate = BooleanField("Enable auto-rotation?", default=True)
|
|
submit = SubmitField('Save Changes')
|
|
|
|
|
|
class EditOriginForm(FlaskForm):
|
|
description = StringField('Description', validators=[DataRequired()])
|
|
group = SelectField('Group', validators=[DataRequired()])
|
|
auto_rotate = BooleanField("Enable auto-rotation?")
|
|
submit = SubmitField('Save Changes')
|
|
|
|
|
|
@bp.route("/new", methods=['GET', 'POST'])
|
|
@bp.route("/new/<group_id>", methods=['GET', 'POST'])
|
|
def origin_new(group_id=None):
|
|
form = NewOriginForm()
|
|
form.group.choices = [(x.id, x.group_name) for x in Group.query.all()]
|
|
if form.validate_on_submit():
|
|
origin = Origin()
|
|
origin.group_id = form.group.data
|
|
origin.domain_name = form.domain_name.data
|
|
origin.description = form.description.data
|
|
origin.auto_rotation = form.auto_rotate.data
|
|
origin.created = datetime.utcnow()
|
|
origin.updated = datetime.utcnow()
|
|
try:
|
|
db.session.add(origin)
|
|
db.session.commit()
|
|
flash(f"Created new origin {origin.domain_name}.", "success")
|
|
return redirect(url_for("portal.origin.origin_edit", origin_id=origin.id))
|
|
except exc.SQLAlchemyError as e:
|
|
print(e)
|
|
flash("Failed to create new origin.", "danger")
|
|
return redirect(url_for("portal.origin.origin_list"))
|
|
if group_id:
|
|
form.group.data = group_id
|
|
return render_template("new.html.j2", section="origin", form=form)
|
|
|
|
|
|
@bp.route('/edit/<origin_id>', methods=['GET', 'POST'])
|
|
def origin_edit(origin_id):
|
|
origin = Origin.query.filter(Origin.id == origin_id).first()
|
|
if origin is None:
|
|
return Response(render_template("error.html.j2",
|
|
section="origin",
|
|
header="404 Origin Not Found",
|
|
message="The requested origin could not be found."),
|
|
status=404)
|
|
form = EditOriginForm(group=origin.group_id,
|
|
description=origin.description,
|
|
auto_rotate=origin.auto_rotation)
|
|
form.group.choices = [(x.id, x.group_name) for x in Group.query.all()]
|
|
if form.validate_on_submit():
|
|
origin.group_id = form.group.data
|
|
origin.description = form.description.data
|
|
origin.auto_rotation = form.auto_rotate.data
|
|
origin.updated = datetime.utcnow()
|
|
try:
|
|
db.session.commit()
|
|
flash("Saved changes to group.", "success")
|
|
except exc.SQLAlchemyError:
|
|
flash("An error occurred saving the changes to the group.", "danger")
|
|
return render_template("origin.html.j2",
|
|
section="origin",
|
|
origin=origin, form=form)
|
|
|
|
|
|
@bp.route("/list")
|
|
def origin_list():
|
|
origins = Origin.query.order_by(Origin.domain_name).all()
|
|
return render_template("list.html.j2",
|
|
section="origin",
|
|
title="Web Origins",
|
|
item="origin",
|
|
new_link=url_for("portal.origin.origin_new"),
|
|
items=origins,
|
|
extra_buttons=[{
|
|
"link": url_for("portal.origin.origin_onion"),
|
|
"text": "Onion services",
|
|
"style": "onion"
|
|
}])
|
|
|
|
|
|
@bp.route("/onion")
|
|
def origin_onion():
|
|
origins = Origin.query.order_by(Origin.domain_name).all()
|
|
return render_template("list.html.j2",
|
|
section="origin",
|
|
title="Onion Sites",
|
|
item="onion service",
|
|
new_link=url_for("portal.onion.onion_new"),
|
|
items=origins)
|
|
|
|
|
|
@bp.route("/destroy/<origin_id>", methods=['GET', 'POST'])
|
|
def origin_destroy(origin_id: int):
|
|
origin = Origin.query.filter(Origin.id == origin_id, Origin.destroyed == None).first()
|
|
if origin is None:
|
|
return response_404("The requested origin could not be found.")
|
|
return view_lifecycle(
|
|
header=f"Destroy origin {origin.domain_name}",
|
|
message=origin.description,
|
|
success_message="All proxies from the destroyed origin will shortly be destroyed at their providers.",
|
|
success_view="portal.origin.origin_list",
|
|
section="origin",
|
|
resource=origin,
|
|
action="destroy"
|
|
)
|