Setting folder permissions

3 files changed, 11 insertions, 2 deletions

diff --git a/httracker b/httracker
index ae86d1f..de05fac 100755
--- a/httracker
+++ b/httracker
@@ -10,7 +10,7 @@ source config || exit 1
 source lib/httracker/functions || exit 1
 
 # Create folders
-mkdir -p $MIRRORS $TMP
+httracker_setup_folders
 
 # Get URLs from feed
 # Thanks http://stackoverflow.com/questions/443991/how-to-parse-rss-feeds-xml-in-a-shell-script
diff --git a/lib/httracker/functions b/lib/httracker/functions
index fa52ed4..bb3dc07 100644
--- a/lib/httracker/functions
+++ b/lib/httracker/functions
@@ -79,3 +79,12 @@ function httracker_iterate {
     httracker_get "$link"
   done
 }
+
+function httracker_setup_folders {
+  mkdir -p $MIRRORS $TMP
+  chown -R $USER.$GROUP $TMP
+
+  if [ "`whoami`" == "root" ]; then
+    chown -R $USER.$GROUP $TMP
+  fi
+}
diff --git a/scuttler b/scuttler
index 1136852..b74ddaa 100755
--- a/scuttler
+++ b/scuttler
@@ -10,7 +10,7 @@ source config || exit 1
 source lib/httracker/functions || exit 1
 
 # Create folders
-mkdir -p $MIRRORS $TMP
+httracker_setup_folders
 
 # Other parameters
 BASEDIR=/var/sites/links