Merge branch '0.9.x' into activityexport

[quix0rs-gnu-social.git] / actions / doc.php

diff --git a/actions/doc.php b/actions/doc.php
index 99c2c796653590c943607fbc13b8f5df63783da3..f876fb8beb7df2474f7c0a1c43700de9002d431c 100644 (file)
--- a/actions/doc.php
+++ b/actions/doc.php
@@ -13,7 +13,7 @@
  * @link     http://status.net/
  *
  * StatusNet - the distributed open-source microblogging tool
- * Copyright (C) 2008, 2009, StatusNet, Inc.
+ * Copyright (C) 2008-2010, StatusNet, Inc.
  *
  * This program is free software: you can redistribute it and/or modify
  * it under the terms of the GNU Affero General Public License as published by
@@ -51,7 +51,12 @@ class DocAction extends Action
 
     function prepare($args)
     {
+        parent::prepare($args);
+
         $this->title  = $this->trimmed('title');
+        if (!preg_match('/^[a-zA-Z0-9_-]*$/', $this->title)) {
+            $this->title = 'help';
+        }
         $this->output = null;
 
         $this->loadDoc();
@@ -163,25 +168,62 @@ class DocAction extends Action
 
     function getFilename()
     {
-        $local = array_merge(glob(INSTALLDIR.'/local/doc-src/'.$this->title),
-                             glob(INSTALLDIR.'/local/doc-src/'.$this->title.'.*'));
+        $localDef = null;
+        $local    = null;
+
+        $site = StatusNet::currentSite();
+
+        if (!empty($site) && file_exists(INSTALLDIR.'/local/doc-src/'.$site.'/'.$this->title)) {
+            $localDef = INSTALLDIR.'/local/doc-src/'.$site.'/'.$this->title;
+
+            $local = glob(INSTALLDIR.'/local/doc-src/'.$site.'/'.$this->title.'.*');
+            if ($local === false) {
+                // Some systems return false, others array(), if dir didn't exist.
+                $local = array();
+            }
+        } else {
+            if (file_exists(INSTALLDIR.'/local/doc-src/'.$this->title)) {
+                $localDef = INSTALLDIR.'/local/doc-src/'.$this->title;
+            }
+
+            $local = glob(INSTALLDIR.'/local/doc-src/'.$this->title.'.*');
+            if ($local === false) {
+                $local = array();
+            }
+        }
 
-        if (count($local)) {
-            return $this->negotiateLanguage($local);
+        if (count($local) || isset($localDef)) {
+            return $this->negotiateLanguage($local, $localDef);
         }
 
-        $dist = array_merge(glob(INSTALLDIR.'/doc-src/'.$this->title),
-                            glob(INSTALLDIR.'/doc-src/'.$this->title.'.*'));
+        if (file_exists(INSTALLDIR.'/doc-src/'.$this->title)) {
+            $distDef = INSTALLDIR.'/doc-src/'.$this->title;
+        }
 
-        if (count($dist)) {
-            return $this->negotiateLanguage($dist);
+        $dist = glob(INSTALLDIR.'/doc-src/'.$this->title.'.*');
+        if ($dist === false) {
+            $dist = array();
+        }
+
+        if (count($dist) || isset($distDef)) {
+            return $this->negotiateLanguage($dist, $distDef);
         }
 
         return null;
     }
 
-    function negotiateLanguage($files)
+    function negotiateLanguage($filenames, $defaultFilename=null)
     {
-        // FIXME: write this
+        // XXX: do this better
+
+        $langcode = common_language();
+
+        foreach ($filenames as $filename) {
+            if (preg_match('/\.'.$langcode.'$/', $filename)) {
+                return $filename;
+            }
+        }
+
+        return $defaultFilename;
     }
 }