calibre-web/cps/audio.py

# -*- coding: utf-8 -*-

#  This file is part of the Calibre-Web (https://github.com/janeczku/calibre-web)
#    Copyright (C) 2024 Ozzieisaacs
#
#  This program is free software: you can redistribute it and/or modify
#  it under the terms of the GNU General Public License as published by
#  the Free Software Foundation, either version 3 of the License, or
#  (at your option) any later version.
#
#  This program is distributed in the hope that it will be useful,
#  but WITHOUT ANY WARRANTY; without even the implied warranty of
#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#  GNU General Public License for more details.
#
#  You should have received a copy of the GNU General Public License
#  along with this program. If not, see <http://www.gnu.org/licenses/>.

import os

import mutagen
import base64
from . import cover, logger

from cps.constants import BookMeta

log = logger.create()

def get_audio_file_info(tmp_file_path, original_file_extension, original_file_name, no_cover_processing):
    tmp_cover_name = None
    audio_file = mutagen.File(tmp_file_path)
    comments = None
    if original_file_extension in [".mp3", ".wav", ".aiff"]:
        cover_data = list()
        for key, val in audio_file.tags.items():
            if key.startswith("APIC:"):
                cover_data.append(val)
            if key.startswith("COMM:"):
                comments = val.text[0]
        title = audio_file.tags.get('TIT2').text[0] if "TIT2" in audio_file.tags else None
        author = audio_file.tags.get('TPE1').text[0] if "TPE1" in audio_file.tags else None
        if author is None:
            author = audio_file.tags.get('TPE2').text[0] if "TPE2" in audio_file.tags else None
        tags = audio_file.tags.get('TCON').text[0] if "TCON" in audio_file.tags else None # Genre
        series = audio_file.tags.get('TALB').text[0] if "TALB" in audio_file.tags else None# Album
        series_id = audio_file.tags.get('TRCK').text[0] if "TRCK" in audio_file.tags else None # track no.
        publisher = audio_file.tags.get('TPUB').text[0] if "TPUB" in audio_file.tags else None
        pubdate = str(audio_file.tags.get('TDRL').text[0]) if "TDRL" in audio_file.tags else None
        if not pubdate:
            pubdate = str(audio_file.tags.get('TDRC').text[0]) if "TDRC" in audio_file.tags else None
            if not pubdate:
                pubdate = str(audio_file.tags.get('TDOR').text[0]) if "TDOR" in audio_file.tags else None
        if cover_data and not no_cover_processing:
            tmp_cover_name = os.path.join(os.path.dirname(tmp_file_path), 'cover.jpg')
            cover_info = cover_data[0]
            for dat in cover_data:
                if dat.type == mutagen.id3.PictureType.COVER_FRONT:
                    cover_info = dat
                    break
            cover.cover_processing(tmp_file_path, cover_info.data, "." + cover_info.mime[-3:])
    elif original_file_extension in [".ogg", ".flac", ".opus", ".ogv"]:
        title = audio_file.tags.get('TITLE')[0] if "TITLE" in audio_file else None
        author = audio_file.tags.get('ARTIST')[0] if "ARTIST" in audio_file else None
        comments = audio_file.tags.get('COMMENTS')[0] if "COMMENTS" in audio_file else None
        tags = audio_file.tags.get('GENRE')[0] if "GENRE" in audio_file else None # Genre
        series = audio_file.tags.get('ALBUM')[0] if "ALBUM" in audio_file else None
        series_id = audio_file.tags.get('TRACKNUMBER')[0] if "TRACKNUMBER" in audio_file else None
        publisher = audio_file.tags.get('LABEL')[0] if "LABEL" in audio_file else None
        pubdate = audio_file.tags.get('DATE')[0] if "DATE" in audio_file else None
        cover_data = audio_file.tags.get('METADATA_BLOCK_PICTURE')
        if not no_cover_processing:
            if cover_data:
                tmp_cover_name = os.path.join(os.path.dirname(tmp_file_path), 'cover.jpg')
                cover_info = mutagen.flac.Picture(base64.b64decode(cover_data[0]))
                cover.cover_processing(tmp_file_path, cover_info.data, "." + cover_info.mime[-3:])
            if hasattr(audio_file, "pictures"):
                cover_info = audio_file.pictures[0]
                for dat in audio_file.pictures:
                    if dat.type == mutagen.id3.PictureType.COVER_FRONT:
                        cover_info = dat
                        break
                tmp_cover_name = os.path.join(os.path.dirname(tmp_file_path), 'cover.jpg')
                cover.cover_processing(tmp_file_path, cover_info.data, "." + cover_info.mime[-3:])
    elif original_file_extension in [".aac"]:
        title = audio_file.tags.get('Title').value if "Title" in audio_file else None
        author = audio_file.tags.get('Artist').value if "Artist" in audio_file else None
        comments = audio_file.tags.get('Comment').value if "Comment" in audio_file else None
        tags = audio_file.tags.get('Genre').value if "Genre" in audio_file else None
        series = audio_file.tags.get('Album').value if "Album" in audio_file else None
        series_id = audio_file.tags.get('Track').value if "Track" in audio_file else None
        publisher = audio_file.tags.get('Label').value if "Label" in audio_file else None
        pubdate = audio_file.tags.get('Year').value if "Year" in audio_file else None
        cover_data = audio_file.tags['Cover Art (Front)']
        if cover_data and not no_cover_processing:
            tmp_cover_name = os.path.join(os.path.dirname(tmp_file_path), 'cover.jpg')
            with open(tmp_cover_name, "wb") as cover_file:
                cover_file.write(cover_data.value.split(b"\x00",1)[1])
    elif original_file_extension in [".asf"]:
        title = audio_file.tags.get('Title')[0].value if "Title" in audio_file else None
        author = audio_file.tags.get('Artist')[0].value if "Artist" in audio_file else None
        comments = audio_file.tags.get('Comments')[0].value if "Comments" in audio_file else None
        tags = audio_file.tags.get('Genre')[0].value if "Genre" in audio_file else None
        series = audio_file.tags.get('Album')[0].value if "Album" in audio_file else None
        series_id = audio_file.tags.get('Track')[0].value if "Track" in audio_file else None
        publisher = audio_file.tags.get('Label')[0].value if "Label" in audio_file else None
        pubdate = audio_file.tags.get('Year')[0].value if "Year" in audio_file else None
        cover_data = audio_file.tags.get('WM/Picture', None)
        if cover_data and not no_cover_processing:
            tmp_cover_name = os.path.join(os.path.dirname(tmp_file_path), 'cover.jpg')
            with open(tmp_cover_name, "wb") as cover_file:
                cover_file.write(cover_data[0].value)
    elif original_file_extension in [".mp4", ".m4a", ".m4b"]:
        title = audio_file.tags.get('©nam')[0] if "©nam" in audio_file.tags else None
        author = audio_file.tags.get('©ART')[0] if "©ART" in audio_file.tags else None
        comments = audio_file.tags.get('©cmt')[0] if "©cmt" in audio_file.tags else None
        tags = audio_file.tags.get('©gen')[0] if "©gen" in audio_file.tags else None
        series = audio_file.tags.get('©alb')[0] if "©alb" in audio_file.tags else None
        series_id = str(audio_file.tags.get('trkn')[0][0]) if "trkn" in audio_file.tags else None
        publisher = ""
        pubdate = audio_file.tags.get('©day')[0] if "©day" in audio_file.tags else None
        cover_data = audio_file.tags.get('covr', None)
        if cover_data and not no_cover_processing:
            tmp_cover_name = os.path.join(os.path.dirname(tmp_file_path), 'cover.jpg')
            cover_type = None
            for c in cover_data:
                if c.imageformat == mutagen.mp4.AtomDataType.JPEG:
                    cover_type =".jpg"
                    cover_bin = c
                    break
                elif c.imageformat == mutagen.mp4.AtomDataType.PNG:
                    cover_type = ".png"
                    cover_bin = c
                    break
            if cover_type:
                cover.cover_processing(tmp_file_path, cover_bin, cover_type)
            else:
                logger.error("Unknown covertype in file {} ".format(original_file_name))

    return BookMeta(
        file_path=tmp_file_path,
        extension=original_file_extension,
        title=title or original_file_name ,
        author="Unknown" if author is None else author,
        cover=tmp_cover_name,
        description="" if comments is None else comments,
        tags="" if tags is None else tags,
        series="" if series is None else series,
        series_id="1" if series_id is None else series_id.split("/")[0],
        languages="",
        publisher= "" if publisher is None else publisher,
        pubdate="" if pubdate is None else pubdate,
        identifiers=[],
    )