1
0
mirror of https://github.com/fumiama/RVC-Models-Downloader.git synced 2024-09-29 14:26:25 +09:00
This commit is contained in:
源文雨 2024-04-18 01:52:23 +09:00
parent c5347fa574
commit 04d3d611b0
40 changed files with 1073 additions and 0 deletions

25
.github/workflows/release.yml vendored Normal file
View File

@ -0,0 +1,25 @@
name: 发行版
on:
push:
tags:
- 'v*'
jobs:
goreleaser:
runs-on: ubuntu-latest
steps:
- name: Checkout
uses: actions/checkout@master
with:
fetch-depth: 0
- name: Set up Go
uses: actions/setup-go@master
- name: Run GoReleaser
uses: goreleaser/goreleaser-action@master
with:
version: latest
args: release --clean
env:
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}

6
.gitignore vendored
View File

@ -19,3 +19,9 @@
# Go workspace file
go.work
# Packed configs
*.zip
# MacOS system file
.DS_Store

89
.goreleaser.yml Normal file
View File

@ -0,0 +1,89 @@
project_name: rvcmd
env:
- GO111MODULE=on
before:
hooks:
- go mod tidy
- go generate
builds:
- id: linux
env:
- CGO_ENABLED=0
- GO111MODULE=on
goos:
- linux
goarch:
- 386
- amd64
- arm
- arm64
goarm:
- 6
- 7
mod_timestamp: "{{ .CommitTimestamp }}"
flags:
- -trimpath
ldflags:
- -s -w
- id: macos
env:
- CGO_ENABLED=0
- GO111MODULE=on
goos:
- darwin
goarch:
- amd64
- arm64
mod_timestamp: "{{ .CommitTimestamp }}"
flags:
- -trimpath
ldflags:
- -s -w
- id: windows
env:
- CGO_ENABLED=0
- GO111MODULE=on
goos:
- windows
goarch:
- 386
- amd64
mod_timestamp: "{{ .CommitTimestamp }}"
flags:
- -trimpath
ldflags:
- -s -w
checksum:
name_template: "rvcmd_checksums.txt"
changelog:
sort: asc
filters:
exclude:
- "^docs:"
- "^test:"
- fix typo
- Merge pull request
- Merge branch
- Merge remote-tracking
- go mod tidy
archives:
- id: all
builds:
- linux
- macos
- windows
name_template: "rvcmd_{{ .Os }}_{{ .Arch }}{{ if .Arm }}v{{ .Arm }}{{ end }}"
format_overrides:
- goos: windows
format: zip
nfpms:
- license: AGPL 3.0
homepage: https://github.com/RVC-Project/RVC-Models-Downloader
file_name_template: "rvcmd_{{ .Os }}_{{ .Arch }}{{ if .Arm }}v{{ .Arm }}{{ end }}"
formats:
- deb
- rpm
maintainer: Fumiama Minamoto

7
assets/all.yaml Normal file
View File

@ -0,0 +1,7 @@
BaseURL: https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main
Targets:
- Refer: hubert
- Refer: rmvpe
- Refer: v1
- Refer: v2
- Refer: uvr5

5
assets/hubert.yaml Normal file
View File

@ -0,0 +1,5 @@
BaseURL: https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main
Targets:
- Folder: assets/hubert
Copy:
- hubert_base.pt

6
assets/rmvpe.yaml Normal file
View File

@ -0,0 +1,6 @@
BaseURL: https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main
Targets:
- Folder: .
Copy:
- rmvpe.pt
- rmvpe.onnx

15
assets/uvr5.yaml Normal file
View File

@ -0,0 +1,15 @@
BaseURL: https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main
Targets:
- Folder: assets/uvr5_weights
Copy:
- uvr5_weights/HP2-人声vocals+非人声instrumentals.pth
- uvr5_weights/HP2_all_vocals.pth
- uvr5_weights/HP3_all_vocals.pth
- uvr5_weights/HP5-主旋律人声vocals+其他instrumentals.pth
- uvr5_weights/HP5_only_main_vocal.pth
- uvr5_weights/VR-DeEchoAggressive.pth
- uvr5_weights/VR-DeEchoDeReverb.pth
- uvr5_weights/VR-DeEchoNormal.pth
- Folder: assets/uvr5_weights/onnx_dereverb_By_FoxJoy
Copy:
- uvr5_weights/onnx_dereverb_By_FoxJoy/vocals.onnx

