|
1 | 1 | package assets
|
2 | 2 |
|
3 | 3 | import (
|
4 |
| - "image" |
5 |
| - _ "image/jpeg" |
6 |
| - _ "image/png" |
7 |
| - "os" |
8 |
| - "path/filepath" |
9 |
| - "strings" |
| 4 | + "image" |
| 5 | + _ "image/jpeg" |
| 6 | + _ "image/png" |
| 7 | + "os" |
| 8 | + "path/filepath" |
| 9 | + "strings" |
10 | 10 |
|
11 |
| - "candy/audio" |
| 11 | + "candy/audio" |
12 | 12 | )
|
13 | 13 |
|
14 | 14 | var imageExtensions = map[string]struct{}{"jpg": {}, "png": {}}
|
15 | 15 | var audioExtensions = map[string]struct{}{"mp3": {}, "wav": {}}
|
16 | 16 |
|
17 | 17 | type Assets struct {
|
18 |
| - imageMap map[string]image.Image |
19 |
| - audioMap map[string]audio.Audio |
| 18 | + imageMap map[string]image.Image |
| 19 | + audioMap map[string]audio.Audio |
20 | 20 | }
|
21 | 21 |
|
22 | 22 | func (a Assets) GetImage(imageName string) image.Image {
|
23 |
| - return a.imageMap[convertPath(imageName)] |
| 23 | + return a.imageMap[convertPath(imageName)] |
24 | 24 | }
|
25 | 25 |
|
26 | 26 | func (a Assets) GetAudio(audioName string) audio.Audio {
|
27 |
| - return a.audioMap[convertPath(audioName)] |
| 27 | + return a.audioMap[convertPath(audioName)] |
28 | 28 | }
|
29 | 29 |
|
30 | 30 | func convertPath(path string) string {
|
31 |
| - parts := strings.Split(path, "/") |
32 |
| - return filepath.Join(parts...) |
| 31 | + parts := strings.Split(path, "/") |
| 32 | + return filepath.Join(parts...) |
33 | 33 | }
|
34 | 34 |
|
35 | 35 | func LoadAssets(assetRootDir string) (Assets, error) {
|
36 |
| - imageAssets, err := loadImages(assetRootDir) |
37 |
| - if err != nil { |
38 |
| - return Assets{}, err |
39 |
| - } |
40 |
| - audioAssets, err := loadAudios(assetRootDir) |
41 |
| - return Assets{ |
42 |
| - imageMap: imageAssets, |
43 |
| - audioMap: audioAssets, |
44 |
| - }, err |
| 36 | + imageAssets, err := loadImages(assetRootDir) |
| 37 | + if err != nil { |
| 38 | + return Assets{}, err |
| 39 | + } |
| 40 | + audioAssets, err := loadAudios(assetRootDir) |
| 41 | + return Assets{ |
| 42 | + imageMap: imageAssets, |
| 43 | + audioMap: audioAssets, |
| 44 | + }, err |
45 | 45 | }
|
46 | 46 |
|
47 | 47 | func loadAudios(assetRootDir string) (map[string]audio.Audio, error) {
|
48 |
| - audioAssets := make(map[string]audio.Audio) |
49 |
| - err := lisFiles(assetRootDir, func(path string, ext string, rel string) error { |
50 |
| - if _, ok := audioExtensions[ext]; !ok { |
51 |
| - return nil |
52 |
| - } |
53 |
| - file, err := os.Open(path) |
54 |
| - if err != nil { |
55 |
| - return err |
56 |
| - } |
57 |
| - bp, err := audio.NewAudio(file, ext) |
58 |
| - if err != nil { |
59 |
| - return err |
60 |
| - } |
61 |
| - audioAssets[rel] = bp |
62 |
| - return nil |
63 |
| - }) |
64 |
| - return audioAssets, err |
| 48 | + audioAssets := make(map[string]audio.Audio) |
| 49 | + err := lisFiles(assetRootDir, func(path string, ext string, rel string) error { |
| 50 | + if _, ok := audioExtensions[ext]; !ok { |
| 51 | + return nil |
| 52 | + } |
| 53 | + file, err := os.Open(path) |
| 54 | + if err != nil { |
| 55 | + return err |
| 56 | + } |
| 57 | + bp, err := audio.NewAudio(file, ext) |
| 58 | + if err != nil { |
| 59 | + return err |
| 60 | + } |
| 61 | + audioAssets[rel] = bp |
| 62 | + return nil |
| 63 | + }) |
| 64 | + return audioAssets, err |
65 | 65 | }
|
66 | 66 |
|
67 | 67 | func loadImages(assetRootDir string) (map[string]image.Image, error) {
|
68 |
| - imageAssets := make(map[string]image.Image) |
69 |
| - err := lisFiles(assetRootDir, func(path string, ext string, rel string) error { |
70 |
| - if _, ok := imageExtensions[ext]; !ok { |
71 |
| - return nil |
72 |
| - } |
73 |
| - file, err := os.Open(path) |
74 |
| - defer file.Close() |
75 |
| - if err != nil { |
76 |
| - return err |
77 |
| - } |
78 |
| - img, _, err := image.Decode(file) |
79 |
| - if err != nil { |
80 |
| - return err |
81 |
| - } |
82 |
| - imageAssets[rel] = img |
83 |
| - return nil |
84 |
| - }) |
85 |
| - return imageAssets, err |
| 68 | + imageAssets := make(map[string]image.Image) |
| 69 | + err := lisFiles(assetRootDir, func(path string, ext string, rel string) error { |
| 70 | + if _, ok := imageExtensions[ext]; !ok { |
| 71 | + return nil |
| 72 | + } |
| 73 | + file, err := os.Open(path) |
| 74 | + defer file.Close() |
| 75 | + if err != nil { |
| 76 | + return err |
| 77 | + } |
| 78 | + img, _, err := image.Decode(file) |
| 79 | + if err != nil { |
| 80 | + return err |
| 81 | + } |
| 82 | + imageAssets[rel] = img |
| 83 | + return nil |
| 84 | + }) |
| 85 | + return imageAssets, err |
86 | 86 | }
|
87 | 87 |
|
88 | 88 | func lisFiles(rootDir string, processFile func(path string, ext string, rel string) error) error {
|
89 |
| - return filepath.Walk(rootDir, func(path string, info os.FileInfo, err error) error { |
90 |
| - if info.IsDir() { |
91 |
| - return nil |
92 |
| - } |
93 |
| - parts := strings.Split(info.Name(), ".") |
94 |
| - if len(parts) == 0 { |
95 |
| - return nil |
96 |
| - } |
97 |
| - ext := parts[len(parts)-1] |
98 |
| - rel, err := filepath.Rel(rootDir, path) |
99 |
| - return processFile(path, ext, rel) |
100 |
| - }) |
| 89 | + return filepath.Walk(rootDir, func(path string, info os.FileInfo, err error) error { |
| 90 | + if info.IsDir() { |
| 91 | + return nil |
| 92 | + } |
| 93 | + parts := strings.Split(info.Name(), ".") |
| 94 | + if len(parts) == 0 { |
| 95 | + return nil |
| 96 | + } |
| 97 | + ext := parts[len(parts)-1] |
| 98 | + rel, err := filepath.Rel(rootDir, path) |
| 99 | + return processFile(path, ext, rel) |
| 100 | + }) |
101 | 101 | }
|
0 commit comments