kiri-core/HttpServer/Http/Response.php

<?php
/**
 * Created by PhpStorm.
 * User: whwyy
 * Date: 2018/4/24 0024
 * Time: 19:39
 */

namespace HttpServer\Http;

use HttpServer\Application;
use HttpServer\Http\Formatter\HtmlFormatter;
use HttpServer\Http\Formatter\JsonFormatter;
use HttpServer\Http\Formatter\XmlFormatter;
use Exception;
use Snowflake\Core\Help;
use Snowflake\Exception\ComponentException;
use Snowflake\Snowflake;
use Swoole\Http\Response as SResponse;

/**
 * Class Response
 * @package Snowflake\Snowflake\Http
 */
class Response extends Application
{

	const JSON = 'json';
	const XML = 'xml';
	const HTML = 'html';

	/** @var string */
	public $format = null;

	/** @var int */
	public $statusCode = 200;

	/** @var SResponse */
	public $response;
	public $isWebSocket = false;
	public $headers = [];

	private $startTime = 0;

	private $_format_maps = [
		self::JSON => JsonFormatter::class,
		self::XML  => XmlFormatter::class,
		self::HTML => HtmlFormatter::class
	];

	public $fd = 0;

	/**
	 * @param $format
	 * @return $this
	 */
	public function setFormat($format)
	{
		$this->format = $format;
		return $this;
	}

	/**
	 * 清理无用数据
	 */
	public function clear()
	{
		$this->fd = 0;
		$this->isWebSocket = false;
		$this->format = null;
	}

	/**
	 * @return string
	 */
	public function getContentType()
	{
		if ($this->format == null || $this->format == static::JSON) {
			return 'application/json;charset=utf-8';
		} else if ($this->format == static::XML) {
			return 'application/xml;charset=utf-8';
		} else {
			return 'text/html;charset=utf-8';
		}
	}

	/**
	 * @return mixed
	 * @throws Exception
	 */
	public function sender()
	{
		return $this->send(func_get_args());
	}

	/**
	 * @param $key
	 * @param $value
	 */
	public function addHeader($key, $value)
	{
		$response = Context::getContext('response');
		$response->header($key, $value);
	}

	/**
	 * @param string $context
	 * @param int $statusCode
	 * @param null $response
	 * @return bool
	 * @throws Exception
	 */
	public function send($context = '', $statusCode = 200, $response = null)
	{
		$sendData = $this->parseData($context);
		if ($response instanceof SResponse) {
			$this->response = $response;
		}
		if ($this->response instanceof SResponse) {
			return $this->sendData($this->response, $sendData, $statusCode);
		} else {
			return $this->printResult($sendData);
		}
	}

	/**
	 * @param $context
	 * @return mixed
	 * @throws Exception
	 */
	private function parseData($context)
	{
		if (isset($this->_format_maps[$this->format])) {
			$config['class'] = $this->_format_maps[$this->format];
		} else {
			$config['class'] = HtmlFormatter::class;
		}
		$formatter = Snowflake::createObject($config);
		return $formatter->send($context)->getData();
	}

	/**
	 * @param $result
	 * @return string
	 * @throws Exception
	 */
	private function printResult($result)
	{
		$result = Help::toString($result);

		$string = 'Command Result: ' . PHP_EOL;
		$string .= empty($result) ? 'success!' : $result . PHP_EOL;
		$string .= 'Command Success!' . PHP_EOL;
		echo $string;

		$event = Snowflake::app()->event;
		$event->trigger('CONSOLE_END');

		return 'ok';
	}

	/**
	 * @param $response
	 * @param $sendData
	 * @param $status
	 * @return mixed
	 */
	private function sendData($response, $sendData, $status)
	{
		$response->status($status);
		$response->header('Content-Type', $this->getContentType());
		$response->header('Access-Control-Allow-Origin', '*');
		$response->header('Run-Time', $this->getRuntime());
		return $response->end($sendData);
	}

	/**
	 * @param $url
	 * @param array $param
	 * @return int
	 */
	public function redirect($url, array $param = [])
	{
		if (!empty($param)) {
			$url .= '?' . http_build_query($param);
		}
		$url = ltrim($url, '/');
		if (!preg_match('/^http/', $url)) {
			$url = '/' . $url;
		}
		return $this->response->redirect($url);
	}

	/**
	 * @param null $response
	 * @return mixed
	 * @throws ComponentException
	 */
	public static function create($response = null)
	{
		$ciResponse = Snowflake::app()->clone('response');
		$ciResponse->response = $response;
		$ciResponse->startTime = microtime(true);
		$ciResponse->format = self::JSON;
		return $ciResponse;
	}


	/**
	 * @throws Exception
	 */
	public function sendNotFind()
	{
		$this->format = static::HTML;
		$this->send('', 404);
	}

	/**
	 * @return string
	 */
	public function getRuntime()
	{
		return sprintf('%.5f', microtime(TRUE) - $this->startTime);
	}

}