18
assets/v1.yaml Normal file
View File

@ -0,0 +1,18 @@
BaseURL: https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main
Targets:
- Refer: hubert
- Folder: assets/pretrained
Copy:
- pretrained/D32k.pth
- pretrained/D40k.pth
- pretrained/D48k.pth
- pretrained/G32k.pth
- pretrained/G40k.pth
- pretrained/G48k.pth
- pretrained/f0D32k.pth
- pretrained/f0D40k.pth
- pretrained/f0D48k.pth
- pretrained/f0G32k.pth
- pretrained/f0G40k.pth
- pretrained/f0G48k.pth
- Refer: uvr5

18
assets/v2.yaml Normal file
View File

@ -0,0 +1,18 @@
BaseURL: https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main
Targets:
- Refer: hubert
- Folder: assets/pretrained_v2
Copy:
- pretrained_v2/D32k.pth
- pretrained_v2/D40k.pth
- pretrained_v2/D48k.pth
- pretrained_v2/G32k.pth
- pretrained_v2/G40k.pth
- pretrained_v2/G48k.pth
- pretrained_v2/f0D32k.pth
- pretrained_v2/f0D40k.pth
- pretrained_v2/f0D48k.pth
- pretrained_v2/f0G32k.pth
- pretrained_v2/f0G40k.pth
- pretrained_v2/f0G48k.pth
- Refer: uvr5

143
cfg.go Normal file
View File

@ -0,0 +1,143 @@
package main
import (
"archive/zip"
"bytes"
_ "embed"
"fmt"
"io"
"net/http"
"os"
"runtime"
"strings"
"sync"
"time"
"github.com/pkg/errors"
"github.com/sirupsen/logrus"
"gopkg.in/yaml.v3"
)
//go:embed cfg.zip
var cfgdata []byte
var cfg = func() *zip.Reader {
r, err := zip.NewReader(bytes.NewReader(cfgdata), int64(len(cfgdata)))
if err != nil {
panic(err)
}
for _, f := range r.File {
cmdlst = append(cmdlst, f.Name)
}
return r
}()
func readconfig(path string) (c config, err error) {
fname := path + ".yaml"
f, err := cfg.Open(fname)
if err != nil {
err = errors.Wrap(err, "invalid path")
return
}
defer f.Close()
err = yaml.NewDecoder(f).Decode(&c)
if err != nil {
err = errors.Wrap(err, "invalid config")
return
}
return
}
type config struct {
BaseURL string `yaml:"BaseURL"`
Targets []targets `yaml:"Targets"`
}
type targets struct {
Refer string `yaml:"Refer"`
Folder string `yaml:"Folder"`
Copy []string `yaml:"Copy"`
OS string `yaml:"OS"`
Arch string `yaml:"Arch"`
}
func (c *config) download(path string) error {
for i, t := range c.Targets {
if t.Refer != "" {
refp := path[:strings.LastIndex(path, "/")+1] + t.Refer
logrus.Infof("#%d refer to target '%s'", i+1, refp)
refcfg, err := readconfig(refp)
if err != nil {
return err
}
err = refcfg.download(refp)
if err != nil {
return err
}
continue
}
if t.OS != "" && t.OS != runtime.GOOS {
logrus.Warnf("#%d target required OS: %s but you are %s, skip.", i+1, t.OS, runtime.GOOS)
continue
}
if t.Arch != "" && t.Arch != runtime.GOARCH {
logrus.Warnf("#%d target required Arch: %s but you are %s, skip.", i+1, t.Arch, runtime.GOARCH)
continue
}
err := os.MkdirAll(t.Folder, 0755)
if err != nil {
return errors.Wrap(err, fmt.Sprintf("#%d make target folder '%s'", i+1, t.Folder))
}
logrus.Infof("#%d open target folder '%s'", i+1, t.Folder)
if len(t.Copy) == 0 {
logrus.Warningf("#%d empty copy target", i+1)
continue
}
wg := sync.WaitGroup{}
wg.Add(len(t.Copy))
logrus.Infof("#%d download copy: %v", i+1, t.Copy)
for i, cp := range t.Copy {
go func(i int, cp string) {
defer wg.Done()
sleep := time.Millisecond * 100 * time.Duration(i)
if sleep > time.Millisecond {
time.Sleep(sleep)
}
req, err := http.NewRequest("GET", c.BaseURL+"/"+cp, nil)
if err != nil {
logrus.Errorln("new request to", cp, "err:", err)
return
}
logrus.Infoln("get:", req.URL)
req.Header.Add("user-agent", ua)
resp, err := cli.Do(req)
if err != nil {
logrus.Errorln("get", req.URL, "err:", err)
return
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
err := errors.New(fmt.Sprintf("HTTP %d %s", resp.StatusCode, resp.Status))
logrus.Errorln("get", req.URL, "err:", err)
return
}
fname := t.Folder + "/" + cp
f, err := os.Create(fname)
if err != nil {
logrus.Errorln("create file", fname, "err:", err)
return
}
logrus.Infoln("writing file", fname)
defer f.Close()
_, err = io.Copy(f, resp.Body)
if err != nil {
logrus.Errorln("download file", fname, "err:", err)
return
}
logrus.Infoln("downloaded file", fname)
}(i, cp)
}
wg.Wait()
}
return nil
}

