refactor: 重构导航模块

Author: seth-shi

api/api_topics.go

@@ -17,25 +17,26 @@ func (cli *v2exClient) GetTopics(
 	return func() tea.Msg {
 
 		var (
-			conf      = g.Config.Get()
-			nodeIndex = conf.ActiveTab
-			chooseV2  = conf.ChooseAPIV2
-			node      = g.GetGroupNode(nodeIndex)
-			res       []response.TopicResult
-			total     int
-			err       error
+			conf       = g.Config.Get()
+			nodeIndex  = conf.ActiveTab
+			chooseV2   = conf.ChooseAPIV2
+			node       = g.GetGroupNode(nodeIndex)
+			res        []response.TopicResult
+			total      int
+			err        error
+			cachePages = -1
 		)
 
 		// 如果是 myNodes, 那么就去用 V2 的接口
-		v2 := node.Key == g.NodesMy || chooseV2
+		v2 := chooseV2
 		// 最新最热, 只能用 v1
 		if node.Key == g.HotNode || node.Key == g.LatestNode {
 			v2 = false
 		}
 
 		g.Session.IsApiV2.Store(v2)
 		if v2 {
-			res, total, err = cli.v2TopicApi.GetTopicsByGroupNode(ctx, node, page)
+			res, cachePages, total, err = cli.v2TopicApi.GetTopicsByGroupNode(ctx, node, page)
 		} else {
 			res, total, err = cli.v1TopicApi.GetTopicsByGroupNode(ctx, node, page)
 		}
@@ -50,6 +51,7 @@ func (cli *v2exClient) GetTopics(
 				TotalCount: total,
 				CurrPage:   page,
 			},
+			CachePages: cachePages,
 		}
 	}
 }

api/internal/api_topics/api_v2.go

@@ -12,6 +12,7 @@ import (
 	"github.com/puzpuzpuz/xsync/v4"
 	"github.com/samber/lo"
 	"github.com/seth-shi/go-v2ex/v2/g"
+	"github.com/seth-shi/go-v2ex/v2/pkg"
 	"github.com/seth-shi/go-v2ex/v2/response"
 	"golang.org/x/sync/errgroup"
 	"resty.dev/v3"
@@ -52,11 +53,11 @@ func (api *V2TopicApi) GetTopicsByGroupNode(
 	ctx context.Context,
 	node g.GroupNode,
 	page int,
-) (res []response.TopicResult, total int, err error) {
+) (res []response.TopicResult, cachePages, total int, err error) {
 
 	// 只允许单个请求进来获取 数据
 	if !api.isRequesting.CompareAndSwap(false, true) {
-		return nil, 0, ErrLockingRequestData
+		return nil, 0, 0, ErrLockingRequestData
 	}
 	defer api.finishRequest(&err)
 
@@ -80,8 +81,9 @@ func (api *V2TopicApi) GetTopicsByGroupNode(
 	}
 
 	res = lo.Subset(api.cacheData, (page-1)*perPage, perPage)
+	cachePages = pkg.TotalPages(len(api.cacheData), perPage)
 	total, _ = api.groupTotalCount.Load(node.Key)
-	return res, total, nil
+	return res, cachePages, total, nil
 }
 
 func (api *V2TopicApi) prepareRequest(node g.GroupNode, page int) error {

api/v2ex.go

@@ -39,6 +39,8 @@ func SetUpHttpClient(conf *model.FileConfig) {
 			},
 		)
 		client.SetTransport(&pkg.MockRoundTripper{Mock: mockApiResp})
+	} else if conf.IsDevelopmentEnv() {
+		client.SetResponseBodyUnlimitedReads(true)
 	}
 
 	V2ex = &v2exClient{

messages/boss.go

@@ -1,5 +1,4 @@
 package messages
 
-type BossInitMsg struct{}
 type BossStartInstallPkgMsg struct{}
 type BossEndInstallPkgMsg string

messages/detail.go

@@ -4,10 +4,6 @@ import (
 	"github.com/seth-shi/go-v2ex/v2/response"
 )
 
-type GetDetailRequest struct {
-	ID int64
-}
-
 type GetDetailResponse struct {
 	Data response.V2DetailResult
 }

messages/topic.go

@@ -5,6 +5,7 @@ import (
 )
 
 type GetTopicResponse struct {
-	Data     []response.TopicResult
-	PageInfo *response.PerTenPageInfo
+	Data       []response.TopicResult
+	PageInfo   *response.PerTenPageInfo
+	CachePages int
 }

model/file.go

@@ -4,26 +4,24 @@ import (
 	"encoding/json"
 	"os"
 	"path"
+	"strings"
 
 	"github.com/mitchellh/go-homedir"
 	"github.com/seth-shi/go-v2ex/v2/consts"
 )
 
 const (
-	envProduction  = "production"
-	envDevelopment = "development"
-	envMock        = "mock"
+	envProduction = "production"
 )
 
 type FileConfig struct {
-	Token         string `json:"personal_access_token"`
-	MyNodes       string `json:"my_nodes"`
-	Timeout       uint   `json:"timeout"`
-	ActiveTab     int    `json:"active_tab"`
-	ShowMode      int    `json:"show_mode"`
-	Env           string `json:"env"`
-	ChooseAPIV2   bool   `json:"choose_api_v2"`
-	BossModeBlank bool   `json:"boss_mode_blank"`
+	Token       string `json:"personal_access_token"`
+	MyNodes     string `json:"my_nodes"`
+	Timeout     uint   `json:"timeout"`
+	ActiveTab   int    `json:"active_tab"`
+	ShowMode    int    `json:"show_mode"`
+	Env         string `json:"env"`
+	ChooseAPIV2 bool   `json:"choose_api_v2"`
 }
 
 func NewDefaultFileConfig() *FileConfig {
@@ -40,13 +38,13 @@ func NewDefaultFileConfig() *FileConfig {
 }
 
 func (c *FileConfig) IsProductionEnv() bool {
-	return c.Env == envProduction
+	return strings.Contains(c.Env, "prod")
 }
 func (c *FileConfig) IsDevelopmentEnv() bool {
-	return c.Env == envDevelopment
+	return strings.Contains(c.Env, "dev")
 }
 func (c *FileConfig) IsMockEnv() bool {
-	return c.Env == envMock
+	return strings.Contains(c.Env, "mock")
 }
 
 func (c *FileConfig) SwitchShowMode() int {

nav/event.go

@@ -0,0 +1,11 @@
+package nav
+
+import (
+	tea "github.com/charmbracelet/bubbletea"
+)
+
+// PageLife  will call every time.
+type PageLife interface {
+	OnEntering() (tea.Model, tea.Cmd)
+	OnLeaving() (tea.Model, tea.Cmd)
+}

nav/msg.go

@@ -0,0 +1,52 @@
+package nav
+
+import (
+	"reflect"
+
+	tea "github.com/charmbracelet/bubbletea"
+)
+
+type PopNavigationMsg struct{}
+type PushNavigationMsg tea.Model
+type NoMorePageMsg struct{}
+
+func Push(m tea.Model) tea.Cmd {
+	return postCmd(PushNavigationMsg(m))
+}
+
+func Back() tea.Cmd {
+	return postCmd(PopNavigationMsg{})
+}
+
+func Histories() []tea.Model {
+	return nav.histories
+}
+
+func CurrentPage() tea.Model {
+	return nav.curr
+}
+
+func IsActivePage(m tea.Model) bool {
+	curr := nav.curr
+	if curr == nil || m == nil {
+		return false
+	}
+
+	return reflect.TypeOf(m).String() == reflect.TypeOf(curr).String()
+}
+
+func PushOrBack(m tea.Model) tea.Cmd {
+
+	// 如果当前无东西, 直接返回
+	if IsActivePage(m) {
+		return Back()
+	}
+
+	return Push(m)
+}
+
+func postCmd(msg tea.Msg) tea.Cmd {
+	return func() tea.Msg {
+		return msg
+	}
+}

nav/nav.go

@@ -0,0 +1,118 @@
+package nav
+
+import (
+	tea "github.com/charmbracelet/bubbletea"
+)
+
+var (
+	nav      = &navigator{}
+	maxPages = 1000
+)
+
+type navigator struct {
+	histories []tea.Model
+	curr      tea.Model
+}
+
+func View() string {
+	if nav.curr == nil {
+		return ""
+	}
+
+	return nav.curr.View()
+}
+
+func Update(msg tea.Msg) tea.Cmd {
+	switch msg := msg.(type) {
+	case PushNavigationMsg:
+		return handlePushNavigation(msg)
+	case PopNavigationMsg:
+		return handlePopNavigation()
+	}
+
+	if nav.curr == nil {
+		return nil
+	}
+
+	// update page
+	var c tea.Cmd
+	nav.curr, c = nav.curr.Update(msg)
+	return c
+}
+
+// handlePushNavigation 处理页面压入导航栈的操作
+func handlePushNavigation(msg PushNavigationMsg) tea.Cmd {
+	var (
+		cmds []tea.Cmd
+		c    tea.Cmd
+	)
+
+	// 1. 处理当前页面的离开逻辑
+	if nav.curr != nil {
+		// 将当前页面存入历史记录
+		nav.histories = append(nav.histories, nav.curr)
+		if len(nav.histories) > maxPages {
+			first := nav.histories[0]
+			if p, ok := first.(PageLife); ok {
+				_, c = p.OnLeaving()
+				cmds = append(cmds, c)
+			}
+
+			nav.histories = nav.histories[1:]
+		}
+
+		// 调用当前页面的离开钩子
+		if p, ok := nav.curr.(PageLife); ok {
+			nav.curr, c = p.OnLeaving()
+			cmds = append(cmds, c)
+		}
+	}
+
+	// 2. 压入新页面
+	nav.curr = tea.Model(msg)
+	// 3. 初始化新页面并处理进入逻辑
+	cmds = append(cmds, nav.curr.Init())
+
+	// 页面离开钩子
+	if p, ok := nav.curr.(PageLife); ok {
+		nav.curr, c = p.OnEntering()
+		cmds = append(cmds, c)
+	}
+
+	return tea.Sequence(cmds...)
+}
+
+// handlePopNavigation 处理页面弹出导航栈的操作
+func handlePopNavigation() tea.Cmd {
+
+	// 尝试弹出页面
+	if nav.curr == nil {
+		return postCmd(NoMorePageMsg{})
+	}
+
+	if len(nav.histories) == 0 {
+		return postCmd(NoMorePageMsg{})
+	}
+
+	var (
+		cmds []tea.Cmd
+		c    tea.Cmd
+	)
+	// 1. 处理旧页面的离开逻辑
+	if p, ok := nav.curr.(PageLife); ok {
+		_, c = p.OnLeaving()
+		cmds = append(cmds, c)
+	}
+
+	// 2. 处理新页面的进入逻辑
+	nav.curr = nav.histories[len(nav.histories)-1]
+	nav.histories = nav.histories[:len(nav.histories)-1]
+
+	if p, ok := nav.curr.(PageLife); ok {
+		m, cmd := p.OnEntering()
+		nav.curr = m
+		cmds = append(cmds, cmd)
+	}
+
+	return tea.Sequence(cmds...)
+}