Can vote on Comment

src/main/kotlin/fr/dcproject/Application.kt

@@ -164,7 +164,7 @@ fun Application.module(env: Env = PROD) {
             comment(get())
             commentArticle(get())
             commentConstitution(get())
-            voteArticle(get())
+            voteArticle(get(), get(), get())
             voteConstitution(get())
         }
     }

src/main/kotlin/fr/dcproject/Module.kt

@@ -15,6 +15,7 @@ import fr.dcproject.repository.FollowArticle as FollowArticleRepository
 import fr.dcproject.repository.FollowConstitution as FollowConstitutionRepository
 import fr.dcproject.repository.User as UserRepository
 import fr.dcproject.repository.VoteArticle as VoteArticleRepository
+import fr.dcproject.repository.VoteArticleComment as VoteArticleCommentRepository
 import fr.dcproject.repository.VoteConstitution as VoteConstitutionRepository
 
 val config = Config()
@@ -43,6 +44,7 @@ val Module = module {
     single { CommentConstitutionRepository(get()) }
     single { VoteArticleRepository(get()) }
     single { VoteConstitutionRepository(get()) }
+    single { VoteArticleCommentRepository(get()) }
 
     single { Migrations(connection = get(), directory = config.sqlFiles) }
 }

src/main/kotlin/fr/dcproject/repository/Vote.kt

@@ -2,6 +2,7 @@ package fr.dcproject.repository
 
 import com.fasterxml.jackson.core.type.TypeReference
 import fr.dcproject.entity.Article
+import fr.dcproject.entity.Comment
 import fr.dcproject.entity.Constitution
 import fr.dcproject.entity.VoteAggregation
 import fr.postgresjson.connexion.Paginated
@@ -17,7 +18,14 @@ open class Vote <T: UuidEntity>(override var requester: Requester): RepositoryI<
     override val entityName = VoteEntity::class as KClass<VoteEntity<T>>
 
     fun vote(vote: VoteEntity<T>): VoteAggregation {
-        val reference = vote.target::class.simpleName!!.toLowerCase()
+        val target = vote.target
+        val reference = if (target is Comment<*>) {
+            target::class.simpleName!!.toLowerCase() +
+            "_on_" +
+            target.target::class.simpleName!!.toLowerCase()
+        } else {
+            target::class.simpleName!!.toLowerCase()
+        }
         val author = vote.createdBy ?: error("vote must be contain an author")
         val anonymous = author.voteAnonymous
         return requester
@@ -78,6 +86,21 @@ class VoteArticle (requester: Requester): Vote<Article>(requester) {
         )
 }
 
+class VoteArticleComment (requester: Requester): Vote<Comment<Article>>(requester) {
+    fun findByCitizen(
+        citizen: CitizenEntity,
+        page: Int = 1,
+        limit: Int = 50
+    ): Paginated<VoteEntity<Comment<Article>>> =
+        findByCitizen(
+            citizen.id ?: error("The citizen must have an id"),
+            "article",
+            object: TypeReference<List<VoteEntity<Comment<Article>>>>() {},
+            page,
+            limit
+        )
+}
+
 class VoteConstitution (requester: Requester): Vote<Constitution>(requester) {
     fun findByCitizen(
         citizen: CitizenEntity,

src/main/kotlin/fr/dcproject/routes/VoteArticle.kt

@@ -2,7 +2,10 @@ package fr.dcproject.routes
 
 import fr.dcproject.citizen
 import fr.dcproject.entity.Citizen
-import fr.dcproject.routes.VoteArticlePaths.ArticleVoteRequest.Content
+import fr.dcproject.repository.CommentArticle
+import fr.dcproject.repository.VoteArticleComment
+import fr.dcproject.routes.VoteArticlePaths.ArticleCommentVoteRequest
+import fr.dcproject.routes.VoteArticlePaths.ArticleVoteRequest
 import fr.dcproject.security.voter.VoteVoter.Action.CREATE
 import fr.dcproject.security.voter.VoteVoter.Action.VIEW
 import fr.dcproject.security.voter.assertCan
@@ -27,6 +30,11 @@ object VoteArticlePaths {
         data class Content(var note: Int)
     }
 
+    @Location("/articles/{article}/comments/{comment}/vote")
+    class ArticleCommentVoteRequest(val article: ArticleEntity, val comment: UUID) {
+        data class Content(var note: Int)
+    }
+
     @Location("/citizens/{citizen}/votes/articles")
     class CitizenVoteArticleRequest(
         val citizen: Citizen,
@@ -45,9 +53,9 @@ object VoteArticlePaths {
 }
 
 @KtorExperimentalLocationsAPI
-fun Route.voteArticle(repo: VoteArticleRepository) {
-    put<VoteArticlePaths.ArticleVoteRequest> {
-        val content = call.receive<Content>()
+fun Route.voteArticle(repo: VoteArticleRepository, voteCommentRepo: VoteArticleComment, commentRepo: CommentArticle) {
+    put<ArticleVoteRequest> {
+        val content = call.receive<ArticleVoteRequest.Content>()
         val vote = VoteEntity(
             target = it.article,
             note = content.note,
@@ -58,6 +66,19 @@ fun Route.voteArticle(repo: VoteArticleRepository) {
         call.respond(HttpStatusCode.Created, votes)
     }
 
+    put<ArticleCommentVoteRequest> {
+        val comment = commentRepo.findById(it.comment)!!
+        val content = call.receive<ArticleCommentVoteRequest.Content>()
+        val vote = VoteEntity(
+            target = comment,
+            note = content.note,
+            createdBy = this.citizen
+        )
+        assertCan(CREATE, vote)
+        val votes = voteCommentRepo.vote(vote)
+        call.respond(HttpStatusCode.Created, votes)
+    }
+
     get<VoteArticlePaths.CitizenVoteArticleRequest> {
         val votes = repo.findByCitizen(it.citizen, it.page, it.limit)
         assertCan(VIEW, votes.result)