49
cmd.go Normal file
View File

@ -0,0 +1,49 @@
package main
import (
"strings"
)
type commandlist []string
var cmdlst = make(commandlist, 0, 64)
func (cl commandlist) String() string {
sb := strings.Builder{}
islastdir := false
isfirstloop := true
sb.WriteString(" ")
for _, cmd := range cmdlst {
if len(cmd) == 0 {
continue
}
trimedcmd := strings.TrimSuffix(cmd, "/")
a := strings.LastIndex(trimedcmd, "/") + 1
b := len(cmd) - 1
if a >= b {
continue
}
isdir := cmd[b] == '/'
ident := strings.Count(trimedcmd, "/") + 1
if !isfirstloop && (islastdir || isdir) {
sb.WriteByte('\n')
if !isdir {
ident--
}
for i := 0; i < ident; i++ {
sb.WriteString(" ")
}
}
isfirstloop = false
if isdir {
islastdir = true
sb.WriteString(cmd[a:b])
sb.WriteByte(':')
} else { // is file
islastdir = false
sb.WriteString(" ")
sb.WriteString(strings.TrimSuffix(cmd[a:], ".yaml"))
}
}
return sb.String()
}

14
console/console_ansi.go Normal file
View File

@ -0,0 +1,14 @@
//go:build !windows
// Package console sets console's behavior on init
package console
// https://github.com/FloatTech/ZeroBot-Plugin/blob/master/console/console_ansi.go
import (
"fmt"
)
func init() {
fmt.Print("\033]0;RVC Models Downloader\007")
}

144
console/console_windows.go Normal file
View File

