package api import ( "bufio" "crypto/sha512" "encoding/base32" "encoding/json" "errors" "log" "net/http" "os" "path" "strings" "srs.epita.fr/fic-server/libfic" ) var CloudDAVBase string var CloudUsername string var CloudPassword string func createExerciceFile(theme fic.Theme, exercice fic.Exercice, args map[string]string, body []byte) (interface{}, error) { var uf map[string]string if err := json.Unmarshal(body, &uf); err != nil { return nil, err } var hash [sha512.Size]byte var logStr string var fromURI string var getFile func(string) error if URI, ok := uf["URI"]; ok { hash = sha512.Sum512([]byte(URI)) logStr = "Import file from Cloud: " + URI + " =>" fromURI = URI getFile = func(dest string) error { return getCloudFile(URI, dest) } } else if path, ok := uf["path"]; ok { hash = sha512.Sum512([]byte(path)) logStr = "Import file from local FS: " + path + " =>" fromURI = path getFile = func(dest string) error { return os.Symlink(path, dest) } } else { return nil, errors.New("URI or path not filled") } pathname := path.Join(fic.FilesDir, strings.ToLower(base32.StdEncoding.EncodeToString(hash[:])), path.Base(fromURI)) if _, err := os.Stat(pathname); os.IsNotExist(err) { log.Println(logStr, pathname) if err := os.MkdirAll(path.Dir(pathname), 0777); err != nil { return nil, err } else if err := getFile(pathname); err != nil { return nil, err } } return exercice.ImportFile(pathname, fromURI) } func getCloudFile(pathname string, dest string) error { client := http.Client{} if req, err := http.NewRequest("GET", CloudDAVBase+pathname, nil); err != nil { return err } else { req.SetBasicAuth(CloudUsername, CloudPassword) if resp, err := client.Do(req); err != nil { return err } else { defer resp.Body.Close() if fd, err := os.Create(dest); err != nil { return err } else { defer fd.Close() if resp.StatusCode != http.StatusOK { return errors.New(resp.Status) } else { writer := bufio.NewWriter(fd) reader := bufio.NewReader(resp.Body) reader.WriteTo(writer) writer.Flush() } } } } return nil }