src/server.go

   1 //
   2 // Armadillo File Manager
   3 // Copyright (c) 2010, Robert Sesek <http://www.bluestatic.org>
   4 //
   5 // This program is free software: you can redistribute it and/or modify it under
   6 // the terms of the GNU General Public License as published by the Free Software
   7 // Foundation, either version 3 of the License, or any later version.
   8 //
   9
  10 package server
  11
  12 import (
  13   "fmt"
  14   "http"
  15   "io"
  16   "json"
  17   "os"
  18   "path"
  19   "strings"
  20   "./paths"
  21 )
  22
  23 var dir, file = path.Split(path.Clean(os.Getenv("_")))
  24 var kFrontEndFiles string = path.Join(dir, "fe")
  25
  26 func indexHandler(connection *http.Conn, request *http.Request) {
  27   fd, err := os.Open(path.Join(kFrontEndFiles, "index.html"), os.O_RDONLY, 0)
  28   if err != nil {
  29     fmt.Print("Error opening file ", err.String(), "\n")
  30     return
  31   }
  32   io.Copy(connection, fd)
  33 }
  34
  35 func serviceHandler(connection *http.Conn, request *http.Request) {
  36   if request.Method != "POST" {
  37     io.WriteString(connection, "Error: Not a POST request")
  38     return
  39   }
  40
  41   switch request.FormValue("action") {
  42     case "list":
  43       files, err := paths.List(request.FormValue("path"))
  44       if err != nil {
  45         errorResponse(connection, err.String())
  46       } else {
  47         okResponse(connection, files)
  48       }
  49     case "remove":
  50       err := paths.Remove(request.FormValue("path"))
  51       if err != nil {
  52         errorResponse(connection, err.String())
  53       } else {
  54         response := map[string] int {
  55           "error" : 0,
  56         }
  57         okResponse(connection, response)
  58       }
  59     case "move":
  60       source := request.FormValue("source")
  61       target := request.FormValue("target")
  62       err := paths.Move(source, target)
  63       if err != nil {
  64         errorResponse(connection, err.String())
  65       } else {
  66         response := map[string] string {
  67           "path" : target,
  68         }
  69         okResponse(connection, response)
  70       }
  71     default:
  72       errorResponse(connection, "Unhandled action")
  73   }
  74 }
  75
  76 func errorResponse(connection *http.Conn, message string) {
  77   message = strings.Replace(message, paths.JailRoot, "/", -1)
  78   response := map[string] string {
  79     "error"   : "-1",
  80     "message" : message,
  81   }
  82   json_data, err := json.Marshal(response)
  83
  84   connection.SetHeader("Content-Type", "text/json")
  85   if err != nil {
  86     io.WriteString(connection, "{\"error\":\"-9\",\"message\":\"Internal encoding error\"}")
  87   } else {
  88     connection.Write(json_data)
  89   }
  90 }
  91
  92 func okResponse(connection *http.Conn, data interface{}) {
  93   connection.SetHeader("Content-Type", "text/json")
  94   json_data, err := json.Marshal(data)
  95   if err != nil {
  96     errorResponse(connection, "Internal encoding error")
  97   } else {
  98     connection.Write(json_data)
  99   }
 100 }
 101
 102 func RunFrontEnd(port int) {
 103   mux := http.NewServeMux()
 104   mux.HandleFunc("/", indexHandler)
 105   mux.Handle("/fe/", http.FileServer(kFrontEndFiles, "/fe/"))
 106   mux.HandleFunc("/service", serviceHandler)
 107
 108   error := http.ListenAndServe(fmt.Sprintf(":%d", port), mux)
 109   fmt.Printf("error %v", error)
 110 }