Unbewohnte
4 years ago
committed by
GitHub
4 changed files with 0 additions and 227 deletions
@ -1,38 +0,0 @@
|
||||
package manager |
||||
|
||||
import ( |
||||
"errors" |
||||
"os" |
||||
"strings" |
||||
|
||||
"github.com/Unbewohnte/OBM/util" |
||||
) |
||||
|
||||
// parses given .osu file and returns the filename of its background
|
||||
func GetBackgroundName(pathTobeatmap string) (string, error) { |
||||
beatmapBytes, err := os.ReadFile(pathTobeatmap) |
||||
if err != nil { |
||||
return "", err |
||||
} |
||||
beatmapContents := string(beatmapBytes) |
||||
|
||||
// get index of "[Events]" (this is where BG filename is stored)
|
||||
eventsIndex := strings.Index(beatmapContents, "[Events]") |
||||
if eventsIndex == -1 { |
||||
return "", errors.New("Could not retrieve index of \"[Events]\"") |
||||
} |
||||
// get index of [TimingPoints] (this tag is right after the previous "[Events]" tag,
|
||||
// so we can grab the whole "[Events]" tag contents)
|
||||
timingPointsIndex := strings.Index(beatmapContents, "[TimingPoints]") |
||||
if timingPointsIndex == -1 { |
||||
return "", errors.New("Could not retrieve index of \"[TimingPoints]\"") |
||||
} |
||||
contentBetween := strings.Split(beatmapContents[eventsIndex:timingPointsIndex], ",") |
||||
|
||||
for _, chunk := range contentBetween { |
||||
if util.IsImage(chunk) { |
||||
return strings.Split(chunk, "\"")[1], nil |
||||
} |
||||
} |
||||
return "", nil |
||||
} |
||||
@ -1,45 +0,0 @@
|
||||
package manager |
||||
|
||||
import ( |
||||
"errors" |
||||
"fmt" |
||||
"os" |
||||
"path/filepath" |
||||
) |
||||
|
||||
// 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 |
||||
} |
||||
|
||||
// returns an array of full filepaths to each beatmap from given base Osu! directory
|
||||
func GetBeatmapFolderPaths(baseOsuDir string) ([]string, 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 beatmapFolderPaths []string |
||||
for _, file := range contents { |
||||
if file.IsDir() { |
||||
path := filepath.Join(songsDir, file.Name()) |
||||
beatmapFolderPaths = append(beatmapFolderPaths, path) |
||||
} |
||||
} |
||||
|
||||
return beatmapFolderPaths, nil |
||||
} |
||||
@ -1,71 +0,0 @@
|
||||
package manager |
||||
|
||||
import ( |
||||
"fmt" |
||||
"os" |
||||
"path/filepath" |
||||
|
||||
"github.com/Unbewohnte/OBM/logger" |
||||
"github.com/Unbewohnte/OBM/util" |
||||
) |
||||
|
||||
// reads contents of given dir; searches for .osu files; parses them for background info;
|
||||
// removes original background and replaces it with copied version of given image
|
||||
func ReplaceBackgrounds(beatmapFolder, replacementPicPath string) (successful, failed uint64) { |
||||
files, err := os.ReadDir(beatmapFolder) |
||||
if err != nil { |
||||
logger.LogError(true, fmt.Sprintf("Could not read directory : %s", err)) |
||||
} |
||||
for _, file := range files { |
||||
filename := file.Name() |
||||
|
||||
// if not a beatmap - skip
|
||||
if !util.IsBeatmap(filename) { |
||||
continue |
||||
} |
||||
|
||||
beatmap := filename |
||||
|
||||
// getting BG filename
|
||||
beatmapBackgroundFilename, err := GetBackgroundName(filepath.Join(beatmapFolder, beatmap)) |
||||
if err != nil { |
||||
logger.LogWarning(fmt.Sprintf("BEATMAP: %s: Error getting background filename: %s", beatmap, err)) |
||||
failed++ |
||||
continue |
||||
} |
||||
if beatmapBackgroundFilename == "" { |
||||
logger.LogWarning(fmt.Sprintf("BEATMAP: %s Could not find background filename in this beatmap file", beatmap)) |
||||
failed++ |
||||
continue |
||||
} |
||||
|
||||
pathToBackground := filepath.Join(beatmapFolder, beatmapBackgroundFilename) |
||||
|
||||
// remove old background
|
||||
err = os.Remove(pathToBackground) |
||||
if err != nil { |
||||
failed++ |
||||
logger.LogWarning(fmt.Sprintf("BEATMAP: %s: Could not remove old background : %s", beatmap, err)) |
||||
} |
||||
|
||||
// create new background file
|
||||
bgFile, err := os.Create(pathToBackground) |
||||
if err != nil { |
||||
failed++ |
||||
logger.LogWarning(fmt.Sprintf("BEATMAP: %s: Could not create new background file : %s", beatmap, err)) |
||||
continue |
||||
} |
||||
bgFile.Close() |
||||
|
||||
// copy the contents of a given image to the newly created bg file
|
||||
err = util.CopyFile(replacementPicPath, pathToBackground) |
||||
if err != nil { |
||||
logger.LogWarning(fmt.Sprintf("BEATMAP: %s: Could not copy file: %s", beatmap, err)) |
||||
failed++ |
||||
continue |
||||
} |
||||
successful++ |
||||
} |
||||
|
||||
return successful, failed |
||||
} |
||||
@ -1,73 +0,0 @@
|
||||
package manager |
||||
|
||||
import ( |
||||
"fmt" |
||||
"os" |
||||
"path/filepath" |
||||
|
||||
"github.com/Unbewohnte/OBM/logger" |
||||
"github.com/Unbewohnte/OBM/util" |
||||
) |
||||
|
||||
// retrieves backgrounds from given beatmap folder (same as in `ReplaceBackgrounds`) and copies them to the retrievement path
|
||||
func RetrieveBackgrounds(beatmapFolder, retrievementPath string) (successful, failed uint64) { |
||||
files, err := os.ReadDir(beatmapFolder) |
||||
if err != nil { |
||||
logger.LogError(true, fmt.Sprintf("Could not read directory : %s", err)) |
||||
} |
||||
for _, file := range files { |
||||
filename := file.Name() |
||||
|
||||
// if not a beatmap - skip
|
||||
if !util.IsBeatmap(filename) { |
||||
continue |
||||
} |
||||
|
||||
beatmap := filename |
||||
|
||||
// getting BG filename
|
||||
beatmapBackgroundFilename, err := GetBackgroundName(filepath.Join(beatmapFolder, beatmap)) |
||||
if err != nil { |
||||
logger.LogWarning(fmt.Sprintf("BEATMAP: %s: Error getting background filename: %s", beatmap, err)) |
||||
failed++ |
||||
continue |
||||
} |
||||
if beatmapBackgroundFilename == "" { |
||||
logger.LogWarning(fmt.Sprintf("BEATMAP: %s Could not find background filename in this beatmap file", beatmap)) |
||||
failed++ |
||||
continue |
||||
} |
||||
|
||||
pathToBackground := filepath.Join(beatmapFolder, beatmapBackgroundFilename) |
||||
|
||||
// creating a directory with the name of current beatmap folder in the retrievement path
|
||||
dstPath := filepath.Join(retrievementPath, filepath.Base(beatmapFolder)) |
||||
|
||||
err = os.MkdirAll(dstPath, os.ModePerm) |
||||
if err != nil { |
||||
logger.LogWarning(fmt.Sprintf("BEATMAP: %s: Could not create a directory (%s) for copying", beatmap, dstPath)) |
||||
continue |
||||
} |
||||
|
||||
// creating a copy file
|
||||
fullPathToCopy := filepath.Join(dstPath, beatmapBackgroundFilename) |
||||
dstFile, err := os.Create(fullPathToCopy) |
||||
if err != nil { |
||||
logger.LogWarning(fmt.Sprintf("BEATMAP: %s: Could not create a copy file", beatmap)) |
||||
failed++ |
||||
continue |
||||
} |
||||
dstFile.Close() |
||||
|
||||
// copy the background file to the retrievement path
|
||||
err = util.CopyFile(pathToBackground, fullPathToCopy) |
||||
if err != nil { |
||||
logger.LogWarning(fmt.Sprintf("BEATMAP: %s: Could not copy file: %s", beatmap, err)) |
||||
failed++ |
||||
continue |
||||
} |
||||
successful++ |
||||
} |
||||
|
||||
return successful, failed |
||||
} |
||||
Loading…
Reference in new issue