From 459a4f330b369d4e9b753ee2e68c0479e1fdd31e Mon Sep 17 00:00:00 2001
From: Pacien TRAN-GIRARD
Date: Sat, 8 Sep 2012 16:19:40 +0200
Subject: Add files and dirs manipulation functions.

---
 src/ninjacloud.go | 130 ++++++++++++++++++++++++++++++++++++++++++++++++------
 1 file changed, 117 insertions(+), 13 deletions(-)

(limited to 'src')

diff --git a/src/ninjacloud.go b/src/ninjacloud.go
index 2d3c2f0..57e939a 100644
--- a/src/ninjacloud.go
+++ b/src/ninjacloud.go
@@ -1,12 +1,16 @@
 package main
 
 import (
+	"encoding/json"
+	//"errors"
 	"flag"
+	"io/ioutil"
 	"log"
 	"net/http"
-	"path"
-	//"io/ioutil"
-	//"os"
+	"os"
+	//"path"
+	"io"
+	"path/filepath"
 )
 
 const APP_NAME = "Ninja Go Local Cloud"
@@ -29,32 +33,121 @@ const statusPathLen = len(statusPath)
 
 //////// FILESYSTEM
 
+func exist(path string) bool {
+	_, err := os.Stat(path)
+	if !os.IsNotExist(err) {
+		return true
+	}
+	return false
+}
+
 //// Files
 
-func writeFile() {
+func writeFile(path string, content []byte, overwrite bool) (err error) {
+	if !overwrite {
+		if exist(path) {
+			err = os.ErrExist
+			return
+		}
+	}
+	err = ioutil.WriteFile(path, content, 0600)
+	return
+}
+
+func readFile(path string) (content []byte, err error) {
+	content, err = ioutil.ReadFile(path)
+	return
 }
 
-func readFile() {
+func removeFile(path string) (err error) {
+	err = os.Remove(path)
+	return
 }
 
-func removeFile() {
+func moveFile(source string, dest string) (err error) {
+	err = os.Rename(source, dest)
+	return
 }
 
-func copyFile() {
+func copyFile(source string, dest string) (err error) {
+	// from https://gist.github.com/2876519
+	sf, err := os.Open(source)
+	if err != nil {
+		return err
+	}
+	defer sf.Close()
+	df, err := os.Create(dest)
+	if err != nil {
+		return err
+	}
+	defer df.Close()
+	_, err = io.Copy(df, sf)
+	if err == nil {
+		si, err := os.Stat(source)
+		if err != nil {
+			err = os.Chmod(dest, si.Mode())
+		}
+
+	}
+	return
 }
 
 //// Dirs
 
-func createDir() {
+func createDir(path string) (err error) {
+	err = os.MkdirAll(path, 0600)
+	return
 }
 
-func removeDir() {
+func removeDir(path string) (err error) {
+	err = os.RemoveAll(path)
+	return
 }
 
-func listDir() {
+func listDir(path string) (list []os.FileInfo, err error) {
+	list, err = ioutil.ReadDir(path)
+	return
 }
 
-func copyDir() {
+func moveDir(source string, dest string) (err error) {
+	err = os.Rename(source, dest)
+	return
+}
+
+func copyDir(source string, dest string) (err error) {
+	// from https://gist.github.com/2876519
+	fi, err := os.Stat(source)
+	if err != nil {
+		return
+	}
+	if !fi.IsDir() {
+		return os.ErrInvalid
+	}
+	_, err = os.Open(dest)
+	if !os.IsNotExist(err) {
+		return os.ErrExist
+	}
+	err = os.MkdirAll(dest, fi.Mode())
+	if err != nil {
+		return
+	}
+	entries, err := ioutil.ReadDir(source)
+	for _, entry := range entries {
+		sfp := source + "/" + entry.Name()
+		dfp := dest + "/" + entry.Name()
+		if entry.IsDir() {
+			err = copyDir(sfp, dfp)
+			if err != nil {
+				return
+			}
+		} else {
+			err = copyFile(sfp, dfp)
+			if err != nil {
+				return
+			}
+		}
+	}
+	return
 }
 
 //////// REQUEST HANDLERS
@@ -63,7 +156,7 @@ func copyDir() {
 
 func fileHandler(w http.ResponseWriter, r *http.Request) {
 	p := r.URL.Path[filePathLen:]
-	path.Clean(p)
+	filepath.Clean(p)
 
 	switch r.Method {
 	case "POST":
@@ -86,7 +179,7 @@ func fileHandler(w http.ResponseWriter, r *http.Request) {
 
 func dirHandler(w http.ResponseWriter, r *http.Request) {
 	p := r.URL.Path[dirPathLen:]
-	path.Clean(p)
+	filepath.Clean(p)
 
 	switch r.Method {
 	case "POST":
@@ -111,6 +204,17 @@ func getDataHandler(w http.ResponseWriter, r *http.Request) {
 
 // Get the cloud status JSON
 func getStatusHandler(w http.ResponseWriter, r *http.Request) {
+	cloudStatus := map[string]string{
+		"name":        APP_NAME,
+		"version":     APP_VERSION,
+		"server-root": rootFlag,
+		"status":      "running",
+	}
+	j, err := json.Marshal(cloudStatus)
+	if err != nil {
+		log.Println(err)
+	}
+	w.Write(j)
 }
 
 //////// INIT and MAIN
-- 
cgit v1.2.3