147 lines
2.6 KiB
Go
147 lines
2.6 KiB
Go
package service
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/google/uuid"
|
|
"os"
|
|
"strings"
|
|
)
|
|
|
|
type PrivDirService struct {
|
|
acc *Account
|
|
}
|
|
|
|
type PrivDir struct {
|
|
Id string `json:"id"`
|
|
DirName string `json:"dirname"`
|
|
Owner string `json:"owner"`
|
|
}
|
|
|
|
func init() {
|
|
db, err := Open()
|
|
if err != nil {
|
|
_, _ = fmt.Fprintf(os.Stderr, "%v\n", err)
|
|
return
|
|
}
|
|
defer db.Close()
|
|
|
|
stmt, err := db.Prepare(strings.TrimSpace(`
|
|
create table PrivDir(
|
|
id varchar(36),
|
|
dirname varchar(250) unique,
|
|
owner varchar(25),
|
|
constraint PK_PrivDir_ID primary key(id),
|
|
constraint FK_Owner_ID foreign key(owner)
|
|
references(Account.username) on update cascade on delete cascade
|
|
);
|
|
`))
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer stmt.Close()
|
|
|
|
if _, err = stmt.Exec(); err != nil {
|
|
_, _ = fmt.Fprintf(os.Stderr, "%v\n", err)
|
|
return
|
|
}
|
|
}
|
|
|
|
func NewPrivDirService(acc *Account) *PrivDirService {
|
|
return &PrivDirService{
|
|
acc: acc,
|
|
}
|
|
}
|
|
|
|
func (sv *PrivDirService) CreatePriv(dirname string) error {
|
|
db, err := Open()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer db.Close()
|
|
|
|
id := uuid.NewString()
|
|
stmt, err := db.Prepare("insert into PrivDir(id, dirname, owner) values (?, ?, ?);")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer stmt.Close()
|
|
|
|
_, err = stmt.Exec(id, dirname, sv.acc.Username)
|
|
return nil
|
|
}
|
|
|
|
func (sv *PrivDirService) ReadPriv(name string) (*PrivDir, error) {
|
|
db, err := Open()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer db.Close()
|
|
|
|
stmt, err := db.Prepare("select * from PrivDir where dirname = ? and owner = ?;")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer stmt.Close()
|
|
|
|
row := stmt.QueryRow(name, sv.acc.Username)
|
|
var data PrivDir
|
|
|
|
if err = row.Scan(&data.Id, &data.DirName, &data.Owner); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &data, nil
|
|
}
|
|
|
|
func (sv *PrivDirService) DeletePriv(name string) error {
|
|
db, err := Open()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer db.Close()
|
|
|
|
stmt, err := db.Prepare("delete from PrivDir where dirname = ? and owner = ?;")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer stmt.Close()
|
|
|
|
_, err = stmt.Exec(name, sv.acc.Username)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (sv *PrivDirService) QueryPriv() ([]*PrivDir, error) {
|
|
db, err := Open()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer db.Close()
|
|
|
|
stmt, err := db.Prepare("select * from PrivDir;")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer stmt.Close()
|
|
|
|
rows, err := stmt.Query()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
|
|
var dirs []*PrivDir
|
|
for rows.Next() {
|
|
var data PrivDir
|
|
if err = rows.Scan(&data.Id, &data.DirName, &data.Owner); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
dirs = append(dirs, &data)
|
|
}
|
|
|
|
return dirs, nil
|
|
}
|