gvguichet/directory.go

123 lines
2.9 KiB
Go
Raw Normal View History

2021-07-21 21:21:46 +02:00
package main
import (
"html/template"
"net/http"
2021-08-16 16:27:20 +02:00
"sort"
2021-07-21 21:21:46 +02:00
"strings"
"github.com/go-ldap/ldap/v3"
)
const FIELD_NAME_PROFILE_PICTURE = "profilePicture"
const FIELD_NAME_DIRECTORY_VISIBILITY = "directoryVisibility"
2021-07-21 21:21:46 +02:00
func handleDirectory(w http.ResponseWriter, r *http.Request) {
templateDirectory := template.Must(template.ParseFiles("templates/layout.html", "templates/directory.html"))
login := checkLogin(w, r)
if login == nil {
return
}
templateDirectory.Execute(w, nil)
}
type SearchResult struct {
2021-08-16 16:28:42 +02:00
DN string
Id string
DisplayName string
Email string
Description string
2021-08-16 16:27:20 +02:00
ProfilePicture string
2021-07-21 21:21:46 +02:00
}
2021-08-16 16:27:20 +02:00
type SearchResults struct {
Results []SearchResult
}
2021-07-21 21:21:46 +02:00
2021-08-16 16:27:20 +02:00
func handleDirectorySearch(w http.ResponseWriter, r *http.Request) {
templateDirectoryResults := template.Must(template.ParseFiles("templates/directory_results.html"))
2021-07-21 21:21:46 +02:00
//Get input value by user
2021-08-16 16:27:20 +02:00
r.ParseMultipartForm(1024)
input := strings.TrimSpace(strings.Join(r.Form["query"], ""))
if r.Method != "POST" {
2021-08-16 16:27:20 +02:00
http.Error(w, "Invalid request", http.StatusBadRequest)
return
}
2021-07-21 21:21:46 +02:00
//Log to allow the research
login := checkLogin(w, r)
if login == nil {
2021-08-16 16:27:20 +02:00
http.Error(w, "Login required", http.StatusUnauthorized)
2021-07-21 21:21:46 +02:00
return
}
//Search values with ldap and filter
2021-07-21 21:21:46 +02:00
searchRequest := ldap.NewSearchRequest(
config.UserBaseDN,
ldap.ScopeSingleLevel, ldap.NeverDerefAliases, 0, 0, false,
"(&(objectclass=organizationalPerson)("+FIELD_NAME_DIRECTORY_VISIBILITY+"=on))",
[]string{
config.UserNameAttr,
"displayname",
"mail",
"description",
FIELD_NAME_PROFILE_PICTURE,
},
2021-07-21 21:21:46 +02:00
nil)
sr, err := login.conn.Search(searchRequest)
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
//Transform the researh's result in a correct struct to send JSON
2021-08-16 16:27:20 +02:00
results := []SearchResult{}
2021-07-21 21:21:46 +02:00
2021-08-16 16:27:20 +02:00
for _, values := range sr.Entries {
if input == "" ||
ContainsI(values.GetAttributeValue(config.UserNameAttr), input) ||
2021-08-16 16:27:20 +02:00
ContainsI(values.GetAttributeValue("displayname"), input) ||
ContainsI(values.GetAttributeValue("mail"), input) {
results = append(results, SearchResult{
2021-08-16 16:28:42 +02:00
DN: values.DN,
Id: values.GetAttributeValue(config.UserNameAttr),
DisplayName: values.GetAttributeValue("displayname"),
Email: values.GetAttributeValue("mail"),
Description: values.GetAttributeValue("description"),
2021-08-16 16:27:20 +02:00
ProfilePicture: values.GetAttributeValue(FIELD_NAME_PROFILE_PICTURE),
})
}
}
2021-08-16 16:27:20 +02:00
search_results := SearchResults{
Results: results,
}
2021-08-16 16:27:20 +02:00
sort.Sort(&search_results)
2021-08-16 16:27:20 +02:00
templateDirectoryResults.Execute(w, search_results)
}
func ContainsI(a string, b string) bool {
return strings.Contains(
strings.ToLower(a),
strings.ToLower(b),
)
}
func (r *SearchResults) Len() int {
return len(r.Results)
}
func (r *SearchResults) Less(i, j int) bool {
return r.Results[i].Id < r.Results[j].Id
}
func (r *SearchResults) Swap(i, j int) {
r.Results[i], r.Results[j] = r.Results[j], r.Results[i]
2021-07-21 21:21:46 +02:00
}