Merge pull request #401 from jasonvarga/master

Statamic caching and localization support

Author: adamwathan

cli/drivers/StatamicValetDriver.php

@@ -48,30 +48,99 @@ public function isStaticFile($sitePath, $siteName, $uri)
      */
     public function frontControllerPath($sitePath, $siteName, $uri)
     {
-        if (file_exists($staticPath = $sitePath.'/static'.$uri.'/index.html')) {
+        if ($this->isActualFile($staticPath = $this->getStaticPath($sitePath))) {
             return $staticPath;
         }
 
-        $_SERVER['SCRIPT_NAME'] = '/index.php';
-
-        if (strpos($_SERVER['REQUEST_URI'], '/index.php') === 0) {
-            $_SERVER['REQUEST_URI'] = substr($_SERVER['REQUEST_URI'], 10);
+        if ($uri === '/installer.php') {
+            return $sitePath.'/installer.php';
         }
 
-        if ($uri === '') {
-            $uri = '/';
+        $scriptName = '/index.php';
+
+        if ($this->isActualFile($sitePath.'/index.php')) {
+            $indexPath = $sitePath.'/index.php';
         }
 
-        if ($uri === '/installer.php') {
-            return $sitePath.'/installer.php';
+        if ($isAboveWebroot = $this->isActualFile($sitePath.'/public/index.php')) {
+            $indexPath = $sitePath.'/public/index.php';
         }
 
-        if (file_exists($indexPath = $sitePath.'/index.php')) {
-            return $indexPath;
+        $sitePathPrefix = ($isAboveWebroot) ? $sitePath.'/public' : $sitePath;
+
+        if ($locale = $this->getUriLocale($uri)) {
+            if ($this->isActualFile($localeIndexPath = $sitePathPrefix . '/' . $locale . '/index.php')) {
+                // Force trailing slashes on locale roots.
+                if ($uri === '/' . $locale) {
+                    header('Location: ' . $uri . '/');
+                    die;
+                }
+
+                $indexPath = $localeIndexPath;
+                $scriptName = '/' . $locale . '/index.php';
+            }
         }
 
-        if (file_exists($indexPath = $sitePath.'/public/index.php')) {
-            return $indexPath;
+        $_SERVER['SCRIPT_NAME'] = $scriptName;
+        $_SERVER['SCRIPT_FILENAME'] = $sitePathPrefix . $scriptName;
+
+        return $indexPath;
+    }
+
+    /**
+     * Get the locale from this URI
+     *
+     * @param  string  $uri
+     * @return string|null
+     */
+    public function getUriLocale($uri)
+    {
+        $parts = explode('/', $uri);
+        $locale = $parts[1];
+
+        if (count($parts) < 2 || ! in_array($locale, $this->getLocales())) {
+            return;
         }
+
+        return $locale;
+    }
+
+    /**
+     * Get the list of possible locales used in the first segment of a URI
+     *
+     * @return array
+     */
+    public function getLocales()
+    {
+        return [
+            'af', 'ax', 'al', 'dz', 'as', 'ad', 'ao', 'ai', 'aq', 'ag', 'ar', 'am', 'aw', 'au', 'at', 'az', 'bs', 'bh',
+            'bd', 'bb', 'by', 'be', 'bz', 'bj', 'bm', 'bt', 'bo', 'bq', 'ba', 'bw', 'bv', 'br', 'io', 'bn', 'bg', 'bf',
+            'bi', 'cv', 'kh', 'cm', 'ca', 'ky', 'cf', 'td', 'cl', 'cn', 'cx', 'cc', 'co', 'km', 'cg', 'cd', 'ck', 'cr',
+            'ci', 'hr', 'cu', 'cw', 'cy', 'cz', 'dk', 'dj', 'dm', 'do', 'ec', 'eg', 'sv', 'gq', 'er', 'ee', 'et', 'fk',
+            'fo', 'fj', 'fi', 'fr', 'gf', 'pf', 'tf', 'ga', 'gm', 'ge', 'de', 'gh', 'gi', 'gr', 'gl', 'gd', 'gp', 'gu',
+            'gt', 'gg', 'gn', 'gw', 'gy', 'ht', 'hm', 'va', 'hn', 'hk', 'hu', 'is', 'in', 'id', 'ir', 'iq', 'ie', 'im',
+            'il', 'it', 'jm', 'jp', 'je', 'jo', 'kz', 'ke', 'ki', 'kp', 'kr', 'kw', 'kg', 'la', 'lv', 'lb', 'ls', 'lr',
+            'ly', 'li', 'lt', 'lu', 'mo', 'mk', 'mg', 'mw', 'my', 'mv', 'ml', 'mt', 'mh', 'mq', 'mr', 'mu', 'yt', 'mx',
+            'fm', 'md', 'mc', 'mn', 'me', 'ms', 'ma', 'mz', 'mm', 'na', 'nr', 'np', 'nl', 'nc', 'nz', 'ni', 'ne', 'ng',
+            'nu', 'nf', 'mp', 'no', 'om', 'pk', 'pw', 'ps', 'pa', 'pg', 'py', 'pe', 'ph', 'pn', 'pl', 'pt', 'pr', 'qa',
+            're', 'ro', 'ru', 'rw', 'bl', 'sh', 'kn', 'lc', 'mf', 'pm', 'vc', 'ws', 'sm', 'st', 'sa', 'sn', 'rs', 'sc',
+            'sl', 'sg', 'sx', 'sk', 'si', 'sb', 'so', 'za', 'gs', 'ss', 'es', 'lk', 'sd', 'sr', 'sj', 'sz', 'se', 'ch',
+            'sy', 'tw', 'tj', 'tz', 'th', 'tl', 'tg', 'tk', 'to', 'tt', 'tn', 'tr', 'tm', 'tc', 'tv', 'ug', 'ua', 'ae',
+            'gb', 'us', 'um', 'uy', 'uz', 'vu', 've', 'vn', 'vg', 'vi', 'wf', 'eh', 'ye', 'zm', 'zw',
+        ];
+    }
+
+    /**
+     * Get the path to a statically cached page
+     *
+     * @param  string $sitePath
+     * @return string
+     */
+    protected function getStaticPath($sitePath)
+    {
+        $parts = parse_url($_SERVER['REQUEST_URI']);
+        $query = isset($parts['query']) ? $parts['query'] : '';
+
+        return $sitePath . '/static' . $parts['path'] . '_' . $query . '.html';
     }
 }