mirror of
https://github.com/Jguer/yay.git
synced 2024-11-07 17:47:21 +01:00
113 lines
2.7 KiB
Go
113 lines
2.7 KiB
Go
package settings
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/leonelquinteros/gotext"
|
|
|
|
"github.com/Jguer/yay/v12/pkg/db"
|
|
"github.com/Jguer/yay/v12/pkg/query"
|
|
"github.com/Jguer/yay/v12/pkg/settings/exe"
|
|
"github.com/Jguer/yay/v12/pkg/settings/parser"
|
|
"github.com/Jguer/yay/v12/pkg/text"
|
|
"github.com/Jguer/yay/v12/pkg/vcs"
|
|
|
|
"github.com/Jguer/aur"
|
|
"github.com/Jguer/aur/metadata"
|
|
"github.com/Jguer/aur/rpc"
|
|
"github.com/Jguer/votar/pkg/vote"
|
|
"github.com/Morganamilo/go-pacmanconf"
|
|
)
|
|
|
|
type Runtime struct {
|
|
QueryBuilder query.Builder
|
|
PacmanConf *pacmanconf.Config
|
|
VCSStore vcs.Store
|
|
CmdBuilder exe.ICmdBuilder
|
|
HTTPClient *http.Client
|
|
AURClient *rpc.Client
|
|
VoteClient *vote.Client
|
|
AURCache aur.QueryClient
|
|
DBExecutor db.Executor
|
|
Logger *text.Logger
|
|
}
|
|
|
|
func BuildRuntime(cfg *Configuration, cmdArgs *parser.Arguments, version string) (*Runtime, error) {
|
|
logger := text.NewLogger(os.Stdout, os.Stderr, os.Stdin, cfg.Debug, "runtime")
|
|
cmdBuilder := cfg.CmdBuilder(nil)
|
|
|
|
httpClient := &http.Client{
|
|
CheckRedirect: func(req *http.Request, via []*http.Request) error {
|
|
return http.ErrUseLastResponse
|
|
},
|
|
}
|
|
|
|
userAgent := fmt.Sprintf("Yay/%s", version)
|
|
voteClient, errVote := vote.NewClient(vote.WithUserAgent(userAgent),
|
|
vote.WithHTTPClient(httpClient))
|
|
if errVote != nil {
|
|
return nil, errVote
|
|
}
|
|
|
|
voteClient.SetCredentials(
|
|
os.Getenv("AUR_USERNAME"),
|
|
os.Getenv("AUR_PASSWORD"))
|
|
|
|
userAgentFn := func(ctx context.Context, req *http.Request) error {
|
|
req.Header.Set("User-Agent", userAgent)
|
|
return nil
|
|
}
|
|
|
|
var aurCache aur.QueryClient
|
|
aurCache, errAURCache := metadata.New(
|
|
metadata.WithHTTPClient(httpClient),
|
|
metadata.WithCacheFilePath(filepath.Join(cfg.BuildDir, "aur.json")),
|
|
metadata.WithRequestEditorFn(userAgentFn),
|
|
metadata.WithBaseURL(cfg.AURURL),
|
|
metadata.WithDebugLogger(logger.Debugln),
|
|
)
|
|
if errAURCache != nil {
|
|
return nil, fmt.Errorf(gotext.Get("failed to retrieve aur Cache")+": %w", errAURCache)
|
|
}
|
|
|
|
aurClient, errAUR := rpc.NewClient(
|
|
rpc.WithHTTPClient(httpClient),
|
|
rpc.WithBaseURL(cfg.AURRPCURL),
|
|
rpc.WithRequestEditorFn(userAgentFn),
|
|
rpc.WithLogFn(logger.Debugln))
|
|
if errAUR != nil {
|
|
return nil, errAUR
|
|
}
|
|
|
|
if cfg.UseRPC {
|
|
aurCache = aurClient
|
|
}
|
|
|
|
vcsStore := vcs.NewInfoStore(
|
|
cfg.VCSFilePath, cmdBuilder,
|
|
logger.Child("vcs"))
|
|
|
|
if err := vcsStore.Load(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
runtime := &Runtime{
|
|
QueryBuilder: nil,
|
|
PacmanConf: nil,
|
|
VCSStore: vcsStore,
|
|
CmdBuilder: cmdBuilder,
|
|
HTTPClient: &http.Client{},
|
|
AURClient: aurClient,
|
|
VoteClient: voteClient,
|
|
AURCache: aurCache,
|
|
DBExecutor: nil,
|
|
Logger: text.NewLogger(os.Stdout, os.Stderr, os.Stdin, cfg.Debug, "runtime"),
|
|
}
|
|
|
|
return runtime, nil
|
|
}
|