First commit

overleaf.go

@@ -0,0 +1,118 @@
+package main
+
+import (
+	"bytes"
+	"errors"
+	"fmt"
+	"io"
+	"net/http"
+	"strings"
+
+	"github.com/PuerkitoBio/goquery"
+)
+
+type Overleaf struct {
+	BaseURL string
+	Client  *http.Client
+	CSRF    string
+}
+
+// NewOverleaf initializes a new Overleaf instance
+func NewOverleaf(baseURL string) *Overleaf {
+	return &Overleaf{
+		BaseURL: strings.TrimSuffix(baseURL, "/"),
+		Client:  &http.Client{},
+	}
+}
+
+// get performs a GET request
+func (o *Overleaf) get(endpoint string) (*http.Response, error) {
+	url := fmt.Sprintf("%s%s", o.BaseURL, endpoint)
+	return o.Client.Get(url)
+}
+
+// post performs a POST request
+func (o *Overleaf) post(endpoint string, data map[string]string) (*http.Response, error) {
+	if o.CSRF == "" {
+		if err := o.obtainCSRF(); err != nil {
+			return nil, err
+		}
+	}
+
+	url := fmt.Sprintf("%s%s", o.BaseURL, endpoint)
+	data["_csrf"] = o.CSRF
+	formData := encodeFormData(data)
+
+	req, err := http.NewRequest("POST", url, bytes.NewBufferString(formData))
+	if err != nil {
+		return nil, err
+	}
+	req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
+
+	return o.Client.Do(req)
+}
+
+// obtainCSRF retrieves the CSRF token
+func (o *Overleaf) obtainCSRF() error {
+	resp, err := o.get("/login")
+	if err != nil {
+		return err
+	}
+	defer resp.Body.Close()
+
+	doc, err := goquery.NewDocumentFromReader(resp.Body)
+	if err != nil {
+		return err
+	}
+
+	csrf, exists := doc.Find(`meta[name="ol-csrfToken"]`).Attr("content")
+	if !exists {
+		return errors.New("CSRF token not found")
+	}
+	o.CSRF = csrf
+	return nil
+}
+
+// Login as admin
+func (o *Overleaf) Login(email, password string) error {
+	resp, err := o.post("/login", map[string]string{
+		"email":    email,
+		"password": password,
+	})
+	if err != nil {
+		return err
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != http.StatusOK {
+		return errors.New("failed to log in as admin")
+	}
+	return nil
+}
+
+// RegisterUser registers a new user
+func (o *Overleaf) RegisterUser(email, password string) error {
+	resp, err := o.post("/admin/register", map[string]string{
+		"email":    email,
+		"password": password,
+	})
+	if err != nil {
+		return err
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != http.StatusOK {
+		body, _ := io.ReadAll(resp.Body)
+		return fmt.Errorf("failed to register user: %s", string(body))
+	}
+	return nil
+}
+
+// encodeFormData encodes a map into x-www-form-urlencoded format
+func encodeFormData(data map[string]string) string {
+	var formData []string
+	for key, value := range data {
+		formData = append(formData, fmt.Sprintf("%s=%s", key, value))
+	}
+	return strings.Join(formData, "&")
+}