func Upload(c *middleware.Context) { c.Req.ParseMultipartForm(32 << 20) file, _, err := c.Req.FormFile("uploadfile") if err != nil { fmt.Println(err) return } defer file.Close() name := uuid.NewV4().String() //.fmt.Fprintf(w, "%v", handler.Header) err = os.MkdirAll("data/uploads/"+makePaths(name), 0777) if err != nil { return } f, err := os.OpenFile("data/uploads/"+makePaths(name)+"/"+name, os.O_WRONLY|os.O_CREATE, 0777) if err != nil { fmt.Println(err) return } defer f.Close() io.Copy(f, file) c.JSON(200, makePaths(name)) }
func screen(siteId int64) (string, error) { site, e := models.GetSite(siteId) if e != nil { return "", e } rdr, e := shot.Shot(site.Domain) if e != nil { return "", e } name := uuid.NewV4().String() //.fmt.Fprintf(w, "%v", handler.Header) e = os.MkdirAll("data/uploads/"+makePaths(name), 0777) if e != nil { return "", e } f, e := os.OpenFile("data/uploads/"+makePaths(name)+"/"+name, os.O_WRONLY|os.O_CREATE, 0777) if e != nil { return "", e } defer f.Close() _, e = io.Copy(f, rdr) if e != nil { return "", e } fp := filepath.Join("/", makePaths(name), "/", name) site.PreviewUrl = fp e = models.SaveSite(site) if e != nil { return "", e } return fp, nil }