90 lines
1.6 KiB
Go
90 lines
1.6 KiB
Go
package service
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/google/uuid"
|
|
"os"
|
|
"strings"
|
|
)
|
|
|
|
type PrivDirService struct{}
|
|
|
|
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() *PrivDirService {
|
|
return &PrivDirService{}
|
|
}
|
|
|
|
func (sv *PrivDirService) CreatePriv(dirname string, acc *Account) error {
|
|
db, err := Open()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer db.Close()
|
|
|
|
id := uuid.NewString()
|
|
stmt, err := db.Prepare("insert into PrivDir(id, name, owner) values (?, ?, ?);")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer stmt.Close()
|
|
|
|
_, err = stmt.Exec(id, dirname, 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 name = ?;")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer stmt.Close()
|
|
|
|
row := stmt.QueryRow(name)
|
|
var data PrivDir
|
|
|
|
if err = row.Scan(&data.Id, &data.DirName, &data.Owner); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &data, nil
|
|
}
|