Merge pull request #10 from billyct/master

🐰 增加两个 plugin：PrivateDownloadUrl，RefreshFile

src/Plugins/PrivateDownloadUrl.php

@@ -0,0 +1,18 @@
+<?php
+
+namespace Overtrue\Flysystem\Qiniu\Plugins;
+
+use League\Flysystem\Plugin\AbstractPlugin;
+
+class PrivateDownloadUrl extends AbstractPlugin
+{
+    public function getMethod()
+    {
+        return 'privateDownloadUrl';
+    }
+
+    public function handle($path, $expires = 3600)
+    {
+        return $this->filesystem->getAdapter()->privateDownloadUrl($path, $expires);
+    }
+}

src/Plugins/RefreshFile.php

@@ -0,0 +1,18 @@
+<?php
+
+namespace Overtrue\Flysystem\Qiniu\Plugins;
+
+use League\Flysystem\Plugin\AbstractPlugin;
+
+class RefreshFile extends AbstractPlugin
+{
+    public function getMethod()
+    {
+        return 'refresh';
+    }
+
+    public function handle($path = [])
+    {
+        return $this->filesystem->getAdapter()->refresh($path);
+    }
+}

src/QiniuAdapter.php

@@ -13,6 +13,7 @@
 use League\Flysystem\Adapter\Polyfill\NotSupportingVisibilityTrait;
 use League\Flysystem\Config;
 use Qiniu\Auth;
+use Qiniu\Cdn\CdnManager;
 use Qiniu\Storage\BucketManager;
 use Qiniu\Storage\UploadManager;
 
@@ -60,6 +61,11 @@ class QiniuAdapter extends AbstractAdapter
      */
     protected $bucketManager;
 
+    /**
+     * @var \Qiniu\Cdn\CdnManager
+     */
+    protected $cdnManager;
+
     /**
      * QiniuAdapter constructor.
      *
@@ -127,7 +133,7 @@ public function writeStream($path, $resource, Config $config)
 
         $response = $this->write($path, $contents, $config);
 
-        if ($response === false) {
+        if (false === $response) {
             return $response;
         }
 
@@ -360,6 +366,40 @@ public function fetch($path, $url)
         return $response;
     }
 
+    /**
+     * Get private file download url.
+     *
+     * @param string $path
+     * @param int    $expires
+     *
+     * @return string
+     */
+    public function privateDownloadUrl($path, $expires = 3600)
+    {
+        $url = $this->getUrl($path);
+
+        return  $this->getAuthManager()->privateDownloadUrl($url, $expires);
+    }
+
+    /**
+     * Refresh file cache.
+     *
+     * @param string|array $path
+     *
+     * @return array
+     */
+    public function refresh($path)
+    {
+        if (is_string($path)) {
+            $path = [$path];
+        }
+
+        // 将 $path 变成完整的 url
+        $urls = array_map([$this, 'getUrl'], $path);
+
+        return $this->getCdnManager()->refreshUrls($urls);
+    }
+
     /**
      * Get the mime-type of a file.
      *
@@ -426,6 +466,18 @@ public function setAuthManager(Auth $manager)
         return $this;
     }
 
+    /**
+     * @param CdnManager $manager
+     *
+     * @return $this
+     */
+    public function setCdnManager(CdnManager $manager)
+    {
+        $this->cdnManager = $manager;
+
+        return $this;
+    }
+
     /**
      * @return \Qiniu\Storage\BucketManager
      */
@@ -450,6 +502,14 @@ public function getUploadManager()
         return $this->uploadManager ?: $this->uploadManager = new UploadManager();
     }
 
+    /**
+     * @return \Qiniu\Cdn\CdnManager
+     */
+    public function getCdnManager()
+    {
+        return $this->cdnManager ?: $this->cdnManager = new CdnManager($this->getAuthManager());
+    }
+
     /**
      * Get the upload token.
      *

tests/QiniuAdapterTest.php

@@ -35,17 +35,19 @@ public function qiniuProvider()
         $authManager = \Mockery::mock('stdClass');
         $bucketManager = \Mockery::mock('stdClass');
         $uploadManager = \Mockery::mock('stdClass');
+        $cdnManager = \Mockery::mock('stdClass');
 
         $authManager->allows()->uploadToken('bucket')->andReturns('token');
 
         $adapter->allows([
             'getAuthManager' => $authManager,
             'getUploadManager' => $uploadManager,
             'getBucketManager' => $bucketManager,
+            'getCdnManager' => $cdnManager,
         ]);
 
         return [
-            [$adapter, compact('authManager', 'bucketManager', 'uploadManager')],
+            [$adapter, compact('authManager', 'bucketManager', 'uploadManager', 'cdnManager')],
         ];
     }
 
@@ -289,6 +291,28 @@ public function testGetSize($adapter)
         $this->assertSame('meta-data', $adapter->getSize('foo.md'));
     }
 
+    /**
+     * @dataProvider qiniuProvider
+     */
+    public function testPrivateDownloadUrl($adapter, $managers)
+    {
+        $managers['authManager']->expects()->privateDownloadUrl('http://domain.com/url', 3600)->andReturn('url');
+        $this->assertSame('url', $adapter->privateDownloadUrl('url'));
+
+        $managers['authManager']->expects()->privateDownloadUrl('http://domain.com/url', 7200)->andReturn('url');
+        $this->assertSame('url', $adapter->privateDownloadUrl('url', 7200));
+    }
+
+    /**
+     * @dataProvider qiniuProvider
+     */
+    public function testRefresh($adapter, $managers)
+    {
+        $managers['cdnManager']->expects()->refreshUrls(['http://domain.com/url'])->andReturn('url');
+        $this->assertSame('url', $adapter->refresh('url'));
+        $this->assertSame('url', $adapter->refresh(['url']));
+    }
+
     /**
      * @dataProvider qiniuProvider
      */