// Copyright (C) 2019 Storj Labs, Inc.
// See LICENSE for copying information.

package storagenodedb

import (
	"context"

	"storj.io/storj/pkg/storj"
	"storj.io/storj/storagenode/reputation"
)

// Reputation returns reputation.DB
func (db *InfoDB) Reputation() reputation.DB { return &reputationDB{db} }

// Reputation returns reputation.DB
func (db *DB) Reputation() reputation.DB { return db.info.Reputation() }

// reputation works with node reputation DB
type reputationDB struct {
	*InfoDB
}

// Store inserts or updates reputation stats into the db
func (db *reputationDB) Store(ctx context.Context, stats reputation.Stats) (err error) {
	defer mon.Task()(&ctx)(&err)

	stmt := `INSERT OR REPLACE INTO reputation (
				satellite_id, 
				uptime_success_count,
				uptime_total_count,
				uptime_reputation_alpha,
				uptime_reputation_beta,
				uptime_reputation_score,
				audit_success_count,
				audit_total_count,
				audit_reputation_alpha,
				audit_reputation_beta,
				audit_reputation_score,
				updated_at
			) VALUES(?,?,?,?,?,?,?,?,?,?,?,?)`

	_, err = db.db.ExecContext(ctx, stmt,
		stats.SatelliteID,
		stats.Uptime.SuccessCount,
		stats.Uptime.TotalCount,
		stats.Uptime.Alpha,
		stats.Uptime.Beta,
		stats.Uptime.Score,
		stats.Audit.SuccessCount,
		stats.Audit.TotalCount,
		stats.Audit.Alpha,
		stats.Audit.Beta,
		stats.Audit.Score,
		stats.UpdatedAt.UTC(),
	)

	return nil
}

// Get retrieves stats for specific satellite
func (db *reputationDB) Get(ctx context.Context, satelliteID storj.NodeID) (_ *reputation.Stats, err error) {
	defer mon.Task()(&ctx)(&err)

	stats := reputation.Stats{}

	row := db.db.QueryRowContext(ctx,
		`SELECT * FROM reputation WHERE satellite_id = ?`,
		satelliteID,
	)

	err = row.Scan(&stats.SatelliteID,
		&stats.Uptime.SuccessCount,
		&stats.Uptime.TotalCount,
		&stats.Uptime.Alpha,
		&stats.Uptime.Beta,
		&stats.Uptime.Score,
		&stats.Audit.SuccessCount,
		&stats.Audit.TotalCount,
		&stats.Audit.Alpha,
		&stats.Audit.Beta,
		&stats.Audit.Score,
		&stats.UpdatedAt,
	)
	if err != nil {
		return nil, err
	}

	return &stats, nil
}