@ -0,0 +1,144 @@
// Package console sets console's behavior on init
package console
// https://github.com/FloatTech/ZeroBot-Plugin/blob/master/console/console_windows.go
import (
"bytes"
"os"
"strings"
"syscall"
"unsafe"
"golang.org/x/sys/windows"
"github.com/sirupsen/logrus"
)
var (
//go:linkname modkernel32 golang.org/x/sys/windows.modkernel32
modkernel32 *windows.LazyDLL
procSetConsoleTitle = modkernel32.NewProc("SetConsoleTitleW")
)
//go:linkname errnoErr golang.org/x/sys/windows.errnoErr
func errnoErr(e syscall.Errno) error
func setConsoleTitle(title string) (err error) {
var p0 *uint16
p0, err = syscall.UTF16PtrFromString(title)
if err != nil {
return
}
r1, _, e1 := syscall.Syscall(procSetConsoleTitle.Addr(), 1, uintptr(unsafe.Pointer(p0)), 0, 0)
if r1 == 0 {
err = errnoErr(e1)
}
return
}
func init() {
stdin := windows.Handle(os.Stdin.Fd())
var mode uint32
err := windows.GetConsoleMode(stdin, &mode)
if err != nil {
panic(err)
}
mode &^= windows.ENABLE_QUICK_EDIT_MODE // 禁用快速编辑模式
mode |= windows.ENABLE_EXTENDED_FLAGS // 启用扩展标志
mode &^= windows.ENABLE_MOUSE_INPUT // 禁用鼠标输入
mode |= windows.ENABLE_PROCESSED_INPUT // 启用控制输入
mode &^= windows.ENABLE_INSERT_MODE // 禁用插入模式
mode |= windows.ENABLE_ECHO_INPUT | windows.ENABLE_LINE_INPUT // 启用输入回显&逐行输入
mode &^= windows.ENABLE_WINDOW_INPUT // 禁用窗口输入
mode &^= windows.ENABLE_VIRTUAL_TERMINAL_INPUT // 禁用虚拟终端输入
err = windows.SetConsoleMode(stdin, mode)
if err != nil {
panic(err)
}
stdout := windows.Handle(os.Stdout.Fd())
err = windows.GetConsoleMode(stdout, &mode)
if err != nil {
panic(err)
}
mode |= windows.ENABLE_VIRTUAL_TERMINAL_PROCESSING // 启用虚拟终端处理
mode |= windows.ENABLE_PROCESSED_OUTPUT // 启用处理后的输出
err = windows.SetConsoleMode(stdout, mode)
// windows 带颜色 log 自定义格式
logrus.SetFormatter(&logFormat{hasColor: err == nil})
if err != nil {
logrus.Warnln("VT100设置失败, 将以无色模式输出")
}
err = setConsoleTitle("RVC Models Downloader")
if err != nil {
panic(err)
}
}
const (
colorCodePanic = "\x1b[1;31m" // color.Style{color.Bold, color.Red}.String()
colorCodeFatal = "\x1b[1;31m" // color.Style{color.Bold, color.Red}.String()
colorCodeError = "\x1b[31m" // color.Style{color.Red}.String()
colorCodeWarn = "\x1b[33m" // color.Style{color.Yellow}.String()
colorCodeInfo = "\x1b[37m" // color.Style{color.White}.String()
colorCodeDebug = "\x1b[32m" // color.Style{color.Green}.String()
colorCodeTrace = "\x1b[36m" // color.Style{color.Cyan}.String()
colorReset = "\x1b[0m"
)
// logFormat specialize for zbp
type logFormat struct {
hasColor bool
}
// Format implements logrus.Formatter
func (f logFormat) Format(entry *logrus.Entry) ([]byte, error) {
buf := new(bytes.Buffer)
buf.WriteByte('[')
if f.hasColor {
buf.WriteString(getLogLevelColorCode(entry.Level))
}
buf.WriteString(strings.ToUpper(entry.Level.String()))
if f.hasColor {
buf.WriteString(colorReset)
}
buf.WriteString("] ")
buf.WriteString(entry.Message)
buf.WriteString(" \n")
return buf.Bytes(), nil
}
// getLogLevelColorCode 获取日志等级对应色彩code
func getLogLevelColorCode(level logrus.Level) string {
switch level {
case logrus.PanicLevel:
return colorCodePanic
case logrus.FatalLevel:
return colorCodeFatal
case logrus.ErrorLevel:
return colorCodeError
case logrus.WarnLevel:
return colorCodeWarn
case logrus.InfoLevel:
return colorCodeInfo
case logrus.DebugLevel:
return colorCodeDebug
case logrus.TraceLevel:
return colorCodeTrace
default:
return colorCodeInfo
}
}

163
dns.go Normal file
View File

