aboutsummaryrefslogtreecommitdiff
path: root/S3.go
diff options
context:
space:
mode:
Diffstat (limited to 'S3.go')
-rw-r--r--S3.go216
1 files changed, 216 insertions, 0 deletions
diff --git a/S3.go b/S3.go
new file mode 100644
index 0000000..8dd87b4
--- /dev/null
+++ b/S3.go
@@ -0,0 +1,216 @@
+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"
+)
+
+const PROFILE_PICTURE_FIELD_NAME = "profilePicture"
+
+//Upload image through guichet server.
+func uploadImage(w http.ResponseWriter, r *http.Request, login *LoginStatus) (string, error) {
+ file, _, err := r.FormFile("image")
+
+ if err == http.ErrMissingFile {
+ return "", nil
+ }
+ if err != nil {
+ return "", err
+ }
+ defer file.Close()
+ err = checkImage(file)
+ if err != nil {
+ return "", err
+ }
+
+ buff := bytes.NewBuffer([]byte{})
+ buff_thumbnail := bytes.NewBuffer([]byte{})
+ err = resizeThumb(file, buff, buff_thumbnail)
+ if err != nil {
+ return "", err
+ }
+
+ mc, err := newMinioClient()
+ if err != nil || mc == nil {
+ return "", err
+ }
+
+ var name, nameFull string
+
+ if nameConsul := login.UserEntry.GetAttributeValue(PROFILE_PICTURE_FIELD_NAME); nameConsul != "" {
+ name = nameConsul
+ nameFull = "full_" + name
+ } else {
+ name = uuid.New().String() + ".jpeg"
+ nameFull = "full_" + name
+ }
+
+ _, err = mc.PutObject(context.Background(), config.S3_Bucket, name, buff_thumbnail, int64(buff_thumbnail.Len()), minio.PutObjectOptions{
+ ContentType: "image/jpeg",
+ })
+ if err != nil {
+ return "", err
+ }
+
+ _, err = mc.PutObject(context.Background(), config.S3_Bucket, nameFull, buff, int64(buff.Len()), minio.PutObjectOptions{
+ ContentType: "image/jpeg",
+ })
+ if err != nil {
+ return "", err
+ }
+
+ return name, nil
+}
+
+func newMinioClient() (*minio.Client, error) {
+ endpoint := config.S3_Endpoint
+ accessKeyID := config.S3_AccesKey
+ secretKeyID := config.S3_SecretKey
+ useSSL := true
+
+ //Initialize Minio
+ minioCLient, err := minio.New(endpoint, &minio.Options{
+ Creds: credentials.NewStaticV4(accessKeyID, secretKeyID, ""),
+ Secure: useSSL,
+ Region: config.S3_Region,
+ })
+
+ if err != nil {
+ return nil, err
+ }
+
+ return minioCLient, nil
+
+}
+
+func checkImage(file multipart.File) 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 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 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{PROFILE_PICTURE_FIELD_NAME},
+ 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(PROFILE_PICTURE_FIELD_NAME)
+ if imageName == "" {
+ http.Error(w, "User doesn't have profile image", http.StatusNotFound)
+ return
+ }
+ } else {
+ imageName = "unknown_profile.jpg"
+ }
+
+ if size == "full" {
+ imageName = "full_" + imageName
+ }
+ //Get the object after connect MC
+ mc, err := newMinioClient()
+ 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, "MiniObjet: "+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)))
+ //Copy obj in w
+ writting, err := io.Copy(w, obj)
+ if writting != objStat.Size || err != nil {
+ http.Error(w, fmt.Sprintf("WriteBody: %s, bytes wrote %d on %d", err.Error(), writting, objStat.Size), http.StatusInternalServerError)
+ return
+ }
+
+}