<?php

/**
 * Copyright (c) 2015 cenci0 <alchimist94@gmail.com>
 * Copyright (c) 2015 the Pantsu.cat developers <hostmaster@pantsu.cat>
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */

class Response
{
    private $type;

    public function __construct($response_type = null)
    {
        switch ($response_type) {
            case 'csv':
                header('Content-Type: text/csv; charset=UTF-8');
                $this->type = $response_type;
                break;
            case 'gyazo':
                // Deprecated API since version 2.0.0
                // fallthrough
            case 'html':
                header('Content-Type: text/html; charset=UTF-8');
                $this->type = $response_type;
                break;
            case 'text':
                header('Content-Type: text/plain; charset=UTF-8');
                $this->type = $response_type;
                break;
            case 'json':
                header('Content-Type: application/json; charset=UTF-8');
                $this->type = $response_type;
                break;
            default:
                http_response_code(400);
                echo 'Invalid response type. Valid options are: csv, html, text, json.';
                break;
        }
    }

    public function error($code, $desc)
    {
        $response = null;

        switch ($this->type) {
            case 'csv':
                $response = $this->csv_error($desc);
                break;
            case 'html':
                $response = $this->html_error($code, $desc);
                break;
            case 'text':
                $response = $this->text_error($code, $desc);
                break;
            case 'json':
                $response = $this->json_error($code, $desc);
                break;
        }

        http_response_code($code);
        echo $response;
    }

    public function send($files)
    {
        $response = null;

        switch ($this->type) {
            case 'csv':
                $response = $this->csv_success($files);
                break;
            case 'html':
                $response = $this->html_success($files);
                break;
            case 'text':
                $response = $this->text_success($files);
                break;
            case 'json':
                $response = $this->json_success($files);
                break;
        }

        http_response_code(200);
        echo $response;
    }

    private static function csv_error($description)
    {
        return "error\n".$description."\n";
    }

    private static function csv_success($files)
    {
        $result = "name,url,hash,size\n";
        foreach ($files as $file) {
            $result .= $file['name'].','.
                       $file['url'].','.
                       $file['hash'].','.
                       $file['size']."\n";
        }

        return $result;
    }

    private static function html_error($code, $description)
    {
        return '<p>ERROR: ('.$code.') '.$description.'</p>';
    }

    private static function html_success($files)
    {
        foreach ($files as $file) {
            $result .=  '<a href="'.$file['url'].'">'.$file['url'].'</a><br>';
        }

        return $result;
    }

    private static function text_error($code, $description)
    {
        return 'ERROR: ('.$code.') '.$description;
    }

    private static function text_success($files)
    {
        foreach ($files as $file) {
            $result .= $file['url']."\n";
        }

        return $result;
    }

    private static function json_error($code, $description)
    {
        return json_encode(array(
            'success' => false,
            'errorcode' => $code,
            'description' => $description,
        ));
    }

    private static function json_success($files)
    {
        return json_encode(array(
            'success' => true,
            'files' => $files,
        ));
    }
}