Update repo.go

Author: michal92299

src/repo.go

@@ -1,24 +1,20 @@
 package main
 
 import (
-	"bufio"
 	"fmt"
 	"io"
 	"net/http"
 	"os"
 	"path/filepath"
-	"strings"
+
+	"gopkg.in/yaml.v3"
 )
 
 // Repo structure: map[section]map[category]map[name]url
 type Repo map[string]map[string]map[string]string
 
-const (
-	repoURL = "https://raw.githubusercontent.com/Bytes-Repository/bytes.io/main/repository/bytes.io"
-)
-
-func refreshRepo(localPath string) error {
-	resp, err := http.Get(repoURL)
+func refreshRepo(url, localPath string) error {
+	resp, err := http.Get(url)
 	if err != nil {
 		return err
 	}
@@ -40,65 +36,41 @@ func refreshRepo(localPath string) error {
 }
 
 func parseRepo(path string) (Repo, error) {
-	f, err := os.Open(path)
+	data, err := os.ReadFile(path)
 	if err != nil {
 		return nil, err
 	}
-	defer f.Close()
+	var raw map[string]interface{}
+	if err := yaml.Unmarshal(data, &raw); err != nil {
+		return nil, err
+	}
 	repo := make(Repo)
-	var currentSection string
-	var currentCategory string
-	scanner := bufio.NewScanner(f)
-	for scanner.Scan() {
-		fullLine := scanner.Text()
-		line := strings.TrimSpace(fullLine)
-		if line == "" {
+	for section, secVal := range raw {
+		secMap, ok := secVal.(map[string]interface{})
+		if !ok {
 			continue
 		}
-		indentLevel := len(fullLine) - len(line)
-		if strings.HasPrefix(line, "=") && strings.Contains(line, "[") {
-			parts := strings.SplitN(line, " [", 2)
-			name := strings.TrimSpace(strings.TrimPrefix(strings.TrimSpace(parts[0]), "="))
-			if indentLevel == 0 {
-				// New section
-				repo[name] = make(map[string]map[string]string)
-				currentSection = name
-				currentCategory = ""
-			} else {
-				// New category
-				if currentSection != "" {
-					repo[currentSection][name] = make(map[string]string)
-					currentCategory = name
-				}
-			}
-		} else if strings.Contains(line, "=>") {
-			// Package
-			parts := strings.SplitN(line, "=>", 2)
-			name := strings.TrimSpace(parts[0])
-			url := ""
-			if len(parts) > 1 {
-				url = strings.TrimSpace(parts[1])
+		repo[section] = make(map[string]map[string]string)
+		for category, catVal := range secMap {
+			if catVal == nil {
+				repo[section][category] = make(map[string]string)
+				continue
 			}
-			if name == "" {
+			catMap, ok := catVal.(map[string]interface{})
+			if !ok {
 				continue
 			}
-			if currentSection != "" && currentCategory != "" {
-				repo[currentSection][currentCategory][name] = url
-			} else if currentSection != "" {
-				// Direct package in section
-				if _, ok := repo[currentSection][""]; !ok {
-					repo[currentSection][""] = make(map[string]string)
+			repo[section][category] = make(map[string]string)
+			for name, urlVal := range catMap {
+				url := ""
+				if urlVal != nil {
+					if s, ok := urlVal.(string); ok {
+						url = s
+					}
 				}
-				repo[currentSection][""][name] = url
-			}
-		} else if line == "]" {
-			// End of category or section
-			if indentLevel > 0 {
-				currentCategory = ""
-			} else {
-				currentSection = ""
+				repo[section][category][name] = url
 			}
 		}
 	}
-	return repo, scanner.Err()
+	return repo, nil
 }