forked from tildetown/town
mark code used
parent
7255ee691e
commit
2a07a0e200
|
@ -1,10 +1,12 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"database/sql"
|
||||
"errors"
|
||||
"fmt"
|
||||
"os"
|
||||
"os/exec"
|
||||
"strconv"
|
||||
"strings"
|
||||
|
||||
|
@ -16,7 +18,7 @@ import (
|
|||
"github.com/mattn/go-tty"
|
||||
)
|
||||
|
||||
// TODO consider local-only help command for renaming and adding emails to account
|
||||
// TODO consider local-only help command for renaming, email mgmt, deleting account
|
||||
|
||||
// TODO put colorscheme, prompting stuff into own packages for use in the other commands. would be good to get off of survey.
|
||||
|
||||
|
@ -262,9 +264,27 @@ func redeemCode(db *sql.DB, cs colorScheme, p *Prompter) error {
|
|||
return nil
|
||||
}
|
||||
|
||||
// TODO need to create a new helper: appendkeyfile and use sudoers to allow help to call it. also need to add help user and add it to /etc/ssh/sshd_config
|
||||
cmd := exec.Command("sudo", "--user", user.Username, "/town/bin/appendkeyfile", user.Username)
|
||||
cmd.Stdin = bytes.NewBufferString(key)
|
||||
stdoutBuff := bytes.NewBuffer([]byte{})
|
||||
cmd.Stdout = stdoutBuff
|
||||
if err = cmd.Run(); err != nil {
|
||||
// TODO log error
|
||||
//return fmt.Errorf("appendkeyfile failed with '%s': %w", string(stdoutBuff.Bytes()), err)
|
||||
return errors.New("adding to keys file failed")
|
||||
}
|
||||
|
||||
// TODO add help user
|
||||
// TODO update sshd_config
|
||||
// TODO update sudoers
|
||||
// TODO compile appendkeyfile and add to /town/bin/
|
||||
|
||||
err = code.MarkUsed(db)
|
||||
if err != nil {
|
||||
// TODO log err
|
||||
return errors.New("database was sad")
|
||||
}
|
||||
|
||||
// TODO mark used
|
||||
return nil
|
||||
}
|
||||
|
||||
|
|
|
@ -221,4 +221,32 @@ func (c *AuthCode) Hydrate(db *sql.DB) error {
|
|||
return stmt.QueryRow(c.Code).Scan(&c.ID, &c.Used, &c.Created)
|
||||
}
|
||||
|
||||
// TODO other auth code as needed
|
||||
func (c *AuthCode) MarkUsed(db *sql.DB) error {
|
||||
if c.ID == 0 {
|
||||
return errors.New("not hydrated")
|
||||
}
|
||||
|
||||
stmt, err := db.Prepare(`
|
||||
UPDATE auth_codes SET used = 1 WHERE id = ?`)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer stmt.Close()
|
||||
|
||||
result, err := stmt.Exec(c.ID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
var rowsAffected int64
|
||||
|
||||
if rowsAffected, err = result.RowsAffected(); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if rowsAffected == 0 {
|
||||
return errors.New("no rows affected")
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue