From 5a3764f178192fed2d8f39379b0481d1ba5ac345 Mon Sep 17 00:00:00 2001
From: rubenwardy <rw@rubenwardy.com>
Date: Tue, 20 Mar 2018 02:58:31 +0000
Subject: [PATCH] Add edit package

---
 app/models.py                      | 36 +++++++++++++++++++++++++
 app/templates/package_details.html |  1 +
 app/templates/package_edit.html    | 24 +++++++++++++++++
 app/views/packages.py              | 43 +++++++++++++++++++++++++++---
 4 files changed, 101 insertions(+), 3 deletions(-)
 create mode 100644 app/templates/package_edit.html

diff --git a/app/models.py b/app/models.py
index cfe19315..f71725d8 100644
--- a/app/models.py
+++ b/app/models.py
@@ -58,6 +58,12 @@ class UserRoles(db.Model):
 	user_id = db.Column(db.Integer(), db.ForeignKey('user.id', ondelete='CASCADE'))
 	role_id = db.Column(db.Integer(), db.ForeignKey('role.id', ondelete='CASCADE'))
 
+class Permission(enum.Enum):
+	EDIT_PACKAGE   = "EDIT_PACKAGE"
+	APPROVE        = "APPROVE"
+	DELETE_PACKAGE = "DELETE_PACKAGE"
+	CHANGE_AUTHOR  = "CHANGE_AUTHOR"
+
 class PackageType(enum.Enum):
 	MOD  = "Mod"
 	GAME = "Game"
@@ -82,6 +88,23 @@ class PackageType(enum.Enum):
 		else:
 			return None
 
+	@classmethod
+	def choices(cls):
+		return [(choice, choice.value) for choice in cls]
+
+	@classmethod
+	def coerce(cls, item):
+		"""item will be both type(enum) AND type(unicode).
+		"""
+		if item == 'PackageType.MOD' or item == PackageType.MOD:
+			return PackageType.MOD
+		elif item == 'PackageType.GAME' or item == PackageType.GAME:
+			return PackageType.GAME
+		elif item == 'PackageType.TXP' or item == PackageType.TXP:
+			return PackageType.TXP
+		else:
+			print("Can't coerce", item, type(item))
+
 class Package(db.Model):
 	id           = db.Column(db.Integer, primary_key=True)
 
@@ -99,6 +122,19 @@ class Package(db.Model):
 	issueTracker = db.Column(db.String(200), nullable=True)
 	forums       = db.Column(db.String(200), nullable=False)
 
+	def getDetailsURL(self):
+		return url_for("package_page",
+				type=self.type.getTitle().lower(),
+				author=self.author.username, name=self.name)
+
+	def getEditURL(self):
+		return url_for("edit_package_page",
+				type=self.type.getTitle().lower(),
+				author=self.author.username, name=self.name)
+
+	def checkPerm(self, user, perm):
+		return user == self.author
+
 # Setup Flask-User
 db_adapter = SQLAlchemyAdapter(db, User)        # Register the User model
 user_manager = UserManager(db_adapter, app)     # Initialize Flask-User
diff --git a/app/templates/package_details.html b/app/templates/package_details.html
index b75be243..457673aa 100644
--- a/app/templates/package_details.html
+++ b/app/templates/package_details.html
@@ -20,6 +20,7 @@
 		</table>
 
 		<ul class="buttonset linedbuttonset">
+			{% if package.checkPerm(current_user, "EDIT_PACKAGE") %}<li><a href="{{ package.getEditURL() }}">Edit</a></li>{% endif %}
 			{% if package.repo %}<li><a href="{{ package.repo }}">View Source</a></li>{% endif %}
 			{% if package.forums %}<li><a href="{{ package.forums }}">Forums</a></li>{% endif %}
 			{% if package.issueTracker %}<li><a href="{{ package.issueTracker }}">Issue Tracker</a></li>{% endif %}
diff --git a/app/templates/package_edit.html b/app/templates/package_edit.html
new file mode 100644
index 00000000..f7e34b11
--- /dev/null
+++ b/app/templates/package_edit.html
@@ -0,0 +1,24 @@
+{% extends "base.html" %}
+
+{% block title %}
+	{{ package.title }}
+{% endblock %}
+
+{% block content %}
+	{% from "macros/forms.html" import render_field, render_submit_field %}
+	<form method="POST" action="">
+		{{ form.hidden_tag() }}
+
+		{{ render_field(form.name) }}
+		{{ render_field(form.title) }}
+		{{ render_field(form.shortDesc) }}
+		{{ render_field(form.desc) }}
+		{{ render_field(form.type) }}
+		{{ render_field(form.repo) }}
+		{{ render_field(form.website) }}
+		{{ render_field(form.issueTracker) }}
+		{{ render_field(form.forums) }}
+		{{ render_submit_field(form.submit) }}
+		<input type="submit" value="Go">
+	</form>
+{% endblock %}
diff --git a/app/views/packages.py b/app/views/packages.py
index 5f3f0fe1..e525caac 100644
--- a/app/views/packages.py
+++ b/app/views/packages.py
@@ -4,6 +4,9 @@ from flask.ext import menu
 from app import app
 from app.models import *
 
+from flask_wtf import FlaskForm
+from wtforms import *
+
 
 # TODO: the following could be made into one route, except I'm not sure how
 # to do the menu
@@ -27,9 +30,7 @@ def txp_page():
 	return render_template('packages.html', title="Texture Packs", packages=packages)
 
 
-
-@app.route("/<type>s/<author>/<name>/")
-def package_page(type, author, name):
+def getPageByInfo(type, author, name):
 	user = User.query.filter_by(username=author).first()
 	if user is None:
 		abort(404)
@@ -39,4 +40,40 @@ def package_page(type, author, name):
 	if package is None:
 		abort(404)
 
+	return package
+
+
+@app.route("/<type>s/<author>/<name>/")
+def package_page(type, author, name):
+	package = getPageByInfo(type, author, name)
 	return render_template('package_details.html', package=package)
+
+
+class PackageForm(FlaskForm):
+	name         = StringField("Name")
+	title        = StringField("Title")
+	shortDesc    = StringField("Short Description")
+	desc         = StringField("Long Description")
+	type         = SelectField("Type", choices=PackageType.choices(), coerce=PackageType.coerce)
+	repo         = StringField("Repo URL")
+	website      = StringField("Website URL")
+	issueTracker = StringField("Issue Tracker URL")
+	forums       = StringField("Forum Topic ID")
+	submit       = SubmitField('Save')
+
+@app.route("/<type>s/<author>/<name>/edit/", methods=['GET', 'POST'])
+@login_required
+def edit_package_page(type, author, name):
+	package = getPageByInfo(type, author, name)
+	if not package.checkPerm(current_user, Permission.EDIT_PACKAGE):
+		return redirect(package.getDetailsURL())
+
+	# Initial form class from post data and default data
+	form = PackageForm(formdata=request.form, obj=package)
+	if request.method == "POST" and form.validate():
+		# Successfully submitted!
+		form.populate_obj(package) # copy to row
+		db.session.commit() # save
+		return redirect(package.getDetailsURL()) # redirect
+
+	return render_template('package_edit.html', package=package, form=form)