4
1
Fork 0
backend-core/contact.go

117 lines
3.2 KiB
Go

package main
import (
"encoding/json"
"log"
"net/http"
"github.com/julienschmidt/httprouter"
)
func (h *Handler) CreateContact(w http.ResponseWriter, r *http.Request, p httprouter.Params) {
// Parse
userID := r.Context().Value("user").(string)
contactPhone := PhoneNumber{}
decoder := json.NewDecoder(r.Body)
err := decoder.Decode(&contactPhone)
if err != nil {
http.Error(w, http.StatusText(http.StatusBadRequest), http.StatusBadRequest)
return
}
// Validate
phone, err := ParsePhone(contactPhone.PhoneNumber)
if err != nil || len(contactPhone.PhoneNumber) < 1 {
http.Error(w, http.StatusText(http.StatusBadRequest), http.StatusBadRequest)
return
}
// Generate ID (just in case)
id := "u-" + RandomHex()
// Create contact if not exists, returning the id regardless
contact := User{}
err = h.db.QueryRow(`
INSERT INTO "user" (id, username, bio, profile_pic, first_name, last_name, phone_number)
VALUES ($1, '', '', '', '', '', $2)
ON CONFLICT(phone_number)
DO UPDATE SET phone_number=EXCLUDED.phone_number
RETURNING id, username, bio, profile_pic, first_name, last_name, phone_number
`, id, phone).Scan(&contact.ID, &contact.Username, &contact.Bio, &contact.ProfilePic, &contact.FirstName, &contact.LastName, &contact.PhoneNumber)
if err != nil {
http.Error(w, http.StatusText(http.StatusInternalServerError), http.StatusInternalServerError)
log.Print(err)
return
}
// Insert
_, err = h.db.Exec(`
INSERT INTO contact ("user", contact) VALUES ($1, $2)
`, userID, contact.ID)
if err != nil {
http.Error(w, http.StatusText(http.StatusInternalServerError), http.StatusInternalServerError)
log.Print(err)
return
}
// Publish NATs
if h.nc != nil {
contact := Contact{
UserA: userID,
UserB: contact.ID,
}
contactString, err := json.Marshal(&contact)
if err == nil {
updateMsg := UpdateMsg{
Type: "add",
Data: string(contactString),
}
updateMsgString, err := json.Marshal(&updateMsg)
if err == nil {
h.nc.Publish("contact", updateMsgString)
}
}
}
// Respond
w.WriteHeader(200)
w.Header().Set("Content-Type", "application/json")
json.NewEncoder(w).Encode(contact)
}
func (h *Handler) GetContacts(w http.ResponseWriter, r *http.Request, p httprouter.Params) {
// Parse
userID := r.Context().Value("user").(string)
// Response object
contacts := make([]User, 0)
// Select
rows, err := h.db.Query(`
SELECT id, username, bio, profile_pic, first_name, last_name, phone_number FROM "user"
INNER JOIN contact
ON contact.contact = "user".id AND contact.user = $1
`, userID)
if err != nil {
http.Error(w, http.StatusText(http.StatusInternalServerError), http.StatusInternalServerError)
log.Print(err)
return
}
defer rows.Close()
// Scan
for rows.Next() {
contact := User{}
if err := rows.Scan(&contact.ID, &contact.Username, &contact.Bio, &contact.ProfilePic, &contact.FirstName, &contact.LastName, &contact.PhoneNumber); err != nil {
http.Error(w, http.StatusText(http.StatusInternalServerError), http.StatusInternalServerError)
log.Print(err)
return
}
contacts = append(contacts, contact)
}
// Respond
w.Header().Set("Content-Type", "application/json")
json.NewEncoder(w).Encode(contacts)
}