DC/dc-project
1
1
Fork 0
Code Issues 25 Pull Requests 2 Actions Packages Projects Releases Wiki Activity

Add security for follow

Browse Source
This commit is contained in:
Fabrice Lecomte
2019-08-31 00:14:05 +02:00
parent 52dfaaf814
commit cb91c50e58
8 changed files with 75 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/main/kotlin/fr/dcproject/Application.kt
View File

@@ -109,7 +109,8 @@ fun Application.module(env: Env = PROD) {
ConstitutionVoter(),
CitizenVoter(),
CommentVoter(),
VoteVoter()
VoteVoter(),
FollowVoter()
)
}

4
src/main/kotlin/fr/dcproject/entity/Citizen.kt
View File

@@ -9,8 +9,8 @@ class Citizen(
var name: Name?,
var birthday: DateTime?,
var userId: UUID? = null,
var voteanonymous: Boolean? = null,
var followanonymous: Boolean? = null,
var voteAnonymous: Boolean = true,
var followAnonymous: Boolean = true,
var user: User?
) : UuidEntity(id),
EntityCreatedAt by EntityCreatedAtImp(),

2
src/main/kotlin/fr/dcproject/repository/Vote.kt
View File

@@ -14,7 +14,7 @@ open class Vote <T: UuidEntity>(override var requester: Requester): RepositoryI<
fun vote(vote: VoteEntity<T>) {
val reference = vote.target::class.simpleName!!.toLowerCase()
val author = vote.createdBy ?: error("vote must be contain an author")
val anonymous = author.voteanonymous
val anonymous = author.voteAnonymous
requester
.getFunction("vote")
.sendQuery(

11
src/main/kotlin/fr/dcproject/routes/FollowArticle.kt
View File

@@ -2,6 +2,8 @@ package fr.dcproject.routes
import fr.dcproject.citizen
import fr.dcproject.entity.Citizen
import fr.dcproject.security.voter.FollowVoter.Action.*
import fr.dcproject.security.voter.assertCan
import io.ktor.application.call
import io.ktor.http.HttpStatusCode
import io.ktor.locations.*
@@ -21,24 +23,21 @@ object FollowArticlePaths {
fun Route.followArticle(repo: FollowArticleRepository) {
post<FollowArticlePaths.ArticleFollowRequest> {
val follow = FollowEntity(target = it.article, createdBy = this.citizen)
// TODO create voter
// assertCan(FollowVoter.Action.CREATE, follow)
assertCan(CREATE, follow)
repo.follow(follow)
call.respond(HttpStatusCode.Created)
}
delete<FollowArticlePaths.ArticleFollowRequest> {
val follow = FollowEntity(target = it.article, createdBy = this.citizen)
// TODO create voter
// assertCan(FollowVoter.Action.DELETE, follow)
assertCan(DELETE, follow)
repo.unfollow(follow)
call.respond(HttpStatusCode.NoContent)
}
get<FollowArticlePaths.CitizenFollowArticleRequest> {
val follows = repo.findByCitizen(it.citizen)
// TODO add security
// assertCan(FollowVoter.Action.VIEW, follows)
assertCan(VIEW, follows.result)
call.respond(follows)
}
}

11
src/main/kotlin/fr/dcproject/routes/FollowConstitution.kt
View File

@@ -2,6 +2,8 @@ package fr.dcproject.routes
import fr.dcproject.citizen
import fr.dcproject.entity.Citizen
import fr.dcproject.security.voter.FollowVoter.Action.*
import fr.dcproject.security.voter.assertCan
import io.ktor.application.call
import io.ktor.http.HttpStatusCode
import io.ktor.locations.*
@@ -21,24 +23,21 @@ object FollowConstitutionPaths {
fun Route.followConstitution(repo: FollowConstitutionRepository) {
post<FollowConstitutionPaths.ConstitutionFollowRequest> {
val follow = FollowEntity(target = it.constitution, createdBy = this.citizen)
// TODO create voter
// assertCan(FollowVoter.Action.CREATE, follow)
assertCan(CREATE, follow)
repo.follow(follow)
call.respond(HttpStatusCode.Created)
}
delete<FollowConstitutionPaths.ConstitutionFollowRequest> {
val follow = FollowEntity(target = it.constitution, createdBy = this.citizen)
// TODO create voter
// assertCan(FollowVoter.Action.DELETE, follow)
assertCan(DELETE, follow)
repo.unfollow(follow)
call.respond(HttpStatusCode.NoContent)
}
get<FollowConstitutionPaths.CitizenFollowConstitutionRequest> {
val follows = repo.findByCitizen(it.citizen)
// TODO create voter
// assertCan(FollowVoter.Action.VIEW, follows)
assertCan(VIEW, follows.result)
call.respond(follows)
}
}

53
src/main/kotlin/fr/dcproject/security/voter/FollowVoter.kt Normal file
View File

@@ -0,0 +1,53 @@
package fr.dcproject.security.voter
import io.ktor.application.ApplicationCall
import fr.dcproject.entity.Follow as FollowEntity
import fr.dcproject.entity.User as UserEntity
class FollowVoter: Voter {
enum class Action: ActionI {
CREATE,
DELETE,
VIEW
}
override fun supports(action: ActionI, call: ApplicationCall, subject: Any?): Boolean {
return (action is Action) &&
(subject is List<*> || subject is FollowEntity<*>?)
}
override fun vote(action: ActionI, call: ApplicationCall, subject: Any?): Vote {
val user = call.user
if (action == Action.CREATE) {
return if (user != null) Vote.GRANTED
else Vote.DENIED
}
if (action == Action.DELETE) {
return if (user != null) Vote.GRANTED
else Vote.DENIED
}
if (action == Action.VIEW) {
if (subject is FollowEntity<*>) {
return voteView(user, subject)
}
if (subject is List<*>) {
subject.forEach {
if (it !is FollowEntity<*> || voteView(user, it) == Vote.DENIED) {
return Vote.DENIED
}
}
return Vote.GRANTED
}
return Vote.DENIED
}
return Vote.ABSTAIN
}
private fun voteView(user: UserEntity?, subject: FollowEntity<*>): Vote {
return if ((user != null && subject.createdBy?.user?.id == user.id) || subject.createdBy?.followAnonymous == false) Vote.GRANTED
else Vote.DENIED
}
}