mirror of https://github.com/danielmiessler/fabric
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
69 lines
1.4 KiB
Go
69 lines
1.4 KiB
Go
package db
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/danielmiessler/fabric/common"
|
|
)
|
|
|
|
type Sessions struct {
|
|
*Storage
|
|
}
|
|
|
|
func (o *Sessions) LoadOrCreateSession(name string) (ret *Session, err error) {
|
|
if name == "" {
|
|
return &Session{}, nil
|
|
}
|
|
|
|
path := o.BuildFilePath(name)
|
|
if _, statErr := os.Stat(path); errors.Is(statErr, os.ErrNotExist) {
|
|
fmt.Printf("Creating new session: %s\n", name)
|
|
ret = &Session{Name: name, sessions: o}
|
|
} else {
|
|
ret, err = o.loadSession(name)
|
|
}
|
|
return
|
|
}
|
|
|
|
// LoadSession Load a session from file
|
|
func (o *Sessions) LoadSession(name string) (ret *Session, err error) {
|
|
if name == "" {
|
|
return &Session{}, nil
|
|
}
|
|
ret, err = o.loadSession(name)
|
|
return
|
|
}
|
|
|
|
func (o *Sessions) loadSession(name string) (ret *Session, err error) {
|
|
ret = &Session{Name: name, sessions: o}
|
|
if err = o.LoadAsJson(name, &ret.Messages); err != nil {
|
|
return
|
|
}
|
|
return
|
|
}
|
|
|
|
type Session struct {
|
|
Name string
|
|
Messages []*common.Message
|
|
|
|
sessions *Sessions
|
|
}
|
|
|
|
func (o *Session) Append(messages ...*common.Message) {
|
|
o.Messages = append(o.Messages, messages...)
|
|
}
|
|
|
|
// Save the session on disk
|
|
func (o *Session) Save() (err error) {
|
|
var jsonBytes []byte
|
|
if jsonBytes, err = json.Marshal(o.Messages); err == nil {
|
|
err = o.sessions.Save(o.Name, jsonBytes)
|
|
} else {
|
|
err = fmt.Errorf("could not marshal session %o: %o", o.Name, err)
|
|
}
|
|
return
|
|
}
|