implemented saving cookies and user session to a file

persistence/session.go

@@ -1,22 +1,34 @@
 package persistence
 
 import (
+	"encoding/json"
 	"offline_twitter/scraper"
+	"os"
 )
 
 func (p Profile) SaveSession(api scraper.API) {
-	// TODO session-saving:1
+	data, err := json.Marshal(api)
-	// - To understand what's going on here, look at the `MarshalJSON` function in `scraper/api_request_utils.go`,
+	if err != nil {
-	//   and the output of `git show 390c83154117aa2a339a83f05820fb904a32298e`.
+		panic(err)
-	// - use `json.Marshal` on the API object and write the resulting bytes to like "[api.UserHandle].session" or something
+	}
-	// - use `os.WriteFile` to write the file
+
-	panic("TODO")
+	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 {
-	// TODO session-saving:2
+	data, err := os.ReadFile(p.ProfileDir + "/" + string(userhandle+".session"))
-	// - use `os.ReadFile` to read "[userhandle].session"
+	if err != nil {
-	// - create a variable of type scraper.API
+		panic(err)
-	// - use `json.Unmarshal` to load the file contents into the new API variable
+	}
-	panic("TODO")
+
+	var result scraper.API
+	err = json.Unmarshal(data, &result)
+	if err != nil {
+		panic(err)
+	}
+
+	return result
 }

persistence/session_test.go

@@ -1,28 +1,46 @@
 package persistence_test
 
 import (
-	"github.com/go-test/deep"
+	"fmt"
-	"offline_twitter/persistence"
+	"math/rand"
+	"net/http"
+	"net/http/cookiejar"
 	"offline_twitter/scraper"
 	"testing"
+	"time"
+
+	"github.com/go-test/deep"
 )
 
 // Save and load an API session; it should come back the same
 func TestSaveAndLoadAuthenticatedSession(t *testing.T) {
-	assert := assert.New(t)
+
 	profile_path := "test_profiles/TestSession"
 	profile := create_or_load_profile(profile_path)
 
+	jar, err := cookiejar.New(nil)
+	if err != nil {
+		panic(err)
+	}
+
 	api := scraper.API{
 		// TODO session-saving
 		// - Fill out some fields here like Cookies and CSRFToken and UserHandle
+		UserHandle:      "testUser",
+		IsAuthenticated: true,
+		Client: http.Client{
+			Timeout: 10 * time.Second,
+			Jar:     jar,
+		},
+		CSRFToken: fmt.Sprint(rand.Int()),
 	}
 
 	// Save and load the session; it should come back the same
 	profile.SaveSession(api)
-	new_api = profile.LoadSession(api.UserHandle)
+	new_api := profile.LoadSession(api.UserHandle)
 
 	if diff := deep.Equal(api, new_api); diff != nil {
-		t.Errorf(diff)
+		t.Error(diff)
 	}
+
 }