feat: initial commit
This commit is contained in:
eric
16
pkg/wails3kit/updates/providers/githubreleases/BUILD.bazel
Normal file
16
pkg/wails3kit/updates/providers/githubreleases/BUILD.bazel
Normal file
@@ -0,0 +1,16 @@
|
||||
load("@io_bazel_rules_go//go:def.bzl", "go_library", "go_test")
|
||||
|
||||
go_library(
|
||||
name = "githubreleases",
|
||||
srcs = ["provider.go"],
|
||||
importpath = "github.com/Eriyc/rules_wails/pkg/wails3kit/updates/providers/githubreleases",
|
||||
visibility = ["//visibility:public"],
|
||||
deps = ["//pkg/wails3kit/updates"],
|
||||
)
|
||||
|
||||
go_test(
|
||||
name = "githubreleases_test",
|
||||
srcs = ["provider_test.go"],
|
||||
embed = [":githubreleases"],
|
||||
deps = ["//pkg/wails3kit/updates"],
|
||||
)
|
||||
259
pkg/wails3kit/updates/providers/githubreleases/provider.go
Normal file
259
pkg/wails3kit/updates/providers/githubreleases/provider.go
Normal file
@@ -0,0 +1,259 @@
|
||||
package githubreleases
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"strings"
|
||||
"text/template"
|
||||
"time"
|
||||
|
||||
"github.com/Eriyc/rules_wails/pkg/wails3kit/updates"
|
||||
)
|
||||
|
||||
const (
|
||||
defaultAssetTemplate = "{{ .ProductID }}_{{ .Version }}_{{ .OS }}_{{ .Arch }}.zip"
|
||||
defaultChecksumAssetTemplate = "{{ .AssetName }}.sha256"
|
||||
)
|
||||
|
||||
type Config struct {
|
||||
Owner string
|
||||
Repo string
|
||||
HTTPClient *http.Client
|
||||
PrepareRequest func(*http.Request) error
|
||||
AssetNameTemplate string
|
||||
ChecksumAssetNameTemplate string
|
||||
}
|
||||
|
||||
type Provider struct {
|
||||
config Config
|
||||
client *http.Client
|
||||
assetNameTemplate *template.Template
|
||||
checksumNameTemplate *template.Template
|
||||
baseURL string
|
||||
}
|
||||
|
||||
func New(cfg Config) (*Provider, error) {
|
||||
if cfg.Owner == "" || cfg.Repo == "" {
|
||||
return nil, updates.ErrInvalidConfig
|
||||
}
|
||||
client := cfg.HTTPClient
|
||||
if client == nil {
|
||||
client = http.DefaultClient
|
||||
}
|
||||
assetTemplate := cfg.AssetNameTemplate
|
||||
if assetTemplate == "" {
|
||||
assetTemplate = defaultAssetTemplate
|
||||
}
|
||||
checksumTemplate := cfg.ChecksumAssetNameTemplate
|
||||
if checksumTemplate == "" {
|
||||
checksumTemplate = defaultChecksumAssetTemplate
|
||||
}
|
||||
|
||||
assetNameTemplate, err := template.New("asset").Parse(assetTemplate)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
checksumNameTemplate, err := template.New("checksum").Parse(checksumTemplate)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &Provider{
|
||||
config: cfg,
|
||||
client: client,
|
||||
assetNameTemplate: assetNameTemplate,
|
||||
checksumNameTemplate: checksumNameTemplate,
|
||||
baseURL: "https://api.github.com",
|
||||
}, nil
|
||||
}
|
||||
|
||||
type githubRelease struct {
|
||||
TagName string `json:"tag_name"`
|
||||
Body string `json:"body"`
|
||||
Prerelease bool `json:"prerelease"`
|
||||
PublishedAt time.Time `json:"published_at"`
|
||||
Assets []githubAsset `json:"assets"`
|
||||
}
|
||||
|
||||
type githubAsset struct {
|
||||
Name string `json:"name"`
|
||||
BrowserDownloadURL string `json:"browser_download_url"`
|
||||
}
|
||||
|
||||
func (provider *Provider) Resolve(ctx context.Context, req updates.ResolveRequest) (*updates.Release, error) {
|
||||
request, err := http.NewRequestWithContext(ctx, http.MethodGet, provider.releaseURL(), nil)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if provider.config.PrepareRequest != nil {
|
||||
if err := provider.config.PrepareRequest(request); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
|
||||
response, err := provider.client.Do(request)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer response.Body.Close()
|
||||
|
||||
if response.StatusCode < 200 || response.StatusCode >= 300 {
|
||||
return nil, fmt.Errorf("unexpected github status: %s", response.Status)
|
||||
}
|
||||
|
||||
var releases []githubRelease
|
||||
if err := json.NewDecoder(response.Body).Decode(&releases); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var best *updates.Release
|
||||
for _, release := range releases {
|
||||
version := strings.TrimPrefix(release.TagName, "v")
|
||||
channel := mapChannel(release)
|
||||
if channel != req.App.Channel {
|
||||
continue
|
||||
}
|
||||
comparison, err := updates.CompareVersions(version, req.App.CurrentVersion)
|
||||
if err != nil || comparison <= 0 {
|
||||
continue
|
||||
}
|
||||
|
||||
assetName, err := provider.renderAssetName(req.App, version, "")
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
checksumName, err := provider.renderChecksumName(req.App, version, assetName)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
artifactURL := ""
|
||||
checksumURL := ""
|
||||
for _, asset := range release.Assets {
|
||||
if asset.Name == assetName {
|
||||
artifactURL = asset.BrowserDownloadURL
|
||||
}
|
||||
if asset.Name == checksumName {
|
||||
checksumURL = asset.BrowserDownloadURL
|
||||
}
|
||||
}
|
||||
if artifactURL == "" || checksumURL == "" {
|
||||
continue
|
||||
}
|
||||
|
||||
sha256Value, err := provider.readChecksum(ctx, checksumURL)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
candidate := &updates.Release{
|
||||
ID: version,
|
||||
Version: version,
|
||||
Channel: channel,
|
||||
NotesMarkdown: release.Body,
|
||||
PublishedAt: release.PublishedAt,
|
||||
Artifact: updates.Artifact{
|
||||
Kind: updates.ArtifactKindBundleArchive,
|
||||
Format: updates.ArtifactFormatZip,
|
||||
URL: artifactURL,
|
||||
SHA256: sha256Value,
|
||||
},
|
||||
}
|
||||
if best == nil {
|
||||
best = candidate
|
||||
continue
|
||||
}
|
||||
better, err := updates.CompareVersions(candidate.Version, best.Version)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if better > 0 {
|
||||
best = candidate
|
||||
}
|
||||
}
|
||||
|
||||
return best, nil
|
||||
}
|
||||
|
||||
func (provider *Provider) OpenArtifact(ctx context.Context, release updates.Release) (io.ReadCloser, error) {
|
||||
request, err := http.NewRequestWithContext(ctx, http.MethodGet, release.Artifact.URL, nil)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if provider.config.PrepareRequest != nil {
|
||||
if err := provider.config.PrepareRequest(request); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
response, err := provider.client.Do(request)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if response.StatusCode < 200 || response.StatusCode >= 300 {
|
||||
defer response.Body.Close()
|
||||
return nil, fmt.Errorf("unexpected artifact status: %s", response.Status)
|
||||
}
|
||||
return response.Body, nil
|
||||
}
|
||||
|
||||
func (provider *Provider) readChecksum(ctx context.Context, url string) (string, error) {
|
||||
request, err := http.NewRequestWithContext(ctx, http.MethodGet, url, nil)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
if provider.config.PrepareRequest != nil {
|
||||
if err := provider.config.PrepareRequest(request); err != nil {
|
||||
return "", err
|
||||
}
|
||||
}
|
||||
response, err := provider.client.Do(request)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
defer response.Body.Close()
|
||||
if response.StatusCode < 200 || response.StatusCode >= 300 {
|
||||
return "", fmt.Errorf("unexpected checksum status: %s", response.Status)
|
||||
}
|
||||
bytes, err := io.ReadAll(response.Body)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
return strings.Fields(string(bytes))[0], nil
|
||||
}
|
||||
|
||||
func (provider *Provider) releaseURL() string {
|
||||
return fmt.Sprintf("%s/repos/%s/%s/releases", provider.baseURL, provider.config.Owner, provider.config.Repo)
|
||||
}
|
||||
|
||||
func (provider *Provider) renderAssetName(app updates.AppDescriptor, version string, assetName string) (string, error) {
|
||||
return provider.executeTemplate(provider.assetNameTemplate, app, version, assetName)
|
||||
}
|
||||
|
||||
func (provider *Provider) renderChecksumName(app updates.AppDescriptor, version string, assetName string) (string, error) {
|
||||
return provider.executeTemplate(provider.checksumNameTemplate, app, version, assetName)
|
||||
}
|
||||
|
||||
func (provider *Provider) executeTemplate(templateValue *template.Template, app updates.AppDescriptor, version string, assetName string) (string, error) {
|
||||
var builder strings.Builder
|
||||
err := templateValue.Execute(&builder, map[string]string{
|
||||
"ProductID": app.ProductID,
|
||||
"Version": version,
|
||||
"OS": app.OS,
|
||||
"Arch": app.Arch,
|
||||
"AssetName": assetName,
|
||||
})
|
||||
return builder.String(), err
|
||||
}
|
||||
|
||||
func mapChannel(release githubRelease) updates.Channel {
|
||||
if !release.Prerelease {
|
||||
return updates.ChannelStable
|
||||
}
|
||||
if strings.Contains(strings.ToLower(release.TagName), "alpha") {
|
||||
return updates.ChannelAlpha
|
||||
}
|
||||
return updates.ChannelBeta
|
||||
}
|
||||
@@ -0,0 +1,72 @@
|
||||
package githubreleases
|
||||
|
||||
import (
|
||||
"context"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"testing"
|
||||
|
||||
"github.com/Eriyc/rules_wails/pkg/wails3kit/updates"
|
||||
)
|
||||
|
||||
func TestResolveSelectsReleaseFromGitHubAPI(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
serverURL := ""
|
||||
server := httptest.NewServer(http.HandlerFunc(func(writer http.ResponseWriter, request *http.Request) {
|
||||
switch request.URL.Path {
|
||||
case "/repos/owner/repo/releases":
|
||||
_, _ = writer.Write([]byte(`[
|
||||
{
|
||||
"tag_name": "v1.2.0",
|
||||
"body": "notes",
|
||||
"prerelease": false,
|
||||
"published_at": "2026-03-01T03:10:56Z",
|
||||
"assets": [
|
||||
{"name": "com.example.app_1.2.0_linux_amd64.zip", "browser_download_url": "` + serverURL + `/artifact.zip"},
|
||||
{"name": "com.example.app_1.2.0_linux_amd64.zip.sha256", "browser_download_url": "` + serverURL + `/artifact.zip.sha256"}
|
||||
]
|
||||
}
|
||||
]`))
|
||||
case "/artifact.zip.sha256":
|
||||
_, _ = writer.Write([]byte("abcdef artifact.zip"))
|
||||
default:
|
||||
http.NotFound(writer, request)
|
||||
}
|
||||
}))
|
||||
defer server.Close()
|
||||
serverURL = server.URL
|
||||
|
||||
provider, err := New(Config{
|
||||
Owner: "owner",
|
||||
Repo: "repo",
|
||||
HTTPClient: server.Client(),
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("New returned error: %v", err)
|
||||
}
|
||||
|
||||
provider.baseURL = server.URL
|
||||
|
||||
release, err := provider.Resolve(context.Background(), updates.ResolveRequest{
|
||||
App: updates.AppDescriptor{
|
||||
ProductID: "com.example.app",
|
||||
CurrentVersion: "1.0.0",
|
||||
Channel: updates.ChannelStable,
|
||||
OS: "linux",
|
||||
Arch: "amd64",
|
||||
},
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("Resolve returned error: %v", err)
|
||||
}
|
||||
if release == nil {
|
||||
t.Fatal("Resolve returned nil release")
|
||||
}
|
||||
if release.Artifact.URL != server.URL+"/artifact.zip" {
|
||||
t.Fatalf("unexpected artifact url: %s", release.Artifact.URL)
|
||||
}
|
||||
if release.Artifact.SHA256 != "abcdef" {
|
||||
t.Fatalf("unexpected checksum: %s", release.Artifact.SHA256)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user