38 lines
744 B
Go
38 lines
744 B
Go
package persistence
|
|
|
|
import (
|
|
"encoding/json"
|
|
"gitlab.com/offline-twitter/twitter_offline_engine/pkg/scraper"
|
|
"os"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
func (p Profile) SaveSession(api scraper.API) {
|
|
data, err := json.Marshal(api)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
log.Debug("Profile Dir: " + p.ProfileDir)
|
|
err = os.WriteFile(p.ProfileDir+"/"+string(api.UserHandle+".session"), data, os.FileMode(0644))
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func (p Profile) LoadSession(userhandle scraper.UserHandle) scraper.API {
|
|
data, err := os.ReadFile(p.ProfileDir + "/" + string(userhandle+".session"))
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
var result scraper.API
|
|
err = json.Unmarshal(data, &result)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
return result
|
|
}
|