@ -0,0 +1,163 @@
package main
// https://github.com/fumiama/comandy
import (
"context"
"crypto/tls"
"errors"
"net"
"sync"
"time"
"github.com/fumiama/terasu"
)
var (
errNoDNSAvailable = errors.New("no dns available")
)
var dnsdialer = net.Dialer{
Timeout: time.Second * 8,
}
type dnsstat struct {
A string
E bool
}
type dnsservers struct {
sync.RWMutex
m map[string][]*dnsstat
}
// hasrecord no lock, use under lock
func hasrecord(lst []*dnsstat, a string) bool {
for _, addr := range lst {
if addr.A == a {
return true
}
}
return false
}
func (ds *dnsservers) add(m map[string][]string) {
ds.Lock()
defer ds.Unlock()
addList := map[string][]*dnsstat{}
for host, addrs := range m {
for _, addr := range addrs {
if !hasrecord(ds.m[host], addr) && !hasrecord(addList[host], addr) {
addList[host] = append(addList[host], &dnsstat{addr, true})
}
}
}
for host, addrs := range addList {
ds.m[host] = append(ds.m[host], addrs...)
}
}
func (ds *dnsservers) dial(ctx context.Context) (tlsConn *tls.Conn, err error) {
err = errNoDNSAvailable
ds.RLock()
defer ds.RUnlock()
if dnsdialer.Timeout != 0 {
var cancel context.CancelFunc
ctx, cancel = context.WithTimeout(ctx, dnsdialer.Timeout)
defer cancel()
}
if !dnsdialer.Deadline.IsZero() {
var cancel context.CancelFunc
ctx, cancel = context.WithDeadline(ctx, dnsdialer.Deadline)
defer cancel()
}
var conn net.Conn
for host, addrs := range ds.m {
for _, addr := range addrs {
if !addr.E {
continue
}
conn, err = dnsdialer.DialContext(ctx, "tcp", addr.A)
if err != nil {
addr.E = false // no need to acquire write lock
continue
}
tlsConn = tls.Client(conn, &tls.Config{ServerName: host})
if usetrs {
err = terasu.Use(tlsConn).HandshakeContext(ctx)
} else {
err = tlsConn.HandshakeContext(ctx)
}
if err == nil {
return
}
_ = tlsConn.Close()
addr.E = false // no need to acquire write lock
}
}
return
}
var dotv6servers = dnsservers{
m: map[string][]*dnsstat{
"dot.sb": {
{"[2a09::]:853", true},
{"[2a11::]:853", true},
},
"dns.google": {
{"[2001:4860:4860::8888]:853", true},
{"[2001:4860:4860::8844]:853", true},
},
"cloudflare-dns.com": {
{"[2606:4700:4700::1111]:853", true},
{"[2606:4700:4700::1001]:853", true},
},
"dns.umbrella.com": {
{"[2620:0:ccc::2]:853", true},
{"[2620:0:ccd::2]:853", true},
},
"dns10.quad9.net": {
{"[2620:fe::10]:853", true},
{"[2620:fe::fe:10]:853", true},
},
},
}
var dotv4servers = dnsservers{
m: map[string][]*dnsstat{
"dot.sb": {
{"185.222.222.222:853", true},
{"45.11.45.11:853", true},
},
"dns.google": {
{"8.8.8.8:853", true},
{"8.8.4.4:853", true},
},
"cloudflare-dns.com": {
{"1.1.1.1:853", true},
{"1.0.0.1:853", true},
},
"dns.umbrella.com": {
{"208.67.222.222:853", true},
{"208.67.220.220:853", true},
},
"dns10.quad9.net": {
{"9.9.9.10:853", true},
{"149.112.112.10:853", true},
},
},
}
var resolver = &net.Resolver{
PreferGo: true,
Dial: func(ctx context.Context, _, _ string) (net.Conn, error) {
if canUseIPv6.Get() {
return dotv6servers.dial(ctx)
}
return dotv4servers.dial(ctx)
},
}

19
go.mod Normal file
View File

@ -0,0 +1,19 @@
module rvcmd
go 1.22.1
require (
github.com/pkg/errors v0.9.1
github.com/sirupsen/logrus v1.9.3
gopkg.in/yaml.v3 v3.0.1
)
require (
github.com/FloatTech/ttl v0.0.0-20230307105452-d6f7b2b647d1
github.com/RomiChan/syncx v0.0.0-20221202055724-5f842c53020e
github.com/fumiama/terasu v0.0.0-20240416061047-62d3c9f6be80
golang.org/x/net v0.24.0
golang.org/x/sys v0.19.0
)
require golang.org/x/text v0.14.0 // indirect

31
go.sum Normal file
View File

@ -0,0 +1,31 @@
github.com/FloatTech/ttl v0.0.0-20230307105452-d6f7b2b647d1 h1:g4pTnDJUW4VbJ9NvoRfUvdjDrHz/6QhfN/LoIIpICbo=
github.com/FloatTech/ttl v0.0.0-20230307105452-d6f7b2b647d1/go.mod h1:fHZFWGquNXuHttu9dUYoKuNbm3dzLETnIOnm1muSfDs=
github.com/RomiChan/syncx v0.0.0-20221202055724-5f842c53020e h1:wR3MXQ3VbUlPKOOUwLOYgh/QaJThBTYtsl673O3lqSA=
github.com/RomiChan/syncx v0.0.0-20221202055724-5f842c53020e/go.mod h1:vD7Ra3Q9onRtojoY5sMCLQ7JBgjUsrXDnDKyFxqpf9w=
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/fumiama/terasu v0.0.0-20240416061047-62d3c9f6be80 h1:O1JJZzcd5ggUw/9X8V9KxBZ9JZGWFmX/r1q2TPg+pZQ=
github.com/fumiama/terasu v0.0.0-20240416061047-62d3c9f6be80/go.mod h1:BFl0X1+rGJf8bLHl/kO+v05ryHrj/R4kyCrK89NvegA=
github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=
github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
github.com/sirupsen/logrus v1.9.3 h1:dueUQJ1C2q9oE3F7wvmSGAaVtTmUizReu6fjN8uqzbQ=
github.com/sirupsen/logrus v1.9.3/go.mod h1:naHLuLoDiP4jHNo9R0sCBMtWGeIprob74mVsIT4qYEQ=
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
github.com/stretchr/testify v1.8.0 h1:pSgiaMZlXftHpm5L7V1+rVB+AZJydKsMxsQBIJw4PKk=
github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=
golang.org/x/net v0.24.0 h1:1PcaxkF854Fu3+lvBIx5SYn9wRlBzzcnHZSiaFFAb0w=
golang.org/x/net v0.24.0/go.mod h1:2Q7sJY5mzlzWjKtYUEXSlBWCdyaioyXzRB2RtU8KVE8=
golang.org/x/sys v0.0.0-20220715151400-c0bba94af5f8/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.19.0 h1:q5f1RH2jigJ1MoAWp2KTp3gm5zAGFUTarQZ5U386+4o=
golang.org/x/sys v0.19.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
golang.org/x/text v0.14.0 h1:ScX5w1eTa3QqT8oi6+ziP7dTV1S2+ALU0bI+0zXKWiQ=
golang.org/x/text v0.14.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU=
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=

87
http.go Normal file
View File

@ -0,0 +1,87 @@
package main
// https://github.com/fumiama/comandy
import (
"context"
"crypto/tls"
"errors"
"net"
"net/http"
"strings"
"time"
"github.com/FloatTech/ttl"
"github.com/fumiama/terasu"
"golang.org/x/net/http2"
)
var (
ErrEmptyHostAddress = errors.New("empty host addr")
)
var httpdialer = net.Dialer{
Timeout: time.Minute,
}
var lookupTable = ttl.NewCache[string, []string](time.Hour)
var cli = http.Client{
Transport: &http2.Transport{
DialTLSContext: func(ctx context.Context, network, addr string, cfg *tls.Config) (net.Conn, error) {
if httpdialer.Timeout != 0 {
var cancel context.CancelFunc
ctx, cancel = context.WithTimeout(ctx, httpdialer.Timeout)
defer cancel()
}
if !httpdialer.Deadline.IsZero() {
var cancel context.CancelFunc
ctx, cancel = context.WithDeadline(ctx, httpdialer.Deadline)
defer cancel()
}
host, port, err := net.SplitHostPort(addr)
if err != nil {
return nil, err
}
addrs := lookupTable.Get(host)
if len(addrs) == 0 {
addrs, err = resolver.LookupHost(ctx, host)
if err != nil {
addrs, err = net.DefaultResolver.LookupHost(ctx, host)
if err != nil {
return nil, err
}
}
lookupTable.Set(host, addrs)
}
if len(addr) == 0 {
return nil, ErrEmptyHostAddress
}
var tlsConn *tls.Conn
for _, a := range addrs {
if strings.Contains(a, ":") {
a = "[" + a + "]:" + port
} else {
a += ":" + port
}
conn, err := httpdialer.DialContext(ctx, network, a)
if err != nil {
continue
}
tlsConn = tls.Client(conn, cfg)
if usetrs {
err = terasu.Use(tlsConn).HandshakeContext(ctx)
} else {
err = tlsConn.HandshakeContext(ctx)
}
if err == nil {
break
}
_ = tlsConn.Close()
}
return tlsConn, err
},
},
}

26
ipv6.go Normal file
View File

@ -0,0 +1,26 @@
package main
// https://github.com/fumiama/comandy
import (
"context"
"net/http"
"time"
"github.com/RomiChan/syncx"
)
var canUseIPv6 = syncx.Lazy[bool]{Init: func() bool {
ctx, cancel := context.WithTimeout(context.Background(), 4*time.Second)
defer cancel()
req, err := http.NewRequestWithContext(ctx, "GET", "http://v6.ipv6-test.com/json/widgetdata.php?callback=?", nil)
if err != nil {
return false
}
resp, err := http.DefaultClient.Do(req)
if err != nil {
return false
}
_ = resp.Body.Close()
return true
}}

72
main.go Normal file
View File

@ -0,0 +1,72 @@
package main
import (
"flag"
"fmt"
"os"
"runtime"
"time"
"github.com/sirupsen/logrus"
"gopkg.in/yaml.v3"
_ "rvcmd/console"
)
//go:generate ./pckcfg.sh assets packs tools
var usetrs = true
const ua = "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/123.0.0.0 Safari/537.36 Edg/123.0.0.0"
func main() {
logrus.Infoln("RVC Models Downloader start at", time.Now().Local().Format(time.DateTime+" (MST)"))
logrus.Infof("operating system: %s, architecture: %s\n", runtime.GOOS, runtime.GOARCH)
logrus.Infoln("can use ipv6:", canUseIPv6.Get())
ntrs := flag.Bool("notrs", false, "use standard TLS client")
dnsf := flag.String("dns", "", "custom dns.yaml")
flag.Parse()
args := flag.Args()
if len(args) != 1 {
fmt.Println("Usage:", os.Args[0], "[-notrs] 'target/to/download'")
flag.PrintDefaults()
fmt.Println(" 'target/to/download'\n like packs/general/latest")
fmt.Println("All available targets:")
fmt.Println(cmdlst.String())
return
}
if *ntrs {
usetrs = false
}
if *dnsf != "" {
f, err := os.Open(*dnsf)
if err != nil {
logrus.Errorln("open custom dns file", *dnsf, "err:", err)
return
}
m := map[string][]string{}
err = yaml.NewDecoder(f).Decode(&m)
if err != nil {
logrus.Errorln("decode custom dns file", *dnsf, "err:", err)
return
}
_ = f.Close()
if canUseIPv6.Get() {
dotv6servers.add(m)
} else {
dotv4servers.add(m)
}
fmt.Println("custom dns file added")
}
usercfg, err := readconfig(args[0])
if err != nil {
logrus.Errorln(err)
return
}
err = usercfg.download(args[0])
if err != nil {
logrus.Errorln(err)
return
}
logrus.Info("download tasks finished")
}

3
packs/amd/latest.yaml Normal file
View File

@ -0,0 +1,3 @@
BaseURL: https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main
Targets:
- Refer: v2/20231006

View File

@ -0,0 +1,7 @@
BaseURL: https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main
Targets:
- Folder: .
Copy:
- RVC0813AMD_Intel.7z
OS: windows
Arch: amd64

View File

@ -0,0 +1,7 @@
BaseURL: https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main
Targets:
- Folder: .
Copy:
- RVC1006AMD_Intel.7z
OS: windows
Arch: amd64

View File

@ -0,0 +1,7 @@
BaseURL: https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main
Targets:
- Folder: .
Copy:
- RVC-beta.7z
OS: windows
Arch: amd64

View File

@ -0,0 +1,7 @@
BaseURL: https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main
Targets:
- Folder: .
Copy:
- RVC-beta-v20230331.7z
OS: windows
Arch: amd64

View File

@ -0,0 +1,7 @@
BaseURL: https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main
Targets:
- Folder: .
Copy:
- RVC-beta20230416.7z
OS: windows
Arch: amd64

View File

@ -0,0 +1,7 @@
BaseURL: https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main
Targets:
- Folder: .
Copy:
- RVC-beta20230428.7z
OS: windows
Arch: amd64

View File

@ -0,0 +1,7 @@
BaseURL: https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main
Targets:
- Folder: .
Copy:
- RVC-beta20230508.7z
OS: windows
Arch: amd64

View File

@ -0,0 +1,7 @@
BaseURL: https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main
Targets:
- Folder: .
Copy:
- RVC-beta-20230513.7z
OS: windows
Arch: amd64

View File

@ -0,0 +1,7 @@
BaseURL: https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main
Targets:
- Folder: .
Copy:
- RVC-beta20230516.7z
OS: windows
Arch: amd64

View File

@ -0,0 +1,7 @@
BaseURL: https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main
Targets:
- Folder: .
Copy:
- RVC-beta0717.7z
OS: windows
Arch: amd64

View File

@ -0,0 +1,7 @@
BaseURL: https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main
Targets:
- Folder: .
Copy:
- RVC-beta-v2-0528.7z
OS: windows
Arch: amd64

View File

@ -0,0 +1,7 @@
BaseURL: https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main
Targets:
- Folder: .
Copy:
- RVC-beta-v2-0618.7z
OS: windows
Arch: amd64

3
packs/intel/latest.yaml Normal file
View File

@ -0,0 +1,3 @@
BaseURL: https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main
Targets:
- Refer: v2/20231006

View File

@ -0,0 +1,7 @@
BaseURL: https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main
Targets:
- Folder: .
Copy:
- RVC0813AMD_Intel.7z
OS: windows
Arch: amd64

View File

@ -0,0 +1,7 @@
BaseURL: https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main
Targets:
- Folder: .
Copy:
- RVC1006AMD_Intel.7z
OS: windows
Arch: amd64

3
packs/nvidia/latest.yaml Normal file
View File

@ -0,0 +1,3 @@
BaseURL: https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main
Targets:
- Refer: v2/20231006

View File

@ -0,0 +1,7 @@
BaseURL: https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main
Targets:
- Folder: .
Copy:
- RVC0813Nvidia.7z
OS: windows
Arch: amd64

View File

@ -0,0 +1,7 @@
BaseURL: https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main
Targets:
- Folder: .
Copy:
- RVC1006Nvidia.7z
OS: windows
Arch: amd64

7
pckcfg.sh Executable file
View File

@ -0,0 +1,7 @@
#!/bin/sh
# Should be run by go generate. DO NOT run it directly.
files=($(find $@ | grep -v .DS_Store | sort))
rm -rf cfg.zip
zip -9 cfg.zip "${files[@]}"

8
tools/ffmpeg.yaml Normal file
View File

@ -0,0 +1,8 @@
BaseURL: https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main
Targets:
- Folder: .
Copy:
- ffmpeg.exe
- ffprobe.exe
OS: windows
Arch: amd64