mirror of
https://github.com/neilalexander/yggmail.git
synced 2026-04-19 00:26:28 +03:00
12c153a364
* Update main.go * Update main.go * Set * Set password to db * Main - Password hash, after trimming, cannot be empty * Backend - Assume incoming data is already a hash * Main - CLeaned up * README - Updated command-line argument documentation * Menton this * lower case * SQLITE Driver - Removed comments * Main - `passwordhash` only valiud when non-"" * Main - Try running `Cost([]byte)` on the UTF-8 bytes to verify the hash
95 lines
2.0 KiB
Go
95 lines
2.0 KiB
Go
/*
|
|
* Copyright (c) 2021 Neil Alexander
|
|
*
|
|
* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
*/
|
|
|
|
package sqlite3
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
|
|
"golang.org/x/crypto/bcrypt"
|
|
)
|
|
|
|
type TableConfig struct {
|
|
db *sql.DB
|
|
writer *Writer
|
|
get *sql.Stmt
|
|
set *sql.Stmt
|
|
}
|
|
|
|
const configSchema = `
|
|
CREATE TABLE IF NOT EXISTS config (
|
|
key TEXT NOT NULL,
|
|
value TEXT NOT NULL,
|
|
PRIMARY KEY(key)
|
|
);
|
|
`
|
|
|
|
const configGet = `
|
|
SELECT value FROM config WHERE key = $1
|
|
`
|
|
|
|
const configSet = `
|
|
INSERT OR REPLACE INTO config (key, value) VALUES($1, $2)
|
|
`
|
|
|
|
func NewTableConfig(db *sql.DB, writer *Writer) (*TableConfig, error) {
|
|
t := &TableConfig{
|
|
db: db,
|
|
writer: writer,
|
|
}
|
|
_, err := db.Exec(configSchema)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("db.Exec: %w", err)
|
|
}
|
|
t.get, err = db.Prepare(configGet)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("db.Prepare(get): %w", err)
|
|
}
|
|
t.set, err = db.Prepare(configSet)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("db.Prepare(set): %w", err)
|
|
}
|
|
return t, nil
|
|
}
|
|
|
|
func (t *TableConfig) ConfigGet(key string) (string, error) {
|
|
var value string
|
|
err := t.get.QueryRow(key).Scan(&value)
|
|
if err == sql.ErrNoRows {
|
|
return "", nil
|
|
}
|
|
return value, err
|
|
}
|
|
|
|
func (t *TableConfig) ConfigSet(key, value string) error {
|
|
return t.writer.Do(t.db, nil, func(txn *sql.Tx) error {
|
|
_, err := t.set.Exec(key, value)
|
|
return err
|
|
})
|
|
}
|
|
|
|
func (t *TableConfig) ConfigSetPassword(passwordHash string) error {
|
|
return t.ConfigSet("password", passwordHash)
|
|
}
|
|
|
|
func (t *TableConfig) ConfigTryPassword(password string) (bool, error) {
|
|
dbPasswordHash, err := t.ConfigGet("password")
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
if dbPasswordHash == "" {
|
|
return true, nil // TODO: Do we want to allow login if no password is set?
|
|
}
|
|
err = bcrypt.CompareHashAndPassword([]byte(dbPasswordHash), []byte(password))
|
|
if err == nil {
|
|
return true, nil
|
|
}
|
|
return false, err
|
|
}
|