Unbewohnte
4 years ago
committed by
GitHub
2 changed files with 0 additions and 195 deletions
@ -1,82 +0,0 @@ |
|||||||
package manager |
|
||||||
|
|
||||||
import ( |
|
||||||
"errors" |
|
||||||
"fmt" |
|
||||||
"image" |
|
||||||
"image/color" |
|
||||||
"image/png" |
|
||||||
"io" |
|
||||||
"os" |
|
||||||
"strings" |
|
||||||
) |
|
||||||
|
|
||||||
// creates a complete black image file
|
|
||||||
func CreateBlackBG(width, height int) error { |
|
||||||
bgfile, err := os.Create("blackBG.png") |
|
||||||
if err != nil { |
|
||||||
return errors.New(fmt.Sprintf("Could not create black background file : %s", err)) |
|
||||||
} |
|
||||||
image := image.NewRGBA(image.Rect(0, 0, width, height)) |
|
||||||
bounds := image.Bounds() |
|
||||||
|
|
||||||
for y := 0; y < bounds.Max.Y; y++ { |
|
||||||
for x := 0; x < bounds.Max.X; x++ { |
|
||||||
image.Set(x, y, color.Black) |
|
||||||
} |
|
||||||
} |
|
||||||
err = png.Encode(bgfile, image) |
|
||||||
if err != nil { |
|
||||||
return errors.New(fmt.Sprintf("Could not encode an image : %s", err)) |
|
||||||
} |
|
||||||
err = bgfile.Close() |
|
||||||
if err != nil { |
|
||||||
return errors.New(fmt.Sprintf("Could not close the background file : %s", err)) |
|
||||||
} |
|
||||||
|
|
||||||
return nil |
|
||||||
} |
|
||||||
|
|
||||||
// checks if given string contains ".osu" file extention (NOT EXPORTED !)
|
|
||||||
func isBeatmap(filename string) bool { |
|
||||||
if len(filename) < 5 { |
|
||||||
return false |
|
||||||
} |
|
||||||
if filename[len(filename)-4:] == ".osu" { |
|
||||||
return true |
|
||||||
} |
|
||||||
return false |
|
||||||
} |
|
||||||
|
|
||||||
// checks if given string contains the image file extention (NOT EXPORTED !)
|
|
||||||
func isImage(filename string) bool { |
|
||||||
var imageExtentions []string = []string{"jpeg", "jpg", "png", "JPEG", "JPG", "PNG"} |
|
||||||
for _, extention := range imageExtentions { |
|
||||||
if strings.Contains(filename, extention) { |
|
||||||
return true |
|
||||||
} |
|
||||||
} |
|
||||||
return false |
|
||||||
} |
|
||||||
|
|
||||||
// opens given files, copies one into another (NOT EXPORTED !)
|
|
||||||
func copyFile(src, dst string) error { |
|
||||||
srcFile, err := os.Open(src) |
|
||||||
if err != nil { |
|
||||||
return errors.New(fmt.Sprintf("Could not open src file : %s", err)) |
|
||||||
} |
|
||||||
defer srcFile.Close() |
|
||||||
|
|
||||||
dstFile, err := os.OpenFile(dst, os.O_WRONLY, os.ModePerm) |
|
||||||
if err != nil { |
|
||||||
return errors.New(fmt.Sprintf("Could not open dst file : %s", err)) |
|
||||||
} |
|
||||||
defer dstFile.Close() |
|
||||||
|
|
||||||
_, err = io.Copy(dstFile, srcFile) |
|
||||||
if err != nil { |
|
||||||
return errors.New(fmt.Sprintf("Could not copy file : %s", err)) |
|
||||||
} |
|
||||||
|
|
||||||
return nil |
|
||||||
} |
|
@ -1,113 +0,0 @@ |
|||||||
package manager |
|
||||||
|
|
||||||
import ( |
|
||||||
"errors" |
|
||||||
"fmt" |
|
||||||
"os" |
|
||||||
"path/filepath" |
|
||||||
"strings" |
|
||||||
|
|
||||||
"github.com/Unbewohnte/OBM/logger" |
|
||||||
) |
|
||||||
|
|
||||||
// filepath.Joins the main osu directory with its songs folder
|
|
||||||
func GetSongsDir(osudir string) (string, error) { |
|
||||||
songsDir := filepath.Join(osudir, "Songs") |
|
||||||
|
|
||||||
stat, err := os.Stat(songsDir) |
|
||||||
if err != nil { |
|
||||||
return "", errors.New(fmt.Sprintf("Could not read the given path : %s", err)) |
|
||||||
} |
|
||||||
if !stat.IsDir() { |
|
||||||
return "", errors.New("Given Osu! directory is not a directory !") |
|
||||||
} |
|
||||||
|
|
||||||
return songsDir, nil |
|
||||||
} |
|
||||||
|
|
||||||
// parses given .osu file and returns the filename of its background
|
|
||||||
func GetBackgroundName(pathToOSUbeatmap string) (string, error) { |
|
||||||
beatmapBytes, err := os.ReadFile(pathToOSUbeatmap) |
|
||||||
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 isImage(chunk) { |
|
||||||
return strings.Split(chunk, "\"")[1], nil |
|
||||||
} |
|
||||||
} |
|
||||||
return "", nil |
|
||||||
} |
|
||||||
|
|
||||||
// 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(false, fmt.Sprintf("Wrong path : %s", err)) |
|
||||||
} |
|
||||||
for _, file := range files { |
|
||||||
filename := file.Name() |
|
||||||
|
|
||||||
if isBeatmap(filename) { |
|
||||||
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 |
|
||||||
} |
|
||||||
|
|
||||||
backgroundPath := filepath.Join(beatmapFolder, beatmapBackgroundFilename) |
|
||||||
|
|
||||||
// remove old background
|
|
||||||
err = os.Remove(backgroundPath) |
|
||||||
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(backgroundPath) |
|
||||||
if err != nil { |
|
||||||
failed++ |
|
||||||
logger.LogWarning(fmt.Sprintf("BEATMAP: %s: Could not create new background file : %s", beatmap, err)) |
|
||||||
continue |
|
||||||
} |
|
||||||
defer bgFile.Close() |
|
||||||
|
|
||||||
// copy the contents of a given image to the newly created bg file
|
|
||||||
err = copyFile(replacementPicPath, backgroundPath) |
|
||||||
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