hdk-core/src/Middleware/SessionMiddleware.php

<?php
/**
 * SessionMiddleware.php@HDK-Core
 *
 * @author 李东云 <Dongyun.Li@LuxCreo.Ai>
 * Powered by PhpStorm
 * Created on 2023/3/8
 */

namespace Singularity\HDK\Core\Middleware;

use Carbon\Carbon;
use Hyperf\Contract\ConfigInterface;
use Hyperf\Contract\SessionInterface;
use Hyperf\HttpMessage\Cookie\Cookie;
use Hyperf\HttpMessage\Server\Response;
use Hyperf\Session\SessionManager;
use Psr\Http\Message\ResponseInterface;
use Psr\Http\Message\ServerRequestInterface;

/**
 * Singularity\HDK\Core\Middleware\SessionMiddleware@HDK-Core
 *
 * @author 李东云 <Dongyun.Li@LuxCreo.Ai>
 * Powered by PhpStorm
 * Created on 2023/3/8
 */
class SessionMiddleware extends \Hyperf\Session\Middleware\SessionMiddleware
{
    public function __construct(private SessionManager $sessionManager, private ConfigInterface $config)
    {
        parent::__construct($this->sessionManager, $this->config);
    }

    /**
     * Add the session cookie to the response·.
     */
    private function addCookieToResponse(
        ServerRequestInterface $request,
        ResponseInterface $response,
        SessionInterface $session
    ): ResponseInterface {
        $cookie = new Cookie(
            name: $session->getName(),
            value: $session->getId(),
            expire: $this->getCookieExpirationDate(),
            path: $this->config->get('session.options.path', '/'),
            domain: $this->config->get('session.options.domain', $request->getUri()->getHost()),
            secure: $this->config->get('session.options.secure', strtolower($request->getUri()->getScheme()) === 'https'),
            httpOnly: true,
            sameSite: $this->config->get('session.options.samesite', Cookie::SAMESITE_LAX)
        );
        if (!method_exists($response, 'withCookie')) {
            return $response->withHeader('Set-Cookie', (string)$cookie);
        }
        /* @var Response $response */
        return $response->withCookie($cookie);
    }

    /**
     * Get the session lifetime in seconds.
     */
    private function getCookieExpirationDate(): int
    {
        if ($this->config->get('session.options.expire_on_close')) {
            $expirationDate = 0;
        } else {
            $expireSeconds = $this->config->get('session.options.cookie_lifetime', 5 * 60 * 60);
            $expirationDate = Carbon::now()->addSeconds($expireSeconds)->getTimestamp();
        }
        return $expirationDate;
    }
}