feat(api): create cover letter
This commit is contained in:
@@ -1,11 +1,13 @@
|
||||
package cover
|
||||
|
||||
import (
|
||||
"backend/models/cover"
|
||||
"backend/models/template"
|
||||
"backend/utils"
|
||||
"backend/utils/chatgpt"
|
||||
"backend/utils/jwt"
|
||||
res "backend/utils/responses"
|
||||
"fmt"
|
||||
"net/http"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
@@ -57,7 +59,32 @@ func Post(c *gin.Context) {
|
||||
return
|
||||
}
|
||||
|
||||
res.Success(c, generatedCover)
|
||||
// Find cover name that doesnt exist
|
||||
coverName := generatedCover.Name
|
||||
for i := 1; true; i++ {
|
||||
// Try to find cover with same name
|
||||
covers, err := cover.Get("name = $1 AND user_id = $2", coverName, user.Id)
|
||||
if err != nil {
|
||||
res.Error(c, err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
// Found non existent name
|
||||
if len(covers) == 0 {
|
||||
break
|
||||
}
|
||||
|
||||
// Change number on name
|
||||
coverName = fmt.Sprintf("%s (%d)", generatedCover.Name, i)
|
||||
}
|
||||
|
||||
// Save name in database
|
||||
if err := cover.Create(coverName, generatedCover.Cover, user.Id); err != nil {
|
||||
res.Error(c, err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
res.Success(c, gin.H{"message": "Successfully created " + coverName})
|
||||
}
|
||||
|
||||
func Put(c *gin.Context) {
|
||||
|
||||
@@ -22,7 +22,7 @@ CREATE TABLE IF NOT EXISTS "templates" (
|
||||
CREATE TABLE IF NOT EXISTS "cover_letters" (
|
||||
"id" SERIAL PRIMARY KEY,
|
||||
"user_id" INTEGER NOT NULL,
|
||||
"name" VARCHAR(50) NOT NULL UNIQUE,
|
||||
"name" TEXT NOT NULL UNIQUE,
|
||||
"letter" TEXT NOT NULL,
|
||||
"created_at" TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||
|
||||
|
||||
63
backend/models/cover/cover.go
Normal file
63
backend/models/cover/cover.go
Normal file
@@ -0,0 +1,63 @@
|
||||
package cover
|
||||
|
||||
import (
|
||||
"backend/db"
|
||||
"context"
|
||||
"time"
|
||||
)
|
||||
|
||||
type Template struct {
|
||||
ID int `json:"id"`
|
||||
UserID int `json:"user_id"`
|
||||
Name string `json:"name"`
|
||||
Letter string `json:"letter"`
|
||||
CreatedAt time.Time `json:"created_at"`
|
||||
}
|
||||
|
||||
func Get(where string, args ...any) ([]Template, error) {
|
||||
// Create timeout context
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
|
||||
defer cancel()
|
||||
|
||||
// Build query and execute
|
||||
query := `SELECT * FROM cover_letters`
|
||||
if where != "" {
|
||||
query += " WHERE " + where
|
||||
}
|
||||
|
||||
rows, err := db.Pool.Query(ctx, query, args...)
|
||||
// Query executes query instantly, and returns error instantly
|
||||
// Not like QueryRow, which executes query only on row.Scan
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
// need to tell database to close the rows connection
|
||||
// and free up resources
|
||||
defer rows.Close()
|
||||
|
||||
// Prepeare results now
|
||||
var results []Template
|
||||
for rows.Next() {
|
||||
var t Template
|
||||
if err := rows.Scan(&t.ID, &t.UserID, &t.Name, &t.Letter, &t.CreatedAt); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
results = append(results, t)
|
||||
}
|
||||
|
||||
return results, nil
|
||||
}
|
||||
|
||||
func Create(name string, letter string, userId float64) error {
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
|
||||
defer cancel()
|
||||
|
||||
// build query
|
||||
query := `INSERT INTO cover_letters (name, letter, user_id) VALUES ($1, $2, $3)`
|
||||
|
||||
// execute query
|
||||
_, err := db.Pool.Exec(ctx, query, name, letter, userId)
|
||||
|
||||
return err
|
||||
}
|
||||
Reference in New Issue
Block a user