page.go

package chatgpt

import (
	"github.com/cherish-chat/chatgpt-firefox/config"
	"github.com/playwright-community/playwright-go"
	"github.com/sirupsen/logrus"
	"log"
)

// Deprecated 请使用GetPage
func (h *Helper) MustGetPage(id string) playwright.Page {
	if page, ok := h.pageMap.Load(id); ok {
		return page.(playwright.Page)
	}
	var page playwright.Page
	var err error
	page, err = h.NewPage(id)
	if err != nil {
		logrus.Errorf("Error while creating new page: %v", err)
		log.Fatalf("Error while creating new page: %v", err)
	}
	return page
}

func (h *Helper) GetPage(id string) (playwright.Page, error) {
	if page, ok := h.pageMap.Load(id); ok {
		return page.(playwright.Page), nil
	}
	return h.NewPage(id)
}

func (h *Helper) GetPlusPage(id string) (playwright.Page, error) {
	if page, ok := h.pageMap.Load(id); ok {
		return page.(playwright.Page), nil
	}
	return h.NewPlusPage(id)
}

func (h *Helper) NewPage(id string) (playwright.Page, error) {
	page, err := h.browser.NewPage()
	if err != nil {
		logrus.Errorf("Error while creating new page: %v", err)
		return nil, err
	}
	logrus.Info("New page created successfully")
	{
		// 设置cookie
		cookies, err := config.LoadCookies("cookies.json")
		if err != nil {
			return nil, err
		}
		err = h.browser.AddCookies(cookies...)
		if err != nil {
			logrus.Errorf("Error while adding cookies: %v", err)
			return nil, err
		}
		logrus.Info("Cookies added successfully")
	}
	_, err = page.Goto("https://chat.openai.com/chat")
	if err != nil {
		page.Close()
		logrus.Errorf("Error while navigating to google: %v", err)
		return nil, err
	}

	logrus.Info("Navigated to openai successfully")
	h.pageMap.Store(id, page)
	return page, nil
}

func (h *Helper) NewPlusPage(id string) (playwright.Page, error) {
	page, err := h.browser.NewPage()
	if err != nil {
		logrus.Errorf("Error while creating new page: %v", err)
		return nil, err
	}
	logrus.Info("New page created successfully")
	{
		// 设置cookie
		cookies, err := config.LoadCookies("cookies.json")
		if err != nil {
			return nil, err
		}
		err = h.browser.AddCookies(cookies...)
		if err != nil {
			logrus.Errorf("Error while adding cookies: %v", err)
			return nil, err
		}
		logrus.Info("Cookies added successfully")
	}
	_, err = page.Goto("https://chat.openai.com/chat?model=text-davinci-002-render-paid")
	if err != nil {
		page.Close()
		logrus.Errorf("Error while navigating to google: %v", err)
		return nil, err
	}

	logrus.Info("Navigated to openai successfully")
	h.pageMap.Store(id, page)
	return page, nil
}