diff options
Diffstat (limited to 'mimioClient.go')
| -rw-r--r-- | mimioClient.go | 244 |
1 files changed, 244 insertions, 0 deletions
diff --git a/mimioClient.go b/mimioClient.go new file mode 100644 index 0000000..a3cfdfc --- /dev/null +++ b/mimioClient.go @@ -0,0 +1,244 @@ +package main + +import ( + "bytes" + "context" + "errors" + "fmt" + "io" + "strconv" + + "image" + "image/jpeg" + _ "image/png" + + "mime/multipart" + "net/http" + "strings" + + "github.com/go-ldap/ldap/v3" + "github.com/google/uuid" + "github.com/gorilla/mux" + "github.com/minio/minio-go/v7" + "github.com/minio/minio-go/v7/pkg/credentials" + "github.com/nfnt/resize" +) + +//Upload image through guichet server. +func uploadImage(w http.ResponseWriter, r *http.Request, login *LoginStatus) (bool, string, error) { + file, _, err := r.FormFile("image") + + if err == http.ErrMissingFile { + return false, "", nil + } + if err != nil { + return false, "", err + } + defer file.Close() + fileType, err := checkImage(file) + if err != nil { + return false, "", err + } + if fileType == "" { + return false, "", nil + } + + buff := bytes.NewBuffer([]byte{}) + buff_thumbnail := bytes.NewBuffer([]byte{}) + err = resizeThumb(file, buff, buff_thumbnail) + if err != nil { + return false, "", err + } + + mc, err := newMimioClient() + if err != nil { + return false, "", err + } + if mc == nil { + return false, "", err + } + + var name, nameFull string + + if nameConsul := login.UserEntry.GetAttributeValue("profilImage"); nameConsul != "" { + name = nameConsul + nameFull = "full_" + name + } else { + name = uuid.New().String() + ".jpeg" + nameFull = "full_" + name + } + + _, err = mc.PutObject(context.Background(), "bottin-pictures", name, buff_thumbnail, int64(buff_thumbnail.Len()), minio.PutObjectOptions{ + ContentType: "image/jpeg", + }) + if err != nil { + return false, "", err + } + + _, err = mc.PutObject(context.Background(), "bottin-pictures", nameFull, buff, int64(buff.Len()), minio.PutObjectOptions{ + ContentType: "image/jpeg", + }) + if err != nil { + return false, "", err + } + + return true, name, nil +} + +func newMimioClient() (*minio.Client, error) { + endpoint := config.Endpoint + accessKeyID := config.AccesKey + secretKeyID := config.SecretKey + useSSL := true + + //Initialize Minio + minioCLient, err := minio.New(endpoint, &minio.Options{ + Creds: credentials.NewStaticV4(accessKeyID, secretKeyID, ""), + Secure: useSSL, + Region: "garage", + }) + + if err != nil { + return nil, err + } + + return minioCLient, nil + +} + +func checkImage(file multipart.File) (string, error) { + buff := make([]byte, 512) //Detect read only the first 512 bytes + _, err := file.Read(buff) + if err != nil { + return "", err + } + file.Seek(0, 0) + + fileType := http.DetectContentType(buff) + fileType = strings.Split(fileType, "/")[0] + switch fileType { + case "image": + return fileType, nil + default: + return "", errors.New("bad type") + } + +} + +func resizeThumb(file multipart.File, buff, buff_thumbnail *bytes.Buffer) error { + file.Seek(0, 0) + images, _, err := image.Decode(file) + if err != nil { + return errors.New("Decode: " + err.Error()) + } + //Encode image to jpeg a first time to eliminate all problems + err = jpeg.Encode(buff, images, &jpeg.Options{ + Quality: 100, //Between 1 to 100, higher is better + }) + if err != nil { + return err + } + images, _, err = image.Decode(buff) + if err != nil { + return err + } + buff.Reset() + images = resize.Thumbnail(200, 200, images, resize.Lanczos3) + images_thumbnail := resize.Thumbnail(80, 80, images, resize.Lanczos3) + + err = jpeg.Encode(buff, images, &jpeg.Options{ + Quality: 95, + }) + if err != nil { + return err + } + + err = jpeg.Encode(buff_thumbnail, images_thumbnail, &jpeg.Options{ + Quality: 95, + }) + + return err +} + +func handleDownloadImage(w http.ResponseWriter, r *http.Request) { + //Get input value by user + dn := mux.Vars(r)["name"] + size := mux.Vars(r)["size"] + + //Check login + login := checkLogin(w, r) + if login == nil { + return + } + var imageName string + if dn != "unknown_profile" { + //Search values with ldap and filter + searchRequest := ldap.NewSearchRequest( + dn, + ldap.ScopeBaseObject, ldap.NeverDerefAliases, 0, 0, false, + "(objectclass=*)", + []string{"profilImage"}, + nil) + + sr, err := login.conn.Search(searchRequest) + if err != nil { + http.Error(w, "Search: "+err.Error(), http.StatusInternalServerError) + return + } + if len(sr.Entries) != 1 { + http.Error(w, fmt.Sprintf("Not found user: %s cn: %s and numberEntries: %d", dn, strings.Split(dn, ",")[0], len(sr.Entries)), http.StatusInternalServerError) + return + } + imageName = sr.Entries[0].GetAttributeValue("profilImage") + if imageName == "" { + http.Error(w, "User doesn't have profile image", http.StatusInternalServerError) + return + } + } else { + imageName = "unknown_profile.jpg" + } + + if size == "full" { + imageName = "full_" + imageName + } + + //Get the object after connect MC + mc, err := newMimioClient() + if err != nil { + http.Error(w, "MinioClient: "+err.Error(), http.StatusInternalServerError) + return + } + obj, err := mc.GetObject(context.Background(), "bottin-pictures", imageName, minio.GetObjectOptions{}) + if err != nil { + http.Error(w, "MinioClient: GetObject: "+err.Error(), http.StatusInternalServerError) + return + } + defer obj.Close() + objStat, err := obj.Stat() + if err != nil { + http.Error(w, "MinioObjet: "+err.Error(), http.StatusInternalServerError) + return + } + + //Send JSON through xhttp + w.Header().Set("Content-Type", objStat.ContentType) + w.Header().Set("Content-Length", strconv.Itoa(int(objStat.Size))) + //http.Error(w, fmt.Sprintf("Length buffer: %d", objStat.Size), http.StatusInternalServerError) + buff := make([]byte, objStat.Size) + + obj.Seek(0, 0) + n, err := obj.Read(buff) + if err != nil && err != io.EOF { + http.Error(w, fmt.Sprintf("Read Error: %s, bytes Read: %d, bytes in file: %d", err.Error(), n, objStat.Size), http.StatusInternalServerError) + return + } + if int64(n) != objStat.Size { + http.Error(w, fmt.Sprintf("Read %d bytes on %d bytes", n, objStat.Size), http.StatusInternalServerError) + return + } + + if _, err := w.Write(buff); err != nil { + http.Error(w, "WriteBody: "+err.Error(), http.StatusInternalServerError) + return + } +} |