You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
83 lines
1.9 KiB
83 lines
1.9 KiB
3 years ago
|
package manager
|
||
|
|
||
|
import (
|
||
|
"errors"
|
||
|
"fmt"
|
||
|
"os"
|
||
|
"path/filepath"
|
||
|
|
||
|
"github.com/Unbewohnte/OBM/util"
|
||
|
)
|
||
|
|
||
|
// filepath.Joins the main osu directory with its songs folder
|
||
|
func getSongsDir(baseOsuDir string) (string, error) {
|
||
|
songsDir := filepath.Join(baseOsuDir, "Songs")
|
||
|
|
||
|
stat, err := os.Stat(songsDir)
|
||
|
if err != nil {
|
||
|
return "", errors.New(fmt.Sprintf("Could not process the given path : %s", err))
|
||
|
}
|
||
|
if !stat.IsDir() {
|
||
|
return "", errors.New("Given Osu! directory is not a directory !")
|
||
|
}
|
||
|
|
||
|
return songsDir, nil
|
||
|
}
|
||
|
|
||
|
// checks for .osu files in given path and returns all found instances
|
||
|
func getDiffs(path string) ([]string, error) {
|
||
|
files, err := os.ReadDir(path)
|
||
|
if err != nil {
|
||
|
return nil, errors.New(fmt.Sprintf("Could not read a directory : %s", err))
|
||
|
}
|
||
|
|
||
|
var diffs []string
|
||
|
for _, file := range files {
|
||
|
filename := file.Name()
|
||
|
if util.IsBeatmap(filename) {
|
||
|
diffs = append(diffs, filename)
|
||
|
}
|
||
|
}
|
||
|
return diffs, nil
|
||
|
}
|
||
|
|
||
|
// constructs a `Beatmap` struct and returns it
|
||
|
func newBeatmap(name, path string, diffs []string) Beatmap {
|
||
|
return Beatmap{
|
||
|
Name: name,
|
||
|
Path: path,
|
||
|
Diffs: diffs,
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// returns an array of beatmaps from given base Osu! directory
|
||
|
func GetBeatmaps(baseOsuDir string) ([]Beatmap, error) {
|
||
|
songsDir, err := getSongsDir(baseOsuDir)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
contents, err := os.ReadDir(songsDir)
|
||
|
if err != nil {
|
||
|
return nil, errors.New(fmt.Sprintf("Could not read a directory : %s", err))
|
||
|
}
|
||
|
|
||
|
var beatmaps []Beatmap
|
||
|
// looping through all folders in yourOsuDir/Songs/ directory
|
||
|
for _, file := range contents {
|
||
|
if file.IsDir() {
|
||
|
// retrieving all necessary data for creating a new instance of a beatmap
|
||
|
beatmapName := file.Name()
|
||
|
pathToBeatmap := filepath.Join(songsDir, beatmapName)
|
||
|
diffs, err := getDiffs(pathToBeatmap)
|
||
|
if err != nil {
|
||
|
continue
|
||
|
}
|
||
|
newBeatmap := newBeatmap(beatmapName, pathToBeatmap, diffs)
|
||
|
|
||
|
beatmaps = append(beatmaps, newBeatmap)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return beatmaps, nil
|
||
|
}
|