1 /*
2  *  Make.org Core API
3  *  Copyright (C) 2018 Make.org
4  *
5  * This program is free software: you can redistribute it and/or modify
6  *  it under the terms of the GNU Affero General Public License as
7  *  published by the Free Software Foundation, either version 3 of the
8  *  License, or (at your option) any later version.
9  *
10  *  This program is distributed in the hope that it will be useful,
11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  *  GNU Affero General Public License for more details.
14  *
15  *  You should have received a copy of the GNU Affero General Public License
16  *  along with this program.  If not, see <https://www.gnu.org/licenses/>.
17  *
18  */
19 
20 package org.make.api
21 
22 import scala.io.Source
23 import com.github.tototoshi.csv.CSVReader
24 import org.make.core.proposal.{ProposalId, ProposalKeywordKey, QualificationKey, VoteKey}
25 
26 import java.time.ZonedDateTime
27 import org.make.core.proposal.VoteKey.{Agree, Disagree, Neutral}
28 import org.make.core.user.UserId
29 import org.make.core.proposal._
30 import org.make.core.proposal.QualificationKey._
31 import org.make.core.proposal.indexed.{IndexedProposal, IndexedProposalKeyword}
32 
33 object ProposalsUtils {
34 
35   private def parseVote(values: Map[String, String], key: VoteKey): Vote =
36     Vote(
37       key = key,
38       count = values(s"vote_verified_${key.value}").toInt,
39       countVerified = values(s"vote_verified_${key.value}").toInt,
40       countSequence = values(s"vote_sequence_${key.value}").toInt,
41       countSegment = 0,
42       Seq.empty
43     )
44 
45   private def parseQualification(
46     values: Map[String, String],
47     voteKey: VoteKey,
48     qualificationKey: QualificationKey,
49     label: String
50   ): Qualification =
51     Qualification(
52       key = qualificationKey,
53       count = values(s"vote_verified_${voteKey.value}_$label").toInt,
54       countVerified = values(s"vote_verified_${voteKey.value}_$label").toInt,
55       countSequence = values(s"vote_sequence_${voteKey.value}_$label").toInt,
56       countSegment = 0
57     )
58 
59   private def parseVotes(values: Map[String, String]): VotingOptions =
60     VotingOptions(
61       agreeVote = AgreeWrapper(
62         vote = parseVote(values, Agree),
63         likeIt = parseQualification(values, Agree, LikeIt, "like"),
64         platitudeAgree = parseQualification(values, Agree, PlatitudeAgree, "platitudeagree"),
65         doable = parseQualification(values, Agree, Doable, "doable")
66       ),
67       neutralVote = NeutralWrapper(
68         vote = parseVote(values, Neutral),
69         noOpinion = parseQualification(values, Neutral, NoOpinion, "noopinion"),
70         doNotUnderstand = parseQualification(values, Neutral, DoNotUnderstand, "donotunderstand"),
71         doNotCare = parseQualification(values, Neutral, DoNotCare, "donotcare")
72       ),
73       disagreeVote = DisagreeWrapper(
74         vote = parseVote(values, Disagree),
75         impossible = parseQualification(values, Disagree, Impossible, "impossible"),
76         noWay = parseQualification(values, Disagree, NoWay, "noway"),
77         platitudeDisagree = parseQualification(values, Disagree, PlatitudeDisagree, "platitudedisagree")
78       )
79     )
80 
81   private def parseProposal(values: Map[String, String]): IndexedProposal = {
82     TestUtils.indexedProposal(
83       id = ProposalId(values("id")),
84       userId = UserId(values("id")),
85       votingOptions = this.parseVotes(values),
86       createdAt = ZonedDateTime.parse(values("date_created")),
87       keywords = values("keywords")
88         .split('|')
89         .map(k => IndexedProposalKeyword(ProposalKeywordKey(k), k))
90         .toSeq
91     )
92   }
93 
94   def getProposalsFromCsv(csvName: String): Seq[IndexedProposal] = {
95     CSVReader
96       .open(Source.fromResource(csvName))
97       .iteratorWithHeaders
98       .map(parseProposal)
99       .toSeq
100   }
101 
102   private val maxVotes = 2000
103 
104   def getNewAndTestedProposals(csvName: String): (Seq[IndexedProposal], Seq[IndexedProposal]) = {
105     getProposalsFromCsv(csvName)
106       .filter(_.votesSequenceCount < maxVotes)
107       .partition(_.votesSequenceCount <= 10)
108   }
109 }