mth/ebiten
1
0
Fork 0
mirror of https://github.com/hajimehoshi/ebiten.git synced 2025-02-04 15:04:28 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fix screenshot/internal-images file names

Browse Source
This commit is contained in:
Hajime Hoshi 2018-10-13 03:23:19 +09:00
parent f59f1990e7
commit 7f291b7bb1
1 changed files with 29 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
run.go
View File

@ -121,6 +121,26 @@ type imageDumper struct {
toDumpInternalImages bool toDumpInternalImages bool
} }
// availableFilename returns a filename that is valid as a new file or directory.
func availableFilename(prefix, postfix string) (string, error) {
const datetimeFormat = "20060102030405"
now := time.Now()
name := fmt.Sprintf("%s%s%s", prefix, now.Format(datetimeFormat), postfix)
for i := 1; ; i++ {
if _, err := os.Stat(name); err != nil {
if os.IsNotExist(err) {
break
}
if !os.IsNotExist(err) {
return "", err
}
}
name = fmt.Sprintf("%s%s_%d%s", prefix, now.Format(datetimeFormat), i, postfix)
}
return name, nil
}
func (i *imageDumper) update(screen *Image) error { func (i *imageDumper) update(screen *Image) error {
const ( const (
envScreenshotKey = "EBITEN_SCREENSHOT_KEY" envScreenshotKey = "EBITEN_SCREENSHOT_KEY"
@ -184,7 +204,7 @@ func (i *imageDumper) update(screen *Image) error {
if i.toTakeScreenshot { if i.toTakeScreenshot {
dump := func() (string, error) { dump := func() (string, error) {
f, err := ioutil.TempFile("", "ebiten_screenshot_") f, err := ioutil.TempFile("", "")
if err != nil { if err != nil {
return "", err return "", err
} }
@ -201,17 +221,9 @@ func (i *imageDumper) update(screen *Image) error {
return err return err
} }
newname := fmt.Sprintf("%s.png", time.Now().Format("20060102030405")) newname, err := availableFilename("screenshot_", ".png")
for i := 1; ; i++ { if err != nil {
if _, err := os.Stat(newname); err != nil { return err
if os.IsNotExist(err) {
break
}
if !os.IsNotExist(err) {
return err
}
}
newname = fmt.Sprintf("%s_%d.png", time.Now().Format("20060102030405"), i)
} }
if err := os.Rename(name, newname); err != nil { if err := os.Rename(name, newname); err != nil {
@ -223,11 +235,15 @@ func (i *imageDumper) update(screen *Image) error {
} }
if i.toDumpInternalImages { if i.toDumpInternalImages {
dir, err := ioutil.TempDir("", "ebiten_internal_images_") dir, err := availableFilename("internalimages_", "")
if err != nil { if err != nil {
return err return err
} }
if err := os.Mkdir(dir, 0755); err != nil {
return err
}
dump := func(img image.Image, index int) error { dump := func(img image.Image, index int) error {
filename := filepath.Join(dir, fmt.Sprintf("%d.png", index)) filename := filepath.Join(dir, fmt.Sprintf("%d.png", index))
f, err := os.Create(filename) f, err := os.Create(filename)