latest
This commit is contained in:
parent
39eefcafee
commit
0dd9505fb3
385
fastmirror.go
385
fastmirror.go
|
|
@ -1,385 +0,0 @@
|
||||||
package fastmirror
|
|
||||||
|
|
||||||
import (
|
|
||||||
"bufio"
|
|
||||||
_ "embed"
|
|
||||||
"fmt"
|
|
||||||
"github.com/schollz/progressbar/v3"
|
|
||||||
"html/template"
|
|
||||||
"io"
|
|
||||||
"log"
|
|
||||||
"net/http"
|
|
||||||
"os"
|
|
||||||
"os/exec"
|
|
||||||
"path/filepath"
|
|
||||||
"runtime"
|
|
||||||
"strconv"
|
|
||||||
"strings"
|
|
||||||
"time"
|
|
||||||
)
|
|
||||||
|
|
||||||
//go:embed sources_list_legacy.tmpl
|
|
||||||
var legacyTemplate string
|
|
||||||
|
|
||||||
//go:embed sources_list_modern.tmpl
|
|
||||||
var modernTemplate string
|
|
||||||
|
|
||||||
func CLIEntry() {
|
|
||||||
log.Println("Starting Fastmirror CLI")
|
|
||||||
|
|
||||||
if runtime.GOOS != "linux" {
|
|
||||||
log.Fatal("This program is only for Linux")
|
|
||||||
}
|
|
||||||
if err := isUbuntu(); err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
sourcesFilePath, err := findSourcesFilePath()
|
|
||||||
if err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
log.Printf("Found sources file: %s", sourcesFilePath)
|
|
||||||
|
|
||||||
// Extract suites from the existing sources file
|
|
||||||
suites, err := extractSuites(sourcesFilePath)
|
|
||||||
if err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
log.Printf("Extracted suites: %v", suites)
|
|
||||||
|
|
||||||
// Create backup of the sources file
|
|
||||||
if err := createBackup(sourcesFilePath); err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Create sources.list.d directory if it doesn't exist
|
|
||||||
if err := createSourcesListD(); err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Fetch and find the fastest mirror
|
|
||||||
fastestMirrorURL, fastestTime, latencies, err := findFastestMirror()
|
|
||||||
if err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
log.Printf("Fastest mirror: %s", fastestMirrorURL)
|
|
||||||
|
|
||||||
// Display latency statistics
|
|
||||||
displayLatencyStatistics(latencies, fastestMirrorURL, fastestTime)
|
|
||||||
|
|
||||||
// Update sources file with the fastest mirror
|
|
||||||
if err := updateSourcesFile(sourcesFilePath, fastestMirrorURL, suites); err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
log.Println("Fastmirror CLI finished successfully")
|
|
||||||
}
|
|
||||||
|
|
||||||
func getUbuntuCodename() (string, error) {
|
|
||||||
cmd := exec.Command("lsb_release", "-cs")
|
|
||||||
output, err := cmd.Output()
|
|
||||||
if err != nil {
|
|
||||||
return "", fmt.Errorf("failed to run lsb_release: %v", err)
|
|
||||||
}
|
|
||||||
return strings.TrimSpace(string(output)), nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func getUbuntuVersion() (int, int, error) {
|
|
||||||
cmd := exec.Command("lsb_release", "-rs")
|
|
||||||
output, err := cmd.Output()
|
|
||||||
if err != nil {
|
|
||||||
return 0, 0, fmt.Errorf("failed to run lsb_release: %v", err)
|
|
||||||
}
|
|
||||||
version := strings.TrimSpace(string(output))
|
|
||||||
parts := strings.Split(version, ".")
|
|
||||||
if len(parts) != 2 {
|
|
||||||
return 0, 0, fmt.Errorf("unexpected version format: %s", version)
|
|
||||||
}
|
|
||||||
majorVersion, err := strconv.Atoi(parts[0])
|
|
||||||
if err != nil {
|
|
||||||
return 0, 0, fmt.Errorf("failed to parse major version: %v", err)
|
|
||||||
}
|
|
||||||
minorVersion, err := strconv.Atoi(parts[1])
|
|
||||||
if err != nil {
|
|
||||||
return 0, 0, fmt.Errorf("failed to parse minor version: %v", err)
|
|
||||||
}
|
|
||||||
return majorVersion, minorVersion, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func isUbuntu() error {
|
|
||||||
cmd := exec.Command("lsb_release", "-is")
|
|
||||||
output, err := cmd.Output()
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("failed to run lsb_release: %v", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if strings.TrimSpace(string(output)) != "Ubuntu" {
|
|
||||||
return fmt.Errorf("this program is only for Ubuntu")
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func findSourcesFilePath() (string, error) {
|
|
||||||
const sourcesListPath = "/etc/apt/sources.list"
|
|
||||||
const sourcesListDPath = "/etc/apt/sources.list.d/"
|
|
||||||
if isUbuntuSourcesFile(sourcesListPath) {
|
|
||||||
log.Printf("Found Ubuntu sources file: %s", sourcesListPath)
|
|
||||||
return sourcesListPath, nil
|
|
||||||
}
|
|
||||||
files, err := os.ReadDir(sourcesListDPath)
|
|
||||||
if err != nil {
|
|
||||||
return "", fmt.Errorf("failed to read directory %s: %v", sourcesListDPath, err)
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, file := range files {
|
|
||||||
if file.IsDir() {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
filePath := filepath.Join(sourcesListDPath, file.Name())
|
|
||||||
if isUbuntuSourcesFile(filePath) {
|
|
||||||
log.Printf("Found Ubuntu sources file: %s", filePath)
|
|
||||||
return filePath, nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return "", fmt.Errorf("no Ubuntu sources file found")
|
|
||||||
}
|
|
||||||
|
|
||||||
func isUbuntuSourcesFile(filePath string) bool {
|
|
||||||
content, err := os.ReadFile(filePath)
|
|
||||||
if err != nil {
|
|
||||||
log.Printf("Failed to read file %s: %v", filePath, err)
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
return strings.Contains(strings.ToLower(string(content)), "ubuntu.com")
|
|
||||||
}
|
|
||||||
|
|
||||||
func createBackup(filePath string) error {
|
|
||||||
backupPath := filePath + ".bak"
|
|
||||||
if _, err := os.Stat(backupPath); err == nil {
|
|
||||||
return fmt.Errorf("backup file %s already exists", backupPath)
|
|
||||||
}
|
|
||||||
if err := os.Rename(filePath, backupPath); err != nil {
|
|
||||||
return fmt.Errorf("failed to create backup of %s: %v", filePath, err)
|
|
||||||
}
|
|
||||||
log.Printf("Created backup: %s", backupPath)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func createSourcesListD() error {
|
|
||||||
const sourcesListDPath = "/etc/apt/sources.list.d/"
|
|
||||||
if _, err := os.Stat(sourcesListDPath); os.IsNotExist(err) {
|
|
||||||
if err := os.Mkdir(sourcesListDPath, 0755); err != nil {
|
|
||||||
return fmt.Errorf("failed to create directory %s: %v", sourcesListDPath, err)
|
|
||||||
}
|
|
||||||
log.Printf("Created directory: %s", sourcesListDPath)
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func findFastestMirror() (string, time.Duration, []time.Duration, error) {
|
|
||||||
log.Println("Fetching mirrors list")
|
|
||||||
resp, err := http.Get("http://mirrors.ubuntu.com/mirrors.txt")
|
|
||||||
if err != nil {
|
|
||||||
return "", 0, nil, fmt.Errorf("failed to fetch mirrors list: %v", err)
|
|
||||||
}
|
|
||||||
defer resp.Body.Close()
|
|
||||||
|
|
||||||
body, err := io.ReadAll(resp.Body)
|
|
||||||
if err != nil {
|
|
||||||
return "", 0, nil, fmt.Errorf("failed to read mirrors list: %v", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
mirrors := strings.Split(string(body), "\n")
|
|
||||||
log.Printf("Found %d mirrors", len(mirrors))
|
|
||||||
|
|
||||||
var fastestMirrorURL string
|
|
||||||
var fastestTime time.Duration
|
|
||||||
var latencies []time.Duration
|
|
||||||
downCount := 0
|
|
||||||
noIndexCount := 0
|
|
||||||
|
|
||||||
codename, err := getUbuntuCodename()
|
|
||||||
if err != nil {
|
|
||||||
return "", 0, nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
httpClient := http.Client{
|
|
||||||
Timeout: 1 * time.Second,
|
|
||||||
}
|
|
||||||
|
|
||||||
log.Println("Testing mirrors for latency")
|
|
||||||
bar := progressbar.Default(int64(len(mirrors)))
|
|
||||||
|
|
||||||
for _, mirror := range mirrors {
|
|
||||||
if err := bar.Add(1); err != nil {
|
|
||||||
log.Printf("Error updating progress bar: %v", err)
|
|
||||||
}
|
|
||||||
if strings.HasPrefix(mirror, "https://") {
|
|
||||||
mirror = strings.TrimSuffix(mirror, "/")
|
|
||||||
startTime := time.Now()
|
|
||||||
isValid := isValidMirror(httpClient, mirror, codename)
|
|
||||||
elapsedTime := time.Since(startTime)
|
|
||||||
|
|
||||||
if isValid {
|
|
||||||
latencies = append(latencies, elapsedTime)
|
|
||||||
if fastestMirrorURL == "" || elapsedTime < fastestTime {
|
|
||||||
fastestMirrorURL = mirror
|
|
||||||
fastestTime = elapsedTime
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
noIndexCount++
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
downCount++
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if fastestMirrorURL == "" {
|
|
||||||
return "", 0, nil, fmt.Errorf("no suitable HTTPS mirror found")
|
|
||||||
}
|
|
||||||
|
|
||||||
displayLatencyStatistics(latencies, fastestMirrorURL, fastestTime)
|
|
||||||
log.Printf("Number of mirrors that were down: %d", downCount)
|
|
||||||
log.Printf("Number of mirrors without required package index: %d", noIndexCount)
|
|
||||||
|
|
||||||
return fastestMirrorURL, fastestTime, latencies, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func isValidMirror(httpClient http.Client, mirrorURL, codename string) bool {
|
|
||||||
uri := fmt.Sprintf("%s/dists/%s/Release", mirrorURL, codename)
|
|
||||||
resp, err := httpClient.Get(uri)
|
|
||||||
if err != nil {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
defer resp.Body.Close()
|
|
||||||
return resp.StatusCode == http.StatusOK
|
|
||||||
}
|
|
||||||
|
|
||||||
func extractSuites(filePath string) ([]string, error) {
|
|
||||||
content, err := os.ReadFile(filePath)
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("failed to read file %s: %v", filePath, err)
|
|
||||||
}
|
|
||||||
|
|
||||||
var suites []string
|
|
||||||
scanner := bufio.NewScanner(strings.NewReader(string(content)))
|
|
||||||
|
|
||||||
isModernFormat := false
|
|
||||||
for scanner.Scan() {
|
|
||||||
line := scanner.Text()
|
|
||||||
if strings.HasPrefix(line, "Types:") {
|
|
||||||
isModernFormat = true
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
scanner = bufio.NewScanner(strings.NewReader(string(content)))
|
|
||||||
if isModernFormat {
|
|
||||||
for scanner.Scan() {
|
|
||||||
line := scanner.Text()
|
|
||||||
if strings.HasPrefix(line, "Suites:") {
|
|
||||||
parts := strings.Fields(line)
|
|
||||||
suites = append(suites, parts[1:]...)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
for scanner.Scan() {
|
|
||||||
line := scanner.Text()
|
|
||||||
if strings.HasPrefix(line, "deb ") || strings.HasPrefix(line, "deb-src ") {
|
|
||||||
parts := strings.Fields(line)
|
|
||||||
if len(parts) >= 4 {
|
|
||||||
suites = append(suites, parts[3])
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if err := scanner.Err(); err != nil {
|
|
||||||
return nil, fmt.Errorf("failed to scan file %s: %v", filePath, err)
|
|
||||||
}
|
|
||||||
return suites, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func displayLatencyStatistics(latencies []time.Duration, fastestMirrorURL string, fastestTime time.Duration) {
|
|
||||||
if len(latencies) == 0 {
|
|
||||||
log.Println("No latencies recorded")
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
var totalLatency time.Duration
|
|
||||||
var minLatency time.Duration = latencies[0]
|
|
||||||
var maxLatency time.Duration = latencies[0]
|
|
||||||
|
|
||||||
for _, latency := range latencies {
|
|
||||||
totalLatency += latency
|
|
||||||
if latency < minLatency {
|
|
||||||
minLatency = latency
|
|
||||||
}
|
|
||||||
if latency > maxLatency {
|
|
||||||
maxLatency = latency
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
averageLatency := totalLatency / time.Duration(len(latencies))
|
|
||||||
|
|
||||||
log.Printf("Minimum latency: %s", minLatency)
|
|
||||||
log.Printf("Maximum latency: %s", maxLatency)
|
|
||||||
log.Printf("Average latency: %s", averageLatency)
|
|
||||||
log.Printf("Chosen fastest mirror: %s with latency: %s", fastestMirrorURL, fastestTime)
|
|
||||||
}
|
|
||||||
|
|
||||||
func updateSourcesFile(filePath, mirrorURL string, suites []string) error {
|
|
||||||
codename, err := getUbuntuCodename()
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
majorVersion, minorVersion, err := getUbuntuVersion()
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
suitesString := strings.Join(suites, " ")
|
|
||||||
|
|
||||||
var content string
|
|
||||||
data := struct {
|
|
||||||
MirrorURL string
|
|
||||||
Codename string
|
|
||||||
Suites string
|
|
||||||
}{
|
|
||||||
MirrorURL: mirrorURL,
|
|
||||||
Codename: codename,
|
|
||||||
Suites: suitesString,
|
|
||||||
}
|
|
||||||
|
|
||||||
if majorVersion > 24 || (majorVersion == 24 && minorVersion >= 4) {
|
|
||||||
tmpl, err := template.New("modern").Parse(modernTemplate)
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("failed to parse modern template: %v", err)
|
|
||||||
}
|
|
||||||
var buf strings.Builder
|
|
||||||
err = tmpl.Execute(&buf, data)
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("failed to execute modern template: %v", err)
|
|
||||||
}
|
|
||||||
content = buf.String()
|
|
||||||
} else {
|
|
||||||
tmpl, err := template.New("legacy").Parse(legacyTemplate)
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("failed to parse legacy template: %v", err)
|
|
||||||
}
|
|
||||||
var buf strings.Builder
|
|
||||||
err = tmpl.Execute(&buf, data)
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("failed to execute legacy template: %v", err)
|
|
||||||
}
|
|
||||||
content = buf.String()
|
|
||||||
}
|
|
||||||
|
|
||||||
err = os.WriteFile(filePath, []byte(content), 0644)
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("failed to update sources file %s: %v", filePath, err)
|
|
||||||
}
|
|
||||||
log.Printf("Updated sources file: %s", filePath)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
@ -0,0 +1,56 @@
|
||||||
|
package fastmirror
|
||||||
|
|
||||||
|
import (
|
||||||
|
"log"
|
||||||
|
"runtime"
|
||||||
|
)
|
||||||
|
|
||||||
|
func CLIEntry() {
|
||||||
|
log.Println("Starting Fastmirror CLI")
|
||||||
|
|
||||||
|
if runtime.GOOS != "linux" {
|
||||||
|
log.Fatal("This program is only for Linux")
|
||||||
|
}
|
||||||
|
if err := isUbuntu(); err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
sourcesFilePath, err := findSourcesFilePath()
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
log.Printf("Found sources file: %s", sourcesFilePath)
|
||||||
|
|
||||||
|
// Extract suites from the existing sources file
|
||||||
|
suites, err := extractSuites(sourcesFilePath)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
log.Printf("Extracted suites: %v", suites)
|
||||||
|
|
||||||
|
// Create backup of the sources file
|
||||||
|
if err := createBackup(sourcesFilePath); err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create sources.list.d directory if it doesn't exist
|
||||||
|
if err := createSourcesListD(); err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Fetch and find the fastest mirror
|
||||||
|
fastestMirrorURL, fastestTime, latencies, err := findFastestMirror(suites)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
log.Printf("Fastest mirror: %s", fastestMirrorURL)
|
||||||
|
|
||||||
|
// Display latency statistics
|
||||||
|
displayLatencyStatistics(latencies, fastestMirrorURL, fastestTime)
|
||||||
|
|
||||||
|
// Update sources file with the fastest mirror
|
||||||
|
if err := updateSourcesFile(sourcesFilePath, fastestMirrorURL, suites); err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
log.Println("Fastmirror CLI finished successfully")
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,121 @@
|
||||||
|
package fastmirror
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"log"
|
||||||
|
"net/http"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/schollz/progressbar/v3"
|
||||||
|
)
|
||||||
|
|
||||||
|
func findFastestMirror(suites []string) (string, time.Duration, []time.Duration, error) {
|
||||||
|
log.Println("Fetching mirrors list")
|
||||||
|
resp, err := http.Get("http://mirrors.ubuntu.com/mirrors.txt")
|
||||||
|
if err != nil {
|
||||||
|
return "", 0, nil, fmt.Errorf("failed to fetch mirrors list: %v", err)
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
body, err := io.ReadAll(resp.Body)
|
||||||
|
if err != nil {
|
||||||
|
return "", 0, nil, fmt.Errorf("failed to read mirrors list: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
mirrors := strings.Split(string(body), "\n")
|
||||||
|
log.Printf("Found %d mirrors", len(mirrors))
|
||||||
|
|
||||||
|
var fastestMirrorURL string
|
||||||
|
var fastestTime time.Duration
|
||||||
|
var latencies []time.Duration
|
||||||
|
downCount := 0
|
||||||
|
noIndexCount := 0
|
||||||
|
|
||||||
|
codename, err := getUbuntuCodename()
|
||||||
|
if err != nil {
|
||||||
|
return "", 0, nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
httpClient := http.Client{
|
||||||
|
Timeout: 1 * time.Second,
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Println("Testing mirrors for latency")
|
||||||
|
bar := progressbar.Default(int64(len(mirrors)))
|
||||||
|
|
||||||
|
for _, mirror := range mirrors {
|
||||||
|
if err := bar.Add(1); err != nil {
|
||||||
|
log.Printf("Error updating progress bar: %v", err)
|
||||||
|
}
|
||||||
|
if strings.HasPrefix(mirror, "https://") {
|
||||||
|
mirror = strings.TrimSuffix(mirror, "/")
|
||||||
|
startTime := time.Now()
|
||||||
|
isValid := isValidMirror(httpClient, mirror, codename, suites)
|
||||||
|
elapsedTime := time.Since(startTime)
|
||||||
|
|
||||||
|
if isValid {
|
||||||
|
latencies = append(latencies, elapsedTime)
|
||||||
|
if fastestMirrorURL == "" || elapsedTime < fastestTime {
|
||||||
|
fastestMirrorURL = mirror
|
||||||
|
fastestTime = elapsedTime
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
noIndexCount++
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
downCount++
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if fastestMirrorURL == "" {
|
||||||
|
return "", 0, nil, fmt.Errorf("no suitable HTTPS mirror found")
|
||||||
|
}
|
||||||
|
|
||||||
|
displayLatencyStatistics(latencies, fastestMirrorURL, fastestTime)
|
||||||
|
log.Printf("Number of mirrors that were down: %d", downCount)
|
||||||
|
log.Printf("Number of mirrors without required package index: %d", noIndexCount)
|
||||||
|
|
||||||
|
return fastestMirrorURL, fastestTime, latencies, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func isValidMirror(httpClient http.Client, mirrorURL, codename string, suites []string) bool {
|
||||||
|
for _, suite := range suites {
|
||||||
|
uri := fmt.Sprintf("%s/dists/%s/Release", mirrorURL, suite)
|
||||||
|
resp, err := httpClient.Get(uri)
|
||||||
|
if err != nil || resp.StatusCode != http.StatusOK {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
resp.Body.Close()
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
func displayLatencyStatistics(latencies []time.Duration, fastestMirrorURL string, fastestTime time.Duration) {
|
||||||
|
if len(latencies) == 0 {
|
||||||
|
log.Println("No latencies recorded")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
var totalLatency time.Duration
|
||||||
|
var minLatency time.Duration = latencies[0]
|
||||||
|
var maxLatency time.Duration = latencies[0]
|
||||||
|
|
||||||
|
for _, latency := range latencies {
|
||||||
|
totalLatency += latency
|
||||||
|
if latency < minLatency {
|
||||||
|
minLatency = latency
|
||||||
|
}
|
||||||
|
if latency > maxLatency {
|
||||||
|
maxLatency = latency
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
averageLatency := totalLatency / time.Duration(len(latencies))
|
||||||
|
|
||||||
|
log.Printf("Minimum latency: %s", minLatency)
|
||||||
|
log.Printf("Maximum latency: %s", maxLatency)
|
||||||
|
log.Printf("Average latency: %s", averageLatency)
|
||||||
|
log.Printf("Chosen fastest mirror: %s with latency: %s", fastestMirrorURL, fastestTime)
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,177 @@
|
||||||
|
package fastmirror
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bufio"
|
||||||
|
_ "embed"
|
||||||
|
"fmt"
|
||||||
|
"html/template"
|
||||||
|
"log"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
//go:embed sources_list_legacy.tmpl
|
||||||
|
var legacyTemplate string
|
||||||
|
|
||||||
|
//go:embed sources_list_modern.tmpl
|
||||||
|
var modernTemplate string
|
||||||
|
|
||||||
|
func findSourcesFilePath() (string, error) {
|
||||||
|
const sourcesListPath = "/etc/apt/sources.list"
|
||||||
|
const sourcesListDPath = "/etc/apt/sources.list.d/"
|
||||||
|
if isUbuntuSourcesFile(sourcesListPath) {
|
||||||
|
log.Printf("Found Ubuntu sources file: %s", sourcesListPath)
|
||||||
|
return sourcesListPath, nil
|
||||||
|
}
|
||||||
|
files, err := os.ReadDir(sourcesListDPath)
|
||||||
|
if err != nil {
|
||||||
|
return "", fmt.Errorf("failed to read directory %s: %v", sourcesListDPath, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, file := range files {
|
||||||
|
if file.IsDir() {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
filePath := filepath.Join(sourcesListDPath, file.Name())
|
||||||
|
if isUbuntuSourcesFile(filePath) {
|
||||||
|
log.Printf("Found Ubuntu sources file: %s", filePath)
|
||||||
|
return filePath, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return "", fmt.Errorf("no Ubuntu sources file found")
|
||||||
|
}
|
||||||
|
|
||||||
|
func isUbuntuSourcesFile(filePath string) bool {
|
||||||
|
content, err := os.ReadFile(filePath)
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("Failed to read file %s: %v", filePath, err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
return strings.Contains(strings.ToLower(string(content)), "ubuntu.com")
|
||||||
|
}
|
||||||
|
|
||||||
|
func createBackup(filePath string) error {
|
||||||
|
backupPath := filePath + ".bak"
|
||||||
|
if _, err := os.Stat(backupPath); err == nil {
|
||||||
|
return fmt.Errorf("backup file %s already exists", backupPath)
|
||||||
|
}
|
||||||
|
if err := os.Rename(filePath, backupPath); err != nil {
|
||||||
|
return fmt.Errorf("failed to create backup of %s: %v", filePath, err)
|
||||||
|
}
|
||||||
|
log.Printf("Created backup: %s", backupPath)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func createSourcesListD() error {
|
||||||
|
const sourcesListDPath = "/etc/apt/sources.list.d/"
|
||||||
|
if _, err := os.Stat(sourcesListDPath); os.IsNotExist(err) {
|
||||||
|
if err := os.Mkdir(sourcesListDPath, 0755); err != nil {
|
||||||
|
return fmt.Errorf("failed to create directory %s: %v", sourcesListDPath, err)
|
||||||
|
}
|
||||||
|
log.Printf("Created directory: %s", sourcesListDPath)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func extractSuites(filePath string) ([]string, error) {
|
||||||
|
content, err := os.ReadFile(filePath)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("failed to read file %s: %v", filePath, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var suites []string
|
||||||
|
scanner := bufio.NewScanner(strings.NewReader(string(content)))
|
||||||
|
|
||||||
|
isModernFormat := false
|
||||||
|
for scanner.Scan() {
|
||||||
|
line := scanner.Text()
|
||||||
|
if strings.HasPrefix(line, "Types:") {
|
||||||
|
isModernFormat = true
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
scanner = bufio.NewScanner(strings.NewReader(string(content)))
|
||||||
|
if isModernFormat {
|
||||||
|
for scanner.Scan() {
|
||||||
|
line := scanner.Text()
|
||||||
|
if strings.HasPrefix(line, "Suites:") {
|
||||||
|
parts := strings.Fields(line)
|
||||||
|
suites = append(suites, parts[1:]...)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
for scanner.Scan() {
|
||||||
|
line := scanner.Text()
|
||||||
|
if strings.HasPrefix(line, "deb ") || strings.HasPrefix(line, "deb-src ") {
|
||||||
|
parts := strings.Fields(line)
|
||||||
|
if len(parts) >= 4 {
|
||||||
|
suites = append(suites, parts[3])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := scanner.Err(); err != nil {
|
||||||
|
return nil, fmt.Errorf("failed to scan file %s: %v", filePath, err)
|
||||||
|
}
|
||||||
|
return suites, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func updateSourcesFile(filePath, mirrorURL string, suites []string) error {
|
||||||
|
codename, err := getUbuntuCodename()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
majorVersion, minorVersion, err := getUbuntuVersion()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
suitesString := strings.Join(suites, " ")
|
||||||
|
|
||||||
|
var content string
|
||||||
|
data := struct {
|
||||||
|
MirrorURL string
|
||||||
|
Codename string
|
||||||
|
Suites string
|
||||||
|
}{
|
||||||
|
MirrorURL: mirrorURL,
|
||||||
|
Codename: codename,
|
||||||
|
Suites: suitesString,
|
||||||
|
}
|
||||||
|
|
||||||
|
if majorVersion > 24 || (majorVersion == 24 && minorVersion >= 4) {
|
||||||
|
tmpl, err := template.New("modern").Parse(modernTemplate)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to parse modern template: %v", err)
|
||||||
|
}
|
||||||
|
var buf strings.Builder
|
||||||
|
err = tmpl.Execute(&buf, data)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to execute modern template: %v", err)
|
||||||
|
}
|
||||||
|
content = buf.String()
|
||||||
|
} else {
|
||||||
|
tmpl, err := template.New("legacy").Parse(legacyTemplate)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to parse legacy template: %v", err)
|
||||||
|
}
|
||||||
|
var buf strings.Builder
|
||||||
|
err = tmpl.Execute(&buf, data)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to execute legacy template: %v", err)
|
||||||
|
}
|
||||||
|
content = buf.String()
|
||||||
|
}
|
||||||
|
|
||||||
|
err = os.WriteFile(filePath, []byte(content), 0644)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to update sources file %s: %v", filePath, err)
|
||||||
|
}
|
||||||
|
log.Printf("Updated sources file: %s", filePath)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,52 @@
|
||||||
|
package fastmirror
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"os/exec"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
func getUbuntuCodename() (string, error) {
|
||||||
|
cmd := exec.Command("lsb_release", "-cs")
|
||||||
|
output, err := cmd.Output()
|
||||||
|
if err != nil {
|
||||||
|
return "", fmt.Errorf("failed to run lsb_release: %v", err)
|
||||||
|
}
|
||||||
|
return strings.TrimSpace(string(output)), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func getUbuntuVersion() (int, int, error) {
|
||||||
|
cmd := exec.Command("lsb_release", "-rs")
|
||||||
|
output, err := cmd.Output()
|
||||||
|
if err != nil {
|
||||||
|
return 0, 0, fmt.Errorf("failed to run lsb_release: %v", err)
|
||||||
|
}
|
||||||
|
version := strings.TrimSpace(string(output))
|
||||||
|
parts := strings.Split(version, ".")
|
||||||
|
if len(parts) != 2 {
|
||||||
|
return 0, 0, fmt.Errorf("unexpected version format: %s", version)
|
||||||
|
}
|
||||||
|
majorVersion, err := strconv.Atoi(parts[0])
|
||||||
|
if err != nil {
|
||||||
|
return 0, 0, fmt.Errorf("failed to parse major version: %v", err)
|
||||||
|
}
|
||||||
|
minorVersion, err := strconv.Atoi(parts[1])
|
||||||
|
if err != nil {
|
||||||
|
return 0, 0, fmt.Errorf("failed to parse minor version: %v", err)
|
||||||
|
}
|
||||||
|
return majorVersion, minorVersion, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func isUbuntu() error {
|
||||||
|
cmd := exec.Command("lsb_release", "-is")
|
||||||
|
output, err := cmd.Output()
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to run lsb_release: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if strings.TrimSpace(string(output)) != "Ubuntu" {
|
||||||
|
return fmt.Errorf("this program is only for Ubuntu")
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue