1 files changed, 31 insertions, 0 deletions
diff --git a/www/functions/func_rewrite.php b/www/functions/func_rewrite.php
new file mode 100644
index 0000000..5ca2171
--- /dev/null
+++ b/www/functions/func_rewrite.php
@@ -0,0 +1,31 @@
+<?
+function rewrite($db){
+
+        /* test if first argument a username or folder */
+
+        $name = SQLite3::escapeString("$_GET[name]");
+
+        $name_id_db = $db->query("SELECT id FROM user WHERE name='" . $name . "';");
+        $name_id_ar = $name_id_db->fetchArray(SQLITE3_NUM);
+
+        if(empty($name_id_ar)){
+
+                /* if first argument wasn't valid, rewrite the URL to include the username */
+
+                if($_SESSION["login"]){
+                        header("Refresh: 0; /" . $_SESSION[username] . "/" . $_GET["name"] . "/" . $_GET["folder"] . "");
+                        exit;
+                } else {
+
+                        $wrong_folder = $_GET["name"];
+                        $working_path = $_GET["name"];
+
+                        get_404($working_path, $wrong_folder);
+			return false;
+                }
+                return false;
+        }
+
+        return true;
+}
+