|
1 | 1 | import AnitomyNative from './dist/anitomyscript.js' |
2 | | -let anitomyModule |
3 | 2 |
|
4 | | -export default async function (file) { |
5 | | - if (!Array.isArray(file) && typeof file !== 'string') { |
6 | | - return new Error('Input must be either an Array or a string') |
| 3 | +const FIELDS = [ |
| 4 | + 'anime_season', 'season_prefix', 'anime_title', |
| 5 | + 'anime_type', 'anime_year', 'audio_term', |
| 6 | + 'device_compatibility', 'episode_number', 'episode_number_alt', |
| 7 | + 'episode_prefix', 'episode_title', 'file_checksum', |
| 8 | + 'file_extension', 'file_name', 'language', |
| 9 | + 'other', 'release_group', 'release_information', |
| 10 | + 'release_version', 'source', 'subtitles', |
| 11 | + 'video_resolution', 'video_term', 'volume_number', |
| 12 | + 'volume_prefix', 'unknown', |
| 13 | +] |
| 14 | + |
| 15 | +let modPromise, pairs |
| 16 | + |
| 17 | +export default async function parse (input) { |
| 18 | + if (!Array.isArray(input) && typeof input !== 'string') { |
| 19 | + throw new Error('Input must be either an Array or a string') |
7 | 20 | } |
8 | 21 |
|
9 | | - if (anitomyModule) { |
10 | | - return parse(file) |
11 | | - } |
12 | | - |
13 | | - anitomyModule = await AnitomyNative() |
14 | | - return parse(file) |
15 | | -} |
16 | | - |
17 | | -async function parse (file) { |
18 | | - const vector = mapArray(file) |
19 | | - const result = mapVector(anitomyModule.parseMultiple(vector)) |
20 | | - vector.delete() |
21 | | - return result.map((each) => elements(each)) |
22 | | -} |
23 | | - |
24 | | -function elements (elements) { |
25 | | - const returnObj = { |
26 | | - anime_season: elementEntry(elements, anitomyModule.ElementCategory.kElementAnimeSeason), |
27 | | - season_prefix: elementEntry(elements, anitomyModule.ElementCategory.kElementAnimeSeasonPrefix), |
28 | | - anime_title: elementEntry(elements, anitomyModule.ElementCategory.kElementAnimeTitle), |
29 | | - anime_type: elementEntry(elements, anitomyModule.ElementCategory.kElementAnimeType), |
30 | | - anime_year: elementEntry(elements, anitomyModule.ElementCategory.kElementAnimeYear), |
31 | | - audio_term: elementEntry(elements, anitomyModule.ElementCategory.kElementAudioTerm), |
32 | | - device_compatibility: elementEntry(elements, anitomyModule.ElementCategory.kElementDeviceCompatibility), |
33 | | - episode_number: elementEntry(elements, anitomyModule.ElementCategory.kElementEpisodeNumber), |
34 | | - episode_prefix: elementEntry(elements, anitomyModule.ElementCategory.kElementEpisodePrefix), |
35 | | - episode_title: elementEntry(elements, anitomyModule.ElementCategory.kElementEpisodeTitle), |
36 | | - file_checksum: elementEntry(elements, anitomyModule.ElementCategory.kElementFileChecksum), |
37 | | - file_extension: elementEntry(elements, anitomyModule.ElementCategory.kElementFileExtension), |
38 | | - file_name: elementEntry(elements, anitomyModule.ElementCategory.kElementFileName), |
39 | | - language: elementEntry(elements, anitomyModule.ElementCategory.kElementLanguage), |
40 | | - other: elementEntry(elements, anitomyModule.ElementCategory.kElementOther), |
41 | | - release_group: elementEntry(elements, anitomyModule.ElementCategory.kElementReleaseGroup), |
42 | | - release_information: elementEntry(elements, anitomyModule.ElementCategory.kElementReleaseInformation), |
43 | | - release_version: elementEntry(elements, anitomyModule.ElementCategory.kElementReleaseVersion), |
44 | | - source: elementEntry(elements, anitomyModule.ElementCategory.kElementSource), |
45 | | - subtitles: elementEntry(elements, anitomyModule.ElementCategory.kElementSubtitles), |
46 | | - video_resolution: elementEntry(elements, anitomyModule.ElementCategory.kElementVideoResolution), |
47 | | - video_term: elementEntry(elements, anitomyModule.ElementCategory.kElementVideoTerm), |
48 | | - volume_number: elementEntry(elements, anitomyModule.ElementCategory.kElementVolumeNumber), |
49 | | - volume_prefix: elementEntry(elements, anitomyModule.ElementCategory.kElementVolumePrefix), |
50 | | - unknown: elementEntry(elements, anitomyModule.ElementCategory.kElementUnknown) |
51 | | - } |
52 | | - elements.delete() |
53 | | - return returnObj |
54 | | -} |
55 | | - |
56 | | -function elementEntry (elements, key) { |
57 | | - return mapVector(elements.get_all(key)) |
58 | | -} |
59 | | - |
60 | | -function mapArray (array) { |
61 | | - const vector = new anitomyModule.StringVector() |
62 | | - array.forEach((element, index) => { |
63 | | - if (typeof element !== 'string') { |
64 | | - throw new Error(`Element at index ${index} is not a string`) |
| 22 | + const mod = await (modPromise ??= AnitomyNative()) |
| 23 | + pairs ??= FIELDS.map((name, i) => [mod.ElementCategory.values[i], name]) |
| 24 | + const files = typeof input === 'string' ? [input] : input |
| 25 | + const anit = new mod.Anitomy() |
| 26 | + const results = [] |
| 27 | + |
| 28 | + try { |
| 29 | + for (let i = 0; i < files.length; i++) { |
| 30 | + const file = files[i] |
| 31 | + if (typeof file !== 'string') throw new Error(`Element at index ${i} is not a string`) |
| 32 | + anit.parse(file) |
| 33 | + const obj = {} |
| 34 | + for (const [cat, field] of pairs) { |
| 35 | + const vec = anit.get_all(cat) |
| 36 | + const length = vec.size() |
| 37 | + if (!length) continue |
| 38 | + obj[field] = Array.from({ length }, (_, j) => vec.get(j)) |
| 39 | + } |
| 40 | + results.push(obj) |
65 | 41 | } |
66 | | - vector.push_back(element) |
67 | | - }) |
68 | | - return vector |
69 | | -} |
70 | | - |
71 | | -function mapVector (vector) { |
72 | | - const array = [] |
73 | | - for (let index = 0; index < vector.size(); index++) { |
74 | | - array.push(vector.get(index)) |
| 42 | + } finally { |
| 43 | + anit.delete() |
75 | 44 | } |
76 | | - vector.delete() |
77 | | - return array |
| 45 | + |
| 46 | + return results |
78 | 47 | } |
0 commit comments