upper/src/drivers/Varnish.php at d119d8381d6d5549573d0e7bb02c2d2c17f70573 · ostark/upper · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
<?php namespace ostark\upper\drivers;

use Craft;
use GuzzleHttp\Client;
use GuzzleHttp\Exception\GuzzleException;

/**
 * Class Varnish Driver
 *
 * @package ostark\upper\drivers
 */
class Varnish extends AbstractPurger implements CachePurgeInterface
{
    /**
     * @var string
     */
    public $purgeHeaderName;

    /**
     * @var string
     */
    public $purgeUrl;

    /**
     * @var array
     */
    public $headers = [];

    /**
     * @param string $tag
     */
    public function purgeTag(string $tag)
    {
        if ($this->useLocalTags) {
            return $this->purgeUrlsByTag($tag);
        }

        return $this->sendPurgeRequest([
                'headers'  => $this->headers + [$this->purgeHeaderName => $tag]
            ]
        );
    }

    /**
     * @param array $urls
     *
     * @return bool
     */
    public function purgeUrls(array $urls)
    {
        foreach ($urls as $url) {
            $success = $this->sendPurgeRequest([
                    'url'      => $url,
                    'headers'  => $this->headers
                ]
            );

            if (!$success) {
                return false;
            }
        }

        return true;

    }


    /**
     * Purge entire cache
     *
     * Requires a custom vcl config
     *
     * @see https://varnish-cache.org/docs/6.0/users-guide/purging.html#bans
     *
     * @return bool
     */
    public function purgeAll()
    {
        $this->headers['Varnish_Ban_Key'] = getenv('VARNISH_BAN_KEY');
        return $this->sendPurgeRequest([
            'headers'  => $this->headers
        ], 'BAN');
    }


    protected function sendPurgeRequest(array $options = [], $method = 'PURGE')
    {
        $success = true;
        $purgeUrls = explode(',', $this->purgeUrl);
        foreach ($purgeUrls as $purgeUrl) {
            Craft::info($method . ' Varnish cache ' . $purgeUrl);

            $options['base_uri'] = $purgeUrl;
            if (isset($options['url'])) {
                $options['base_uri'] .= $options['url'];
            }

            try {
                $response = (new Client($options))->request($method);

                if ($success) {
                    $success = in_array($response->getStatusCode(), [204, 200]);
                }
            } catch (GuzzleException $guzzleException) {
                $success = false;
                Craft::warning($guzzleException->getMessage());
            }
        }

        return $success;
    }
}