FilesExpand file tree

main

/

gui_tray_provider.go

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

122 lines (109 loc) · 3.15 KB

main

/

gui_tray_provider.go

Copy path

Top

File metadata and controls

• Code
• Blame

122 lines (109 loc) · 3.15 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

//go:build wailsgui && windows

package main

import (

"context"

"encoding/json"

"fmt"

"net/http"

"net/url"

"strings"

"time"

)

type guiTrayProvider struct {

ID string `json:"id"`

Name string `json:"name"`

BaseURL string `json:"base_url"`

IsActive bool `json:"is_active"`

}

type guiTrayProviderSnapshot struct {

Providers []guiTrayProvider

ActiveID string

ActiveName string

OfficialActive bool

ConfigMatchesActive bool

}

func (s guiTrayProviderSnapshot) currentName() string {

if s.OfficialActive {

return "官方账号"

}

if s.ActiveName != "" {

return s.ActiveName

}

return "未设置"

}

func (s guiTrayProviderSnapshot) drifted() bool {

return s.ActiveID != "" && !s.ConfigMatchesActive

}

func (s guiTrayProviderSnapshot) fingerprint() string {

data, _ := json.Marshal(s)

return string(data)

}

type guiTrayProviderClient struct {

baseURL string

client *http.Client

}

func newGUITrayProviderClient(baseURL string) *guiTrayProviderClient {

return &guiTrayProviderClient{

baseURL: strings.TrimRight(baseURL, "/"),

client: &http.Client{Timeout: 3 * time.Second},

}

}

func (c *guiTrayProviderClient) snapshot(ctx context.Context) (guiTrayProviderSnapshot, error) {

var status struct {

ActiveProfile struct {

ID string `json:"id"`

Name string `json:"name"`

} `json:"active_profile"`

OfficialActive bool `json:"official_active"`

ConfigMatchesActive bool `json:"config_matches_active"`

}

if err := c.do(ctx, http.MethodGet, "/api/status", &status); err != nil {

return guiTrayProviderSnapshot{}, err

}

var providers []guiTrayProvider

if err := c.do(ctx, http.MethodGet, "/api/profiles", &providers); err != nil {

return guiTrayProviderSnapshot{}, err

}

return guiTrayProviderSnapshot{

Providers: providers,

ActiveID: status.ActiveProfile.ID,

ActiveName: status.ActiveProfile.Name,

OfficialActive: status.OfficialActive,

ConfigMatchesActive: status.ConfigMatchesActive,

}, nil

}

func (c *guiTrayProviderClient) activate(ctx context.Context, id string) error {

return c.do(ctx, http.MethodPost, "/api/profiles/"+url.PathEscape(id)+"/activate", nil)

}

func (c *guiTrayProviderClient) activateOfficial(ctx context.Context) error {

return c.do(ctx, http.MethodPost, "/api/official/activate", nil)

}

func (c *guiTrayProviderClient) do(ctx context.Context, method, path string, out any) error {

req, err := http.NewRequestWithContext(ctx, method, c.baseURL+path, nil)

if err != nil {

return err

}

req.Header.Set("Accept", "application/json")

resp, err := c.client.Do(req)

if err != nil {

return err

}

defer resp.Body.Close()

if resp.StatusCode < http.StatusOK || resp.StatusCode >= http.StatusMultipleChoices {

var apiError struct {

Error string `json:"error"`

}

_ = json.NewDecoder(resp.Body).Decode(&apiError)

if apiError.Error != "" {

return fmt.Errorf("%s", apiError.Error)

}

return fmt.Errorf("本地服务返回 HTTP %d", resp.StatusCode)

}

if out == nil {

return nil

}

if err := json.NewDecoder(resp.Body).Decode(out); err != nil {

return fmt.Errorf("解析本地服务响应失败: %w", err)

}

return nil

}