app/models.py

   1 # Content DB
   2 # Copyright (C) 2018  rubenwardy
   3 #
   4 # This program is free software: you can redistribute it and/or modify
   5 # it under the terms of the GNU General Public License as published by
   6 # the Free Software Foundation, either version 3 of the License, or
   7 # (at your option) any later version.
   8 #
   9 # This program is distributed in the hope that it will be useful,
  10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 # GNU General Public License for more details.
  13 #
  14 # You should have received a copy of the GNU General Public License
  15 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
  16
  17
  18 from flask import Flask, url_for
  19 from flask_sqlalchemy import SQLAlchemy
  20 from flask_migrate import Migrate
  21 from urllib.parse import urlparse
  22 from app import app, gravatar
  23 from datetime import datetime
  24 from sqlalchemy.orm import validates
  25 from flask_user import login_required, UserManager, UserMixin, SQLAlchemyAdapter
  26 import enum
  27
  28 # Initialise database
  29 db = SQLAlchemy(app)
  30 migrate = Migrate(app, db)
  31
  32
  33 class UserRank(enum.Enum):
  34         BANNED         = 0
  35         NOT_JOINED     = 1
  36         NEW_MEMBER     = 2
  37         MEMBER         = 3
  38         TRUSTED_MEMBER = 4
  39         EDITOR         = 5
  40         MODERATOR      = 6
  41         ADMIN          = 7
  42
  43         def atLeast(self, min):
  44                 return self.value >= min.value
  45
  46         def getTitle(self):
  47                 return self.name.replace("_", " ").title()
  48
  49         def toName(self):
  50                 return self.name.lower()
  51
  52         def __str__(self):
  53                 return self.name
  54
  55         @classmethod
  56         def choices(cls):
  57                 return [(choice, choice.getTitle()) for choice in cls]
  58
  59         @classmethod
  60         def coerce(cls, item):
  61                 return item if type(item) == UserRank else UserRank[item]
  62
  63
  64 class Permission(enum.Enum):
  65         EDIT_PACKAGE       = "EDIT_PACKAGE"
  66         APPROVE_CHANGES    = "APPROVE_CHANGES"
  67         DELETE_PACKAGE     = "DELETE_PACKAGE"
  68         CHANGE_AUTHOR      = "CHANGE_AUTHOR"
  69         MAKE_RELEASE       = "MAKE_RELEASE"
  70         ADD_SCREENSHOTS    = "ADD_SCREENSHOTS"
  71         APPROVE_SCREENSHOT = "APPROVE_SCREENSHOT"
  72         APPROVE_RELEASE    = "APPROVE_RELEASE"
  73         APPROVE_NEW        = "APPROVE_NEW"
  74         CHANGE_RELEASE_URL = "CHANGE_RELEASE_URL"
  75         CHANGE_DNAME       = "CHANGE_DNAME"
  76         CHANGE_RANK        = "CHANGE_RANK"
  77         CHANGE_EMAIL       = "CHANGE_EMAIL"
  78         EDIT_EDITREQUEST   = "EDIT_EDITREQUEST"
  79         SEE_THREAD         = "SEE_THREAD"
  80         CREATE_THREAD      = "CREATE_THREAD"
  81         UNAPPROVE_PACKAGE  = "UNAPPROVE_PACKAGE"
  82         TOPIC_DISCARD      = "TOPIC_DISCARD"
  83
  84         # Only return true if the permission is valid for *all* contexts
  85         # See Package.checkPerm for package-specific contexts
  86         def check(self, user):
  87                 if not user.is_authenticated:
  88                         return False
  89
  90                 if self == Permission.APPROVE_NEW or \
  91                                 self == Permission.APPROVE_CHANGES    or \
  92                                 self == Permission.APPROVE_RELEASE    or \
  93                                 self == Permission.APPROVE_SCREENSHOT or \
  94                                 self == Permission.SEE_THREAD:
  95                         return user.rank.atLeast(UserRank.EDITOR)
  96                 else:
  97                         raise Exception("Non-global permission checked globally. Use Package.checkPerm or User.checkPerm instead.")
  98
  99 class User(db.Model, UserMixin):
 100         id           = db.Column(db.Integer, primary_key=True)
 101
 102         # User authentication information
 103         username     = db.Column(db.String(50, collation="NOCASE"), nullable=False, unique=True, index=True)
 104         password     = db.Column(db.String(255), nullable=True)
 105         reset_password_token = db.Column(db.String(100), nullable=False, server_default="")
 106
 107         rank         = db.Column(db.Enum(UserRank))
 108
 109         # Account linking
 110         github_username = db.Column(db.String(50, collation="NOCASE"), nullable=True, unique=True)
 111         forums_username = db.Column(db.String(50, collation="NOCASE"), nullable=True, unique=True)
 112
 113         # User email information
 114         email         = db.Column(db.String(255), nullable=True, unique=True)
 115         confirmed_at  = db.Column(db.DateTime())
 116
 117         # User information
 118         profile_pic   = db.Column(db.String(255), nullable=True, server_default=None)
 119         active        = db.Column("is_active", db.Boolean, nullable=False, server_default="0")
 120         display_name  = db.Column(db.String(100), nullable=False, server_default="")
 121
 122         # Content
 123         notifications = db.relationship("Notification", primaryjoin="User.id==Notification.user_id")
 124
 125         # causednotifs  = db.relationship("Notification", backref="causer", lazy="dynamic")
 126         packages      = db.relationship("Package", backref="author", lazy="dynamic")
 127         requests      = db.relationship("EditRequest", backref="author", lazy="dynamic")
 128         threads       = db.relationship("Thread", backref="author", lazy="dynamic")
 129         replies       = db.relationship("ThreadReply", backref="author", lazy="dynamic")
 130
 131         def __init__(self, username, active=False, email=None, password=None):
 132                 import datetime
 133
 134                 self.username = username
 135                 self.confirmed_at = datetime.datetime.now() - datetime.timedelta(days=6000)
 136                 self.display_name = username
 137                 self.active = active
 138                 self.email = email
 139                 self.password = password
 140                 self.rank = UserRank.NOT_JOINED
 141
 142         def canAccessTodoList(self):
 143                 return Permission.APPROVE_NEW.check(self) or \
 144                                 Permission.APPROVE_RELEASE.check(self) or \
 145                                 Permission.APPROVE_CHANGES.check(self)
 146
 147         def isClaimed(self):
 148                 return self.rank.atLeast(UserRank.NEW_MEMBER)
 149
 150         def getProfilePicURL(self):
 151                 if self.profile_pic:
 152                         return self.profile_pic
 153                 else:
 154                         return gravatar(self.email or "")
 155
 156         def checkPerm(self, user, perm):
 157                 if not user.is_authenticated:
 158                         return False
 159
 160                 if type(perm) == str:
 161                         perm = Permission[perm]
 162                 elif type(perm) != Permission:
 163                         raise Exception("Unknown permission given to User.checkPerm()")
 164
 165                 # Members can edit their own packages, and editors can edit any packages
 166                 if perm == Permission.CHANGE_AUTHOR:
 167                         return user.rank.atLeast(UserRank.EDITOR)
 168                 elif perm == Permission.CHANGE_RANK or perm == Permission.CHANGE_DNAME:
 169                         return user.rank.atLeast(UserRank.MODERATOR)
 170                 elif perm == Permission.CHANGE_EMAIL:
 171                         return user == self or (user.rank.atLeast(UserRank.MODERATOR) and user.rank.atLeast(self.rank))
 172                 else:
 173                         raise Exception("Permission {} is not related to users".format(perm.name))
 174
 175 class UserEmailVerification(db.Model):
 176         id      = db.Column(db.Integer, primary_key=True)
 177         user_id = db.Column(db.Integer, db.ForeignKey("user.id"))
 178         email   = db.Column(db.String(100))
 179         token   = db.Column(db.String(32))
 180         user    = db.relationship("User", foreign_keys=[user_id])
 181
 182 class Notification(db.Model):
 183         id        = db.Column(db.Integer, primary_key=True)
 184         user_id   = db.Column(db.Integer, db.ForeignKey("user.id"))
 185         causer_id = db.Column(db.Integer, db.ForeignKey("user.id"))
 186         user      = db.relationship("User", foreign_keys=[user_id])
 187         causer    = db.relationship("User", foreign_keys=[causer_id])
 188
 189         title     = db.Column(db.String(100), nullable=False)
 190         url       = db.Column(db.String(200), nullable=True)
 191
 192         def __init__(self, us, cau, titl, ur):
 193                 self.user   = us
 194                 self.causer = cau
 195                 self.title  = titl
 196                 self.url    = ur
 197
 198
 199 class License(db.Model):
 200         id      = db.Column(db.Integer, primary_key=True)
 201         name    = db.Column(db.String(50), nullable=False, unique=True)
 202         is_foss = db.Column(db.Boolean,    nullable=False, default=True)
 203
 204         def __init__(self, v, is_foss=True):
 205                 self.name = v
 206                 self.is_foss = is_foss
 207
 208         def __str__(self):
 209                 return self.name
 210
 211
 212 class PackageType(enum.Enum):
 213         MOD  = "Mod"
 214         GAME = "Game"
 215         TXP  = "Texture Pack"
 216
 217         def toName(self):
 218                 return self.name.lower()
 219
 220         def __str__(self):
 221                 return self.name
 222
 223         @classmethod
 224         def get(cls, name):
 225                 try:
 226                         return PackageType[name.upper()]
 227                 except KeyError:
 228                         return None
 229
 230         @classmethod
 231         def choices(cls):
 232                 return [(choice, choice.value) for choice in cls]
 233
 234         @classmethod
 235         def coerce(cls, item):
 236                 return item if type(item) == PackageType else PackageType[item]
 237
 238
 239 class PackagePropertyKey(enum.Enum):
 240         name          = "Name"
 241         title         = "Title"
 242         shortDesc     = "Short Description"
 243         desc          = "Description"
 244         type          = "Type"
 245         license       = "License"
 246         media_license = "Media License"
 247         tags          = "Tags"
 248         provides      = "Provides"
 249         repo          = "Repository"
 250         website       = "Website"
 251         issueTracker  = "Issue Tracker"
 252         forums        = "Forum Topic ID"
 253
 254         def convert(self, value):
 255                 if self == PackagePropertyKey.tags:
 256                         return ",".join([t.title for t in value])
 257                 elif self == PackagePropertyKey.provides:
 258                         return ",".join([t.name for t in value])
 259                 else:
 260                         return str(value)
 261
 262 provides = db.Table("provides",
 263         db.Column("package_id",    db.Integer, db.ForeignKey("package.id"), primary_key=True),
 264     db.Column("metapackage_id", db.Integer, db.ForeignKey("meta_package.id"), primary_key=True)
 265 )
 266
 267 tags = db.Table("tags",
 268     db.Column("tag_id", db.Integer, db.ForeignKey("tag.id"), primary_key=True),
 269     db.Column("package_id", db.Integer, db.ForeignKey("package.id"), primary_key=True)
 270 )
 271
 272 class Dependency(db.Model):
 273         id              = db.Column(db.Integer, primary_key=True)
 274         depender_id     = db.Column(db.Integer, db.ForeignKey("package.id"),     nullable=True)
 275         package_id      = db.Column(db.Integer, db.ForeignKey("package.id"),     nullable=True)
 276         package         = db.relationship("Package", foreign_keys=[package_id])
 277         meta_package_id = db.Column(db.Integer, db.ForeignKey("meta_package.id"), nullable=True)
 278         optional        = db.Column(db.Boolean, nullable=False, default=False)
 279         __table_args__  = (db.UniqueConstraint('depender_id', 'package_id', 'meta_package_id', name='_dependency_uc'), )
 280
 281         def __init__(self, depender=None, package=None, meta=None):
 282                 if depender is None:
 283                         return
 284
 285                 self.depender = depender
 286
 287                 packageProvided = package is not None
 288                 metaProvided = meta is not None
 289
 290                 if packageProvided and not metaProvided:
 291                         self.package = package
 292                 elif metaProvided and not packageProvided:
 293                         self.meta_package = meta
 294                 else:
 295                         raise Exception("Either meta or package must be given, but not both!")
 296
 297         def __str__(self):
 298                 if self.package is not None:
 299                         return self.package.author.username + "/" + self.package.name
 300                 elif self.meta_package is not None:
 301                         return self.meta_package.name
 302                 else:
 303                         raise Exception("Meta and package are both none!")
 304
 305         @staticmethod
 306         def SpecToList(depender, spec, cache={}):
 307                 retval = []
 308                 arr = spec.split(",")
 309
 310                 import re
 311                 pattern1 = re.compile("^([a-z0-9_]+)$")
 312                 pattern2 = re.compile("^([A-Za-z0-9_]+)/([a-z0-9_]+)$")
 313
 314                 for x in arr:
 315                         x = x.strip()
 316                         if x == "":
 317                                 continue
 318
 319                         if pattern1.match(x):
 320                                 meta = MetaPackage.GetOrCreate(x, cache)
 321                                 retval.append(Dependency(depender, meta=meta))
 322                         else:
 323                                 m = pattern2.match(x)
 324                                 username = m.group(1)
 325                                 name     = m.group(2)
 326                                 user = User.query.filter_by(username=username).first()
 327                                 if user is None:
 328                                         raise Exception("Unable to find user " + username)
 329
 330                                 package = Package.query.filter_by(author=user, name=name).first()
 331                                 if package is None:
 332                                         raise Exception("Unable to find package " + name + " by " + username)
 333
 334                                 retval.append(Dependency(depender, package=package))
 335
 336                 return retval
 337
 338
 339
 340 class Package(db.Model):
 341         id           = db.Column(db.Integer, primary_key=True)
 342
 343         # Basic details
 344         author_id    = db.Column(db.Integer, db.ForeignKey("user.id"))
 345         name         = db.Column(db.String(100), nullable=False)
 346         title        = db.Column(db.String(100), nullable=False)
 347         shortDesc    = db.Column(db.String(200), nullable=False)
 348         desc         = db.Column(db.Text, nullable=True)
 349         type         = db.Column(db.Enum(PackageType))
 350         created_at   = db.Column(db.DateTime, nullable=False, default=datetime.utcnow)
 351
 352         license_id   = db.Column(db.Integer, db.ForeignKey("license.id"), nullable=False, default=1)
 353         license      = db.relationship("License", foreign_keys=[license_id])
 354         media_license_id = db.Column(db.Integer, db.ForeignKey("license.id"), nullable=False, default=1)
 355         media_license    = db.relationship("License", foreign_keys=[media_license_id])
 356
 357         approved     = db.Column(db.Boolean, nullable=False, default=False)
 358         soft_deleted = db.Column(db.Boolean, nullable=False, default=False)
 359
 360         score        = db.Column(db.Float, nullable=False, default=0)
 361
 362         review_thread_id = db.Column(db.Integer, db.ForeignKey("thread.id"), nullable=True, default=None)
 363         review_thread    = db.relationship("Thread", foreign_keys=[review_thread_id])
 364
 365         # Downloads
 366         repo         = db.Column(db.String(200), nullable=True)
 367         website      = db.Column(db.String(200), nullable=True)
 368         issueTracker = db.Column(db.String(200), nullable=True)
 369         forums       = db.Column(db.Integer,     nullable=True)
 370
 371         provides = db.relationship("MetaPackage", secondary=provides, lazy="subquery",
 372                         backref=db.backref("packages", lazy="dynamic"))
 373
 374         dependencies = db.relationship("Dependency", backref="depender", lazy="dynamic", foreign_keys=[Dependency.depender_id])
 375
 376         tags = db.relationship("Tag", secondary=tags, lazy="subquery",
 377                         backref=db.backref("packages", lazy=True))
 378
 379         releases = db.relationship("PackageRelease", backref="package",
 380                         lazy="dynamic", order_by=db.desc("package_release_releaseDate"))
 381
 382         screenshots = db.relationship("PackageScreenshot", backref="package",
 383                         lazy="dynamic", order_by=db.asc("package_screenshot_id"))
 384
 385         requests = db.relationship("EditRequest", backref="package",
 386                         lazy="dynamic")
 387
 388         def __init__(self, package=None):
 389                 if package is None:
 390                         return
 391
 392                 self.author_id = package.author_id
 393                 self.created_at = package.created_at
 394                 self.approved = package.approved
 395
 396                 for e in PackagePropertyKey:
 397                         setattr(self, e.name, getattr(package, e.name))
 398
 399         def getAsDictionaryShort(self, base_url):
 400                 tnurl = self.getThumbnailURL(1)
 401                 return {
 402                         "name": self.name,
 403                         "title": self.title,
 404                         "author": self.author.display_name,
 405                         "short_description": self.shortDesc,
 406                         "type": self.type.toName(),
 407                         "release": self.getDownloadRelease().id if self.getDownloadRelease() is not None else None,
 408                         "thumbnail": (base_url + tnurl) if tnurl is not None else None,
 409                         "score": round(self.score * 10) / 10
 410                 }
 411
 412         def getAsDictionary(self, base_url):
 413                 tnurl = self.getThumbnailURL(1)
 414                 return {
 415                         "author": self.author.display_name,
 416                         "name": self.name,
 417                         "title": self.title,
 418                         "short_description": self.shortDesc,
 419                         "desc": self.desc,
 420                         "type": self.type.toName(),
 421                         "created_at": self.created_at,
 422
 423                         "license": self.license.name,
 424                         "media_license": self.media_license.name,
 425
 426                         "repo": self.repo,
 427                         "website": self.website,
 428                         "issue_tracker": self.issueTracker,
 429                         "forums": self.forums,
 430
 431                         "provides": [x.name for x in self.provides],
 432                         "thumbnail": (base_url + tnurl) if tnurl is not None else None,
 433                         "screenshots": [base_url + ss.url for ss in self.screenshots],
 434
 435                         "url": base_url + self.getDownloadURL(),
 436                         "release": self.getDownloadRelease().id if self.getDownloadRelease() is not None else None,
 437
 438                         "score": round(self.score * 10) / 10
 439                 }
 440
 441         def getThumbnailURL(self, level=2):
 442                 screenshot = self.screenshots.filter_by(approved=True).order_by(db.asc(PackageScreenshot.id)).first()
 443                 return screenshot.getThumbnailURL(level) if screenshot is not None else None
 444
 445         def getMainScreenshotURL(self):
 446                 screenshot = self.screenshots.filter_by(approved=True).order_by(db.asc(PackageScreenshot.id)).first()
 447                 return screenshot.url if screenshot is not None else None
 448
 449         def getDetailsURL(self):
 450                 return url_for("package_page",
 451                                 author=self.author.username, name=self.name)
 452
 453         def getEditURL(self):
 454                 return url_for("create_edit_package_page",
 455                                 author=self.author.username, name=self.name)
 456
 457         def getApproveURL(self):
 458                 return url_for("approve_package_page",
 459                                 author=self.author.username, name=self.name)
 460
 461         def getRemoveURL(self):
 462                 return url_for("remove_package_page",
 463                                 author=self.author.username, name=self.name)
 464
 465         def getNewScreenshotURL(self):
 466                 return url_for("create_screenshot_page",
 467                                 author=self.author.username, name=self.name)
 468
 469         def getCreateReleaseURL(self):
 470                 return url_for("create_release_page",
 471                                 author=self.author.username, name=self.name)
 472
 473         def getCreateEditRequestURL(self):
 474                 return url_for("create_edit_editrequest_page",
 475                                 author=self.author.username, name=self.name)
 476
 477         def getDownloadURL(self):
 478                 return url_for("package_download_page",
 479                                 author=self.author.username, name=self.name)
 480
 481         def getDownloadRelease(self):
 482                 for rel in self.releases:
 483                         if rel.approved:
 484                                 return rel
 485
 486                 return None
 487
 488         def checkPerm(self, user, perm):
 489                 if not user.is_authenticated:
 490                         return False
 491
 492                 if type(perm) == str:
 493                         perm = Permission[perm]
 494                 elif type(perm) != Permission:
 495                         raise Exception("Unknown permission given to Package.checkPerm()")
 496
 497                 isOwner = user == self.author
 498
 499                 # Members can edit their own packages, and editors can edit any packages
 500                 if perm == Permission.MAKE_RELEASE or perm == Permission.ADD_SCREENSHOTS or perm == Permission.CREATE_THREAD:
 501                         return isOwner or user.rank.atLeast(UserRank.EDITOR)
 502
 503                 if perm == Permission.EDIT_PACKAGE or perm == Permission.APPROVE_CHANGES:
 504                         if isOwner:
 505                                 return user.rank.atLeast(UserRank.MEMBER if self.approved else UserRank.NEW_MEMBER)
 506                         else:
 507                                 return user.rank.atLeast(UserRank.EDITOR)
 508
 509                 # Editors can change authors and approve new packages
 510                 elif perm == Permission.APPROVE_NEW or perm == Permission.CHANGE_AUTHOR:
 511                         return user.rank.atLeast(UserRank.EDITOR)
 512
 513                 elif perm == Permission.APPROVE_RELEASE or perm == Permission.APPROVE_SCREENSHOT:
 514                         return user.rank.atLeast(UserRank.TRUSTED_MEMBER if isOwner else UserRank.EDITOR)
 515
 516                 # Moderators can delete packages
 517                 elif perm == Permission.DELETE_PACKAGE or perm == Permission.UNAPPROVE_PACKAGE \
 518                                 or perm == Permission.CHANGE_RELEASE_URL:
 519                         return user.rank.atLeast(UserRank.MODERATOR)
 520
 521                 else:
 522                         raise Exception("Permission {} is not related to packages".format(perm.name))
 523
 524         def recalcScore(self):
 525                 import datetime
 526
 527                 self.score = 10
 528
 529                 if self.forums is not None:
 530                         topic = ForumTopic.query.get(self.forums)
 531                         if topic:
 532                                 days   = (datetime.datetime.now() - topic.created_at).days
 533                                 months = days / 30
 534                                 years  = days / 365
 535                                 self.score = topic.views / max(years, 0.0416) + 80*min(max(months, 0.5), 6)
 536
 537                 if self.getMainScreenshotURL() is None:
 538                         self.score *= 0.8
 539
 540                 if not self.license.is_foss or not self.media_license.is_foss:
 541                         self.score *= 0.1
 542
 543 class MetaPackage(db.Model):
 544         id           = db.Column(db.Integer, primary_key=True)
 545         name         = db.Column(db.String(100), unique=True, nullable=False)
 546         dependencies = db.relationship("Dependency", backref="meta_package", lazy="dynamic")
 547
 548         def __init__(self, name=None):
 549                 self.name = name
 550
 551         def __str__(self):
 552                 return self.name
 553
 554         @staticmethod
 555         def ListToSpec(list):
 556                 return ",".join([str(x) for x in list])
 557
 558         @staticmethod
 559         def GetOrCreate(name, cache={}):
 560                 mp = cache.get(name)
 561                 if mp is None:
 562                         mp = MetaPackage.query.filter_by(name=name).first()
 563
 564                 if mp is None:
 565                         mp = MetaPackage(name)
 566                         db.session.add(mp)
 567
 568                 cache[name] = mp
 569                 return mp
 570
 571         @staticmethod
 572         def SpecToList(spec, cache={}):
 573                 retval = []
 574                 arr = spec.split(",")
 575
 576                 import re
 577                 pattern = re.compile("^([a-z0-9_]+)$")
 578
 579                 for x in arr:
 580                         x = x.strip()
 581                         if x == "":
 582                                 continue
 583
 584                         if not pattern.match(x):
 585                                 continue
 586
 587                         retval.append(MetaPackage.GetOrCreate(x, cache))
 588
 589                 return retval
 590
 591 class Tag(db.Model):
 592         id              = db.Column(db.Integer,    primary_key=True)
 593         name            = db.Column(db.String(100), unique=True, nullable=False)
 594         title           = db.Column(db.String(100), nullable=False)
 595         backgroundColor = db.Column(db.String(6),   nullable=False)
 596         textColor       = db.Column(db.String(6),   nullable=False)
 597
 598         def __init__(self, title, backgroundColor="000000", textColor="ffffff"):
 599                 self.title           = title
 600                 self.backgroundColor = backgroundColor
 601                 self.textColor       = textColor
 602
 603                 import re
 604                 regex = re.compile("[^a-z_]")
 605                 self.name = regex.sub("", self.title.lower().replace(" ", "_"))
 606
 607 class PackageRelease(db.Model):
 608         id           = db.Column(db.Integer, primary_key=True)
 609
 610         package_id   = db.Column(db.Integer, db.ForeignKey("package.id"))
 611         title        = db.Column(db.String(100), nullable=False)
 612         releaseDate  = db.Column(db.DateTime,        nullable=False)
 613         url          = db.Column(db.String(200), nullable=False)
 614         approved     = db.Column(db.Boolean, nullable=False, default=False)
 615         task_id      = db.Column(db.String(37), nullable=True)
 616         commit_hash  = db.Column(db.String(41), nullable=True, default=None)
 617
 618
 619         def getEditURL(self):
 620                 return url_for("edit_release_page",
 621                                 author=self.package.author.username,
 622                                 name=self.package.name,
 623                                 id=self.id)
 624
 625         def getDownloadURL(self):
 626                 return url_for("download_release_page",
 627                                 author=self.package.author.username,
 628                                 name=self.package.name,
 629                                 id=self.id)
 630
 631
 632         def __init__(self):
 633                 self.releaseDate = datetime.now()
 634
 635
 636 class PackageReview(db.Model):
 637         id         = db.Column(db.Integer, primary_key=True)
 638         package_id = db.Column(db.Integer, db.ForeignKey("package.id"))
 639         thread_id  = db.Column(db.Integer, db.ForeignKey("thread.id"), nullable=False)
 640         recommend  = db.Column(db.Boolean, nullable=False, default=True)
 641
 642
 643 class PackageScreenshot(db.Model):
 644         id         = db.Column(db.Integer, primary_key=True)
 645         package_id = db.Column(db.Integer, db.ForeignKey("package.id"))
 646         title      = db.Column(db.String(100), nullable=False)
 647         url        = db.Column(db.String(100), nullable=False)
 648         approved   = db.Column(db.Boolean, nullable=False, default=False)
 649
 650
 651         def getEditURL(self):
 652                 return url_for("edit_screenshot_page",
 653                                 author=self.package.author.username,
 654                                 name=self.package.name,
 655                                 id=self.id)
 656
 657         def getThumbnailURL(self, level=2):
 658                 return self.url.replace("/uploads/", ("/thumbnails/{:d}/").format(level))
 659
 660
 661
 662 class EditRequest(db.Model):
 663         id           = db.Column(db.Integer, primary_key=True)
 664
 665         package_id   = db.Column(db.Integer, db.ForeignKey("package.id"))
 666         author_id    = db.Column(db.Integer, db.ForeignKey("user.id"))
 667
 668         title        = db.Column(db.String(100), nullable=False)
 669         desc         = db.Column(db.String(1000), nullable=True)
 670
 671         # 0 - open
 672         # 1 - merged
 673         # 2 - rejected
 674         status       = db.Column(db.Integer, nullable=False, default=0)
 675
 676         changes = db.relationship("EditRequestChange", backref="request",
 677                         lazy="dynamic")
 678
 679         def getURL(self):
 680                 return url_for("view_editrequest_page",
 681                                 author=self.package.author.username,
 682                                 name=self.package.name,
 683                                 id=self.id)
 684
 685         def getApproveURL(self):
 686                 return url_for("approve_editrequest_page",
 687                                 author=self.package.author.username,
 688                                 name=self.package.name,
 689                                 id=self.id)
 690
 691         def getRejectURL(self):
 692                 return url_for("reject_editrequest_page",
 693                                 author=self.package.author.username,
 694                                 name=self.package.name,
 695                                 id=self.id)
 696
 697         def getEditURL(self):
 698                 return url_for("create_edit_editrequest_page",
 699                                 author=self.package.author.username,
 700                                 name=self.package.name,
 701                                 id=self.id)
 702
 703         def applyAll(self, package):
 704                 for change in self.changes:
 705                         change.apply(package)
 706
 707
 708         def checkPerm(self, user, perm):
 709                 if not user.is_authenticated:
 710                         return False
 711
 712                 if type(perm) == str:
 713                         perm = Permission[perm]
 714                 elif type(perm) != Permission:
 715                         raise Exception("Unknown permission given to EditRequest.checkPerm()")
 716
 717                 isOwner = user == self.author
 718
 719                 # Members can edit their own packages, and editors can edit any packages
 720                 if perm == Permission.EDIT_EDITREQUEST:
 721                         return isOwner or user.rank.atLeast(UserRank.EDITOR)
 722
 723                 else:
 724                         raise Exception("Permission {} is not related to packages".format(perm.name))
 725
 726
 727
 728
 729 class EditRequestChange(db.Model):
 730         id           = db.Column(db.Integer, primary_key=True)
 731
 732         request_id   = db.Column(db.Integer, db.ForeignKey("edit_request.id"))
 733         key          = db.Column(db.Enum(PackagePropertyKey), nullable=False)
 734
 735         # TODO: make diff instead
 736         oldValue     = db.Column(db.Text, nullable=True)
 737         newValue     = db.Column(db.Text, nullable=True)
 738
 739         def apply(self, package):
 740                 if self.key == PackagePropertyKey.tags:
 741                         package.tags.clear()
 742                         for tagTitle in self.newValue.split(","):
 743                                 tag = Tag.query.filter_by(title=tagTitle.strip()).first()
 744                                 package.tags.append(tag)
 745
 746                 else:
 747                         setattr(package, self.key.name, self.newValue)
 748
 749
 750 watchers = db.Table("watchers",
 751     db.Column("user_id", db.Integer, db.ForeignKey("user.id"), primary_key=True),
 752     db.Column("thread_id", db.Integer, db.ForeignKey("thread.id"), primary_key=True)
 753 )
 754
 755 class Thread(db.Model):
 756         id         = db.Column(db.Integer, primary_key=True)
 757
 758         package_id = db.Column(db.Integer, db.ForeignKey("package.id"), nullable=True)
 759         package    = db.relationship("Package", foreign_keys=[package_id])
 760
 761         author_id  = db.Column(db.Integer, db.ForeignKey("user.id"), nullable=False)
 762         title      = db.Column(db.String(100), nullable=False)
 763         private    = db.Column(db.Boolean, server_default="0")
 764
 765         created_at = db.Column(db.DateTime, nullable=False, default=datetime.utcnow)
 766
 767         replies    = db.relationship("ThreadReply", backref="thread", lazy="dynamic")
 768
 769         watchers   = db.relationship("User", secondary=watchers, lazy="subquery", \
 770                                                 backref=db.backref("watching", lazy=True))
 771
 772
 773         def getSubscribeURL(self):
 774                 return url_for("thread_subscribe_page",
 775                                 id=self.id)
 776
 777         def getUnsubscribeURL(self):
 778                 return url_for("thread_unsubscribe_page",
 779                                 id=self.id)
 780
 781         def checkPerm(self, user, perm):
 782                 if not user.is_authenticated:
 783                         return not self.private
 784
 785                 if type(perm) == str:
 786                         perm = Permission[perm]
 787                 elif type(perm) != Permission:
 788                         raise Exception("Unknown permission given to Thread.checkPerm()")
 789
 790                 isOwner = user == self.author or (self.package is not None and self.package.author == user)
 791
 792                 if perm == Permission.SEE_THREAD:
 793                         return not self.private or isOwner or user.rank.atLeast(UserRank.EDITOR)
 794
 795                 else:
 796                         raise Exception("Permission {} is not related to threads".format(perm.name))
 797
 798 class ThreadReply(db.Model):
 799         id         = db.Column(db.Integer, primary_key=True)
 800         thread_id  = db.Column(db.Integer, db.ForeignKey("thread.id"), nullable=False)
 801         comment    = db.Column(db.String(500), nullable=False)
 802         author_id  = db.Column(db.Integer, db.ForeignKey("user.id"), nullable=False)
 803         created_at = db.Column(db.DateTime, nullable=False, default=datetime.utcnow)
 804
 805
 806 REPO_BLACKLIST = [".zip", "mediafire.com", "dropbox.com", "weebly.com", \
 807                 "minetest.net", "dropboxusercontent.com", "4shared.com", \
 808                 "digitalaudioconcepts.com", "hg.intevation.org", "www.wtfpl.net", \
 809                 "imageshack.com", "imgur.com"]
 810
 811 class ForumTopic(db.Model):
 812         topic_id  = db.Column(db.Integer, primary_key=True, autoincrement=False)
 813         author_id = db.Column(db.Integer, db.ForeignKey("user.id"), nullable=False)
 814         author    = db.relationship("User")
 815
 816         wip       = db.Column(db.Boolean, server_default="0")
 817         discarded = db.Column(db.Boolean, server_default="0")
 818
 819         type      = db.Column(db.Enum(PackageType), nullable=False)
 820         title     = db.Column(db.String(200), nullable=False)
 821         name      = db.Column(db.String(30), nullable=True)
 822         link      = db.Column(db.String(200), nullable=True)
 823
 824         posts     = db.Column(db.Integer, nullable=False)
 825         views     = db.Column(db.Integer, nullable=False)
 826
 827         created_at = db.Column(db.DateTime, nullable=False, default=datetime.utcnow)
 828
 829         def getRepoURL(self):
 830                 if self.link is None:
 831                         return None
 832
 833                 for item in REPO_BLACKLIST:
 834                         if item in self.link:
 835                                 return None
 836
 837                 return self.link.replace("repo.or.cz/w/", "repo.or.cz/")
 838
 839         def getAsDictionary(self):
 840                 return {
 841                         "author": self.author.username,
 842                         "name":   self.name,
 843                         "type":   self.type.toName(),
 844                         "title":  self.title,
 845                         "id":     self.topic_id,
 846                         "link":   self.link,
 847                         "posts":  self.posts,
 848                         "views":  self.views,
 849                         "is_wip": self.wip,
 850                         "discarded":  self.discarded,
 851                         "created_at": self.created_at.isoformat(),
 852                 }
 853
 854         def checkPerm(self, user, perm):
 855                 if not user.is_authenticated:
 856                         return False
 857
 858                 if type(perm) == str:
 859                         perm = Permission[perm]
 860                 elif type(perm) != Permission:
 861                         raise Exception("Unknown permission given to ForumTopic.checkPerm()")
 862
 863                 if perm == Permission.TOPIC_DISCARD:
 864                         return self.author == user or user.rank.atLeast(UserRank.EDITOR)
 865
 866                 else:
 867                         raise Exception("Permission {} is not related to topics".format(perm.name))
 868
 869
 870 # Setup Flask-User
 871 db_adapter = SQLAlchemyAdapter(db, User)        # Register the User model
 872 user_manager = UserManager(db_adapter, app)     # Initialize Flask-User