/* Copyright 2017-2023 CAcert Inc. SPDX-License-Identifier: Apache-2.0 Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0 Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License. */ package models import ( "context" "database/sql" "database/sql/driver" "errors" "fmt" "strings" "time" "github.com/jmoiron/sqlx" ) type VoteType struct { label string id uint8 } var ( VoteTypeMotion = &VoteType{label: "motion", id: 0} VoteTypeVeto = &VoteType{label: "veto", id: 1} ) func (v *VoteType) String() string { return v.label } func VoteTypeFromString(label string) (*VoteType, error) { for _, vt := range []*VoteType{VoteTypeMotion, VoteTypeVeto} { if strings.EqualFold(vt.label, label) { return vt, nil } } return nil, fmt.Errorf("unknown vote type %s", label) } func VoteTypeFromInt(id int64) (*VoteType, error) { for _, vt := range []*VoteType{VoteTypeMotion, VoteTypeVeto} { if int64(vt.id) == id { return vt, nil } } return nil, fmt.Errorf("unknown vote type id %d", id) } func (v *VoteType) Scan(src any) error { value, ok := src.(int64) if !ok { return fmt.Errorf("could not cast %v of %T to uint8", src, src) } vt, err := VoteTypeFromInt(value) if err != nil { return err } *v = *vt return nil } func (v *VoteType) Value() (driver.Value, error) { return int64(v.id), nil } func (v *VoteType) QuorumAndMajority() (int, float32) { const ( majorityDefault = 0.99 majorityMotion = 0.50 quorumDefault = 1 quorumMotion = 3 ) if v == VoteTypeMotion { return quorumMotion, majorityMotion } return quorumDefault, majorityDefault } type VoteStatus struct { Label string ID int8 } var ( voteStatusDeclined = &VoteStatus{Label: "declined", ID: -1} voteStatusPending = &VoteStatus{Label: "pending", ID: 0} voteStatusApproved = &VoteStatus{Label: "approved", ID: 1} voteStatusWithdrawn = &VoteStatus{Label: "withdrawn", ID: -2} ) func VoteStatusFromInt(id int64) (*VoteStatus, error) { for _, vs := range []*VoteStatus{voteStatusPending, voteStatusApproved, voteStatusWithdrawn, voteStatusDeclined} { if int64(vs.ID) == id { return vs, nil } } return nil, fmt.Errorf("unknown vote status id %d", id) } func (v *VoteStatus) String() string { return v.Label } func (v *VoteStatus) Scan(src any) error { value, ok := src.(int64) if !ok { return fmt.Errorf("could not cast %v of %T to uint8", src, src) } vs, err := VoteStatusFromInt(value) if err != nil { return err } *v = *vs return nil } func (v *VoteStatus) Value() (driver.Value, error) { return int64(v.ID), nil } type VoteChoice struct { Label string ID int8 } var ( VoteAye = &VoteChoice{Label: "aye", ID: 1} VoteNaye = &VoteChoice{Label: "naye", ID: -1} VoteAbstain = &VoteChoice{Label: "abstain", ID: 0} ) func VoteChoiceFromString(label string) (*VoteChoice, error) { for _, vc := range []*VoteChoice{VoteAye, VoteNaye, VoteAbstain} { if strings.EqualFold(vc.Label, label) { return vc, nil } } return nil, fmt.Errorf("unknown vote choice %s", label) } func VoteChoiceFromInt(id int64) (*VoteChoice, error) { for _, vc := range []*VoteChoice{VoteAye, VoteNaye, VoteAbstain} { if int64(vc.ID) == id { return vc, nil } } return nil, fmt.Errorf("unknown vote type id %d", id) } func (v *VoteChoice) String() string { return v.Label } func (v *VoteChoice) Scan(src any) error { value, ok := src.(int64) if !ok { return fmt.Errorf("could not cast %v of %T to uint8", src, src) } vc, err := VoteChoiceFromInt(value) if err != nil { return err } *v = *vc return nil } func (v *VoteChoice) Value() (driver.Value, error) { return int64(v.ID), nil } func (v *VoteChoice) Equal(other *VoteChoice) bool { return v.ID == other.ID } type VoteSums struct { Ayes, Nayes, Abstains int } func (v *VoteSums) VoteCount() int { return v.Ayes + v.Nayes + v.Abstains } func (v *VoteSums) TotalVotes() int { return v.Ayes + v.Nayes } func (v *VoteSums) Percent() int { totalVotes := v.TotalVotes() if totalVotes == 0 { return 0 } return v.Ayes * 100 / totalVotes } func (v *VoteSums) CalculateResult(quorum int, majority float32) (*VoteStatus, string) { if v.VoteCount() < quorum { return voteStatusDeclined, fmt.Sprintf("Needed quorum of %d has not been reached.", quorum) } if (float32(v.Ayes) / float32(v.TotalVotes())) < majority { return voteStatusDeclined, fmt.Sprintf("Needed majority of %0.2f%% has not been reached.", majority) } return voteStatusApproved, "Quorum and majority have been reached" } type Motion struct { ID int64 `db:"id"` Proposed time.Time `db:"proposed"` Proponent int64 `db:"proponent"` Proposer string `db:"proposer"` Title string `db:"title"` Content string `db:"content"` Status *VoteStatus `db:"status"` Due time.Time `db:"due"` Modified time.Time `db:"modified"` Tag string `db:"tag"` Type *VoteType `db:"votetype"` Sums *VoteSums `db:"-"` Votes []*Vote `db:"-"` Reasoning string `db:"-"` } type MotionModel struct { DB *sqlx.DB } // Create a new decision. func (m *MotionModel) Create( ctx context.Context, proponent *User, voteType *VoteType, title, content string, proposed, due time.Time, ) (int64, error) { d := &Motion{ Proposed: proposed.UTC(), Proponent: proponent.ID, Title: title, Content: content, Due: due.UTC(), Type: voteType, Status: voteStatusPending, } result, err := m.DB.NamedExecContext( ctx, `INSERT INTO decisions (proposed, proponent, title, content, votetype, status, due, modified, tag) VALUES (:proposed, :proponent, :title, :content, :votetype, :status, :due, :proposed, 'm' || STRFTIME('%Y%m%d', :proposed) || '.' || ( SELECT COUNT(*)+1 AS num FROM decisions WHERE proposed BETWEEN DATE(:proposed) AND DATE(:proposed, '1 day') ))`, d, ) if err != nil { return 0, fmt.Errorf("creating motion failed: %w", err) } id, err := result.LastInsertId() if err != nil { return 0, fmt.Errorf("could not get inserted decision id: %w", err) } return id, nil } func (m *MotionModel) CloseDecisions(ctx context.Context) ([]*Motion, error) { tx, err := m.DB.BeginTxx(ctx, nil) if err != nil { return nil, errCouldNotStartTransaction(err) } defer func(tx *sqlx.Tx) { _ = tx.Rollback() }(tx) rows, err := tx.NamedQuery(` SELECT decisions.id, decisions.tag, decisions.proponent, decisions.proposed, decisions.title, decisions.content, decisions.votetype, decisions.status, decisions.due, decisions.modified FROM decisions WHERE decisions.status=0 AND :now > due`, struct{ Now time.Time }{Now: time.Now().UTC()}) if err != nil { return nil, fmt.Errorf("fetching closable decisions failed: %w", err) } defer func() { _ = rows.Close() }() decisions := make([]*Motion, 0) for rows.Next() { decision := &Motion{} if err = rows.StructScan(decision); err != nil { return nil, errCouldNotScanResult(err) } if rows.Err() != nil { return nil, fmt.Errorf("row error: %w", err) } decisions = append(decisions, decision) } results := make([]*Motion, 0, len(decisions)) var decisionResult *Motion for _, decision := range decisions { if decisionResult, err = closeDecision(ctx, tx, decision); err != nil { return nil, fmt.Errorf("closing decision %s failed: %w", decision.Tag, err) } results = append(results, decisionResult) } if err = tx.Commit(); err != nil { return nil, errCouldNotCommitTransaction(err) } return results, nil } func closeDecision(ctx context.Context, tx *sqlx.Tx, d *Motion) (*Motion, error) { quorum, majority := d.Type.QuorumAndMajority() var ( voteSums *VoteSums err error reasoning string ) // TODO: implement prefetching in CloseDecisions if voteSums, err = sumsForDecision(ctx, tx, d); err != nil { return nil, fmt.Errorf("getting vote sums failed: %w", err) } d.Status, reasoning = voteSums.CalculateResult(quorum, majority) result, err := tx.NamedExecContext( ctx, `UPDATE decisions SET status=:status, modified=CURRENT_TIMESTAMP WHERE id=:id`, d, ) if err != nil { return nil, errCouldNotExecuteQuery(err) } affectedRows, err := result.RowsAffected() if err != nil { return nil, fmt.Errorf("could not get affected rows count: %w", err) } if affectedRows != 1 { return nil, fmt.Errorf("unexpected number of rows %d instead of 1", affectedRows) } d.Sums = voteSums d.Reasoning = reasoning return d, nil } func (m *MotionModel) UnvotedForVoter(ctx context.Context, voter *User) ([]*Motion, error) { // TODO: implement more efficient variant that fetches unvoted votes for a slice of voters rows, err := m.DB.QueryxContext( ctx, `SELECT decisions.* FROM decisions WHERE due < ? AND status=? AND NOT EXISTS(SELECT * FROM votes WHERE decision = decisions.id AND voter = ?)`, time.Now().UTC(), voteStatusPending, voter.ID) if err != nil { return nil, errCouldNotExecuteQuery(err) } defer func() { _ = rows.Close() }() result := make([]*Motion, 0) for rows.Next() { if err := rows.Err(); err != nil { return nil, errCouldNotFetchRow(err) } var motion Motion if err := rows.StructScan(&motion); err != nil { return nil, errCouldNotScanResult(err) } result = append(result, &motion) } return result, nil } func sumsForDecision(ctx context.Context, tx *sqlx.Tx, d *Motion) (*VoteSums, error) { voteRows, err := tx.QueryxContext( ctx, `SELECT vote, COUNT(vote) FROM votes WHERE decision=$1 GROUP BY vote`, d.ID, ) if err != nil { return nil, fmt.Errorf("fetching vote sums for motion %s failed: %w", d.Tag, err) } defer func() { _ = voteRows.Close() }() sums := &VoteSums{} for voteRows.Next() { var ( vote *VoteChoice count int ) if err = voteRows.Err(); err != nil { return nil, fmt.Errorf("could not fetch vote sums for motion %s: %w", d.Tag, err) } if err = voteRows.Scan(&vote, &count); err != nil { return nil, fmt.Errorf("could not parse row for vote sums of motion %s: %w", d.Tag, err) } switch vote.ID { case VoteAye.ID: sums.Ayes = count case VoteNaye.ID: sums.Nayes = count case VoteAbstain.ID: sums.Abstains = count default: return nil, fmt.Errorf("unknown vote type '%+v'", vote) } } return sums, nil } func (m *MotionModel) NextPendingDue(ctx context.Context, relativeTo time.Time) (*time.Time, error) { row := m.DB.QueryRowContext( ctx, `SELECT due FROM decisions WHERE status=0 AND due >= ? ORDER BY due LIMIT 1`, relativeTo.UTC(), ) if row == nil { return nil, errors.New("no row returned") } if err := row.Err(); err != nil { return nil, fmt.Errorf("could not retrieve row for next pending decision: %w", err) } var due time.Time if err := row.Scan(&due); err != nil { if errors.Is(err, sql.ErrNoRows) { return nil, nil } return nil, fmt.Errorf("parsing result failed: %w", err) } return &due, nil } type MotionListOptions struct { Limit int UnvotedOnly bool Before, After *time.Time VoterID int64 } func (m *MotionModel) TimestampRange(ctx context.Context, options *MotionListOptions) (*time.Time, *time.Time, error) { var row *sqlx.Row if options.UnvotedOnly { row = m.DB.QueryRowxContext( ctx, `SELECT MIN(proposed), MAX(proposed) FROM decisions WHERE due >= ? AND NOT EXISTS(SELECT * FROM votes WHERE decision = decisions.id AND voter = ?)`, time.Now().UTC(), options.VoterID, ) } else { row = m.DB.QueryRowxContext( ctx, `SELECT MIN(proposed), MAX(proposed) FROM decisions`, ) } if err := row.Err(); err != nil { return nil, nil, fmt.Errorf("could not query for motion timestamps: %w", err) } var ( first, last sql.NullString firstTs, lastTs *time.Time err error ) if err := row.Scan(&first, &last); err != nil { return nil, nil, fmt.Errorf("could not scan timestamps: %w", err) } if !first.Valid || !last.Valid { return nil, nil, nil } if firstTs, err = parseSqlite3TimeStamp(first.String); err != nil { return nil, nil, err } if lastTs, err = parseSqlite3TimeStamp(last.String); err != nil { return nil, nil, err } return firstTs, lastTs, nil } func (m *MotionModel) List(ctx context.Context, options *MotionListOptions) ([]*Motion, error) { var ( rows *sqlx.Rows err error ) switch { case options.Before != nil: rows, err = m.rowsBefore(ctx, options) case options.After != nil: rows, err = m.rowsAfter(ctx, options) default: rows, err = m.rowsFirst(ctx, options) } if err != nil { return nil, err } defer func(rows *sqlx.Rows) { _ = rows.Close() }(rows) motions := make([]*Motion, 0, options.Limit) for rows.Next() { var decision Motion if err = rows.Err(); err != nil { return nil, errCouldNotFetchRow(err) } if err = rows.StructScan(&decision); err != nil { return nil, errCouldNotScanResult(err) } motions = append(motions, &decision) } if len(motions) > 0 { err = m.FillVoteSums(ctx, motions) if err != nil { return nil, err } } return motions, nil } func (m *MotionModel) FillVoteSums(ctx context.Context, decisions []*Motion) error { decisionIds := make([]int64, len(decisions)) decisionMap := make(map[int64]*Motion, len(decisions)) for idx, decision := range decisions { decision.Sums = &VoteSums{} decisionIds[idx] = decision.ID decisionMap[decision.ID] = decision } query, args, err := sqlx.In( `SELECT v.decision, v.vote, COUNT(*) FROM votes v WHERE v.decision IN (?) GROUP BY v.decision, v.vote`, decisionIds, ) if err != nil { return fmt.Errorf("could not create IN query: %w", err) } rows, err := m.DB.QueryContext(ctx, query, args...) if err != nil { return errCouldNotExecuteQuery(err) } defer func() { _ = rows.Close() }() for rows.Next() { if err = rows.Err(); err != nil { return errCouldNotFetchRow(err) } var ( decisionID int64 vote *VoteChoice count int ) err = rows.Scan(&decisionID, &vote, &count) if err != nil { return errCouldNotScanResult(err) } switch { case vote.Equal(VoteAye): decisionMap[decisionID].Sums.Ayes = count case vote.Equal(VoteNaye): decisionMap[decisionID].Sums.Nayes = count case vote.Equal(VoteAbstain): decisionMap[decisionID].Sums.Abstains = count } } return nil } func (m *MotionModel) rowsBefore(ctx context.Context, options *MotionListOptions) (*sqlx.Rows, error) { // TODO: implement variant for options.UnvotedOnly rows, err := m.DB.QueryxContext( ctx, `SELECT decisions.id, decisions.tag, decisions.proponent, voters.name AS proposer, decisions.proposed, decisions.title, decisions.content, decisions.votetype, decisions.status, decisions.due, decisions.modified FROM decisions JOIN voters ON decisions.proponent = voters.id WHERE decisions.proposed < $1 ORDER BY proposed DESC LIMIT $2`, options.Before, options.Limit, ) if err != nil { return nil, fmt.Errorf("could not query motions before %s: %w", options.Before, err) } return rows, nil } func (m *MotionModel) rowsAfter(ctx context.Context, options *MotionListOptions) (*sqlx.Rows, error) { // TODO: implement variant for options.UnvotedOnly rows, err := m.DB.QueryxContext( ctx, `WITH display_decision AS (SELECT decisions.id, decisions.tag, decisions.proponent, voters.name AS proposer, decisions.proposed, decisions.title, decisions.content, decisions.votetype, decisions.status, decisions.due, decisions.modified FROM decisions JOIN voters ON decisions.proponent = voters.id WHERE decisions.proposed > $1 ORDER BY proposed LIMIT $2) SELECT * FROM display_decision ORDER BY proposed DESC`, options.After, options.Limit, ) if err != nil { return nil, fmt.Errorf("could not query motions after %s: %w", options.After, err) } return rows, nil } func (m *MotionModel) rowsFirst(ctx context.Context, options *MotionListOptions) (*sqlx.Rows, error) { var ( rows *sqlx.Rows err error ) if options.UnvotedOnly { rows, err = m.DB.QueryxContext( ctx, `SELECT decisions.id, decisions.tag, decisions.proponent, voters.name AS proposer, decisions.proposed, decisions.title, decisions.content, decisions.votetype, decisions.status, decisions.due, decisions.modified FROM decisions JOIN voters ON decisions.proponent = voters.id WHERE status=? AND due >= ? AND NOT EXISTS(SELECT * FROM votes WHERE decision = decisions.id AND voter = ?) ORDER BY decisions.proposed DESC LIMIT ?`, voteStatusPending, time.Now().UTC(), options.VoterID, options.Limit, ) } else { rows, err = m.DB.QueryxContext( ctx, `SELECT decisions.id, decisions.tag, decisions.proponent, voters.name AS proposer, decisions.proposed, decisions.title, decisions.content, decisions.votetype, decisions.status, decisions.due, decisions.modified FROM decisions JOIN voters ON decisions.proponent = voters.id ORDER BY proposed DESC LIMIT ?`, options.Limit, ) } if err != nil { return nil, fmt.Errorf("could not query motions: %w", err) } return rows, nil } func (m *MotionModel) ByTag(ctx context.Context, tag string, withVotes bool) (*Motion, error) { row := m.DB.QueryRowxContext( ctx, `SELECT decisions.id, decisions.tag, decisions.proponent, voters.name AS proposer, decisions.proposed, decisions.title, decisions.content, decisions.votetype, decisions.status, decisions.due, decisions.modified FROM decisions JOIN voters ON decisions.proponent = voters.id WHERE decisions.tag = ?`, tag, ) if err := row.Err(); err != nil { return nil, fmt.Errorf("could not query motion: %w", err) } var result Motion if err := row.StructScan(&result); err != nil { return nil, fmt.Errorf("could not fill motion from query result: %w", err) } if err := m.FillVoteSums(ctx, []*Motion{&result}); err != nil { return nil, fmt.Errorf("could not get vote sums: %w", err) } if result.ID != 0 && withVotes { if err := m.FillVotes(ctx, &result); err != nil { return nil, fmt.Errorf("could not get votes for %s: %w", result.Tag, err) } } return &result, nil } func (m *MotionModel) FillVotes(ctx context.Context, md *Motion) error { rows, err := m.DB.QueryxContext(ctx, `SELECT voters.name, votes.vote FROM voters JOIN votes ON votes.voter = voters.id WHERE votes.decision = ? ORDER BY voters.name`, md.ID) if err != nil { if errors.Is(err, sql.ErrNoRows) { return nil } return fmt.Errorf("could not fetch rows: %w", err) } defer func() { _ = rows.Close() }() for rows.Next() { if err := rows.Err(); err != nil { return fmt.Errorf("could not get row: %w", err) } var vote Vote if err := rows.StructScan(&vote); err != nil { return errCouldNotScanResult(err) } md.Votes = append(md.Votes, &vote) } return nil } func (m *MotionModel) ByID(ctx context.Context, id int64) (*Motion, error) { row := m.DB.QueryRowxContext(ctx, `SELECT * FROM decisions WHERE id=?`, id) if err := row.Err(); err != nil { return nil, fmt.Errorf("could not fetch tag for id %d: %w", id, err) } var motion Motion if err := row.StructScan(&motion); err != nil { return nil, fmt.Errorf("could not get tag from row: %w", err) } return &motion, nil } func (m *MotionModel) Update( ctx context.Context, id int64, updateFn func(*Motion), ) error { tx, err := m.DB.BeginTxx(ctx, nil) if err != nil { return errCouldNotStartTransaction(err) } defer func(tx *sqlx.Tx) { _ = tx.Rollback() }(tx) row := tx.QueryRowxContext(ctx, `SELECT * FROM decisions WHERE id=?`, id) if err := row.Err(); err != nil { return fmt.Errorf("could not select motion: %w", err) } var motion Motion if err := row.StructScan(&motion); err != nil { return errCouldNotScanResult(err) } updateFn(&motion) motion.Modified = time.Now().UTC() _, err = tx.NamedExecContext( ctx, `UPDATE decisions SET title=:title, content=:content, votetype=:votetype, due=:due, modified=:modified, status=:status WHERE id = :id`, motion, ) if err != nil { return fmt.Errorf("could not update decision: %w", err) } if err := tx.Commit(); err != nil { return errCouldNotCommitTransaction(err) } return nil } type Vote struct { UserID int64 `db:"voter"` MotionID int64 `db:"decision"` Vote *VoteChoice `db:"vote"` Voted time.Time `db:"voted"` Notes string `db:"notes"` Name string `db:"name"` } func (m *MotionModel) UpdateVote(ctx context.Context, userID, motionID int64, performVoteFn func(v *Vote)) error { tx, err := m.DB.BeginTxx(ctx, nil) if err != nil { return errCouldNotStartTransaction(err) } defer func(tx *sqlx.Tx) { _ = tx.Rollback() }(tx) row := tx.QueryRowxContext(ctx, `SELECT * FROM votes WHERE voter=? AND decision=?`, userID, motionID) if err := row.Err(); err != nil { return errCouldNotExecuteQuery(err) } vote := Vote{UserID: userID, MotionID: motionID} if err := row.StructScan(&vote); err != nil && !errors.Is(err, sql.ErrNoRows) { return fmt.Errorf("could not scan vote structure: %w", err) } performVoteFn(&vote) if _, err := tx.NamedExecContext( ctx, `INSERT INTO votes (decision, voter, vote, voted, notes) VALUES (:decision, :voter, :vote, :voted, :notes) ON CONFLICT (decision, voter) DO UPDATE SET vote=:vote, voted=:voted, notes=:notes WHERE decision = :decision AND voter = :voter`, vote, ); err != nil { return fmt.Errorf("could not insert or update vote: %w", err) } if err := tx.Commit(); err != nil { return errCouldNotCommitTransaction(err) } return nil } func (m *MotionModel) Withdraw(ctx context.Context, id int64) error { return m.Update(ctx, id, func(m *Motion) { m.Status = voteStatusWithdrawn }) }