<?php

/**
 * OrangeHRM is a comprehensive Human Resource Management (HRM) System that captures
 * all the essential functionalities required for any enterprise.
 * Copyright (C) 2006 OrangeHRM Inc., http://www.orangehrm.com
 *
 * OrangeHRM 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.
 *
 * OrangeHRM 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 OrangeHRM.
 * If not, see <https://www.gnu.org/licenses/>.
 */

namespace OrangeHRM\CorporateBranding\Controller\File;

use OrangeHRM\Core\Controller\AbstractFileController;
use OrangeHRM\CorporateBranding\Dto\ThemeImage;
use OrangeHRM\CorporateBranding\Traits\ThemeServiceTrait;
use OrangeHRM\Framework\Http\Request;
use OrangeHRM\Framework\Http\Response;

class ImageAttachmentController extends AbstractFileController
{
    use ThemeServiceTrait;

    /**
     * @param Request $request
     * @return Response
     */
    public function handle(Request $request): Response
    {
        $imageName = $request->attributes->get('imageName');
        $map = [
            'clientLogo' => 'client_logo',
            'clientBanner' => 'client_banner',
            'loginBanner' => 'login_banner',
        ];

        if (isset($map[$imageName])) {
            $imageKey = $map[$imageName];
            $image = $this->getThemeService()
                ->getThemeDao()
                ->getImageByImageKeyAndThemeName($imageKey);
            if ($image instanceof ThemeImage) {
                $response = $this->getResponse();
                $this->setCommonHeadersToResponse(
                    $image->getFilename(),
                    $image->getFileType(),
                    $image->getFileSize(),
                    $response
                );
                $response->setContent($image->getContent());
                return $response;
            }
        }

        return $this->handleBadRequest();
    }
}
