refactor: repo queries and add endpoint query

Author: ItsNotGoodName

config/config.go

@@ -272,7 +272,7 @@ func (p Parser) Parse(raw Raw) (Config, error) {
 			if value.BodyTemplate != nil {
 				bodyTemplate = *value.BodyTemplate
 			}
-			e, err := endpoint.NewInternal(endpointFactory, endpoint.CreateEndpoint{
+			e, err := endpoint.NewInternal(endpointFactory, models.DTOEndpointCreate{
 				Name:              value.Name,
 				AttachmentDisable: value.AttachmentDisable,
 				TextDisable:       value.TextDisable,

internal/endpoint/endpoint.go

@@ -25,20 +25,10 @@ func validate(f Factory, end models.Endpoint) error {
 	return nil
 }
 
-type CreateEndpoint struct {
-	Name              string
-	AttachmentDisable bool
-	TextDisable       bool
-	TitleTemplate     string
-	BodyTemplate      string
-	Kind              string
-	Config            models.EndpointConfig
-}
-
 const DefaultTitleTemplate = "{{ .Message.Subject }}"
 const DefaultBodyTemplate = "{{ .Message.Text }}"
 
-func new(f Factory, r CreateEndpoint) models.Endpoint {
+func new(f Factory, r models.DTOEndpointCreate) models.Endpoint {
 	return models.Endpoint{
 		Internal:          false,
 		InternalID:        sql.NullString{},
@@ -52,12 +42,12 @@ func new(f Factory, r CreateEndpoint) models.Endpoint {
 	}
 }
 
-func New(f Factory, r CreateEndpoint) (models.Endpoint, error) {
+func New(f Factory, r models.DTOEndpointCreate) (models.Endpoint, error) {
 	end := new(f, r)
 	return end, validate(f, end)
 }
 
-func NewInternal(f Factory, r CreateEndpoint, internalID string) (models.Endpoint, error) {
+func NewInternal(f Factory, r models.DTOEndpointCreate, internalID string) (models.Endpoint, error) {
 	if r.Name == "" {
 		r.Name = internalID
 	}

internal/models/dto.go

@@ -84,3 +84,24 @@ type DTORuleUpdate struct {
 	Enable     *bool
 	Endpoints  *[]int64
 }
+
+type DTOEndpointCreate struct {
+	Name              string
+	AttachmentDisable bool
+	TextDisable       bool
+	TitleTemplate     string
+	BodyTemplate      string
+	Kind              string
+	Config            EndpointConfig
+}
+
+type DTOEndpointUpdate struct {
+	ID                int64
+	Name              *string
+	AttachmentDisable *bool
+	TextDisable       *bool
+	TitleTemplate     *string
+	BodyTemplate      *string
+	Kind              *string
+	Config            *EndpointConfig
+}

internal/repo/endpoint.go

@@ -2,6 +2,7 @@ package repo
 
 import (
 	"context"
+	"time"
 
 	"github.com/ItsNotGoodName/smtpbridge/internal/database"
 	. "github.com/ItsNotGoodName/smtpbridge/internal/jet/table"
@@ -22,6 +23,39 @@ var endpointPJ ProjectionList = ProjectionList{
 	Endpoints.Config.AS("endpoint.config"),
 }
 
+func EndpointCreate(ctx context.Context, db database.Querier, end models.Endpoint) (int64, error) {
+	now := models.NewTime(time.Now())
+	m := struct {
+		models.Endpoint
+		CreatedAt models.Time
+		UpdatedAt models.Time
+	}{
+		Endpoint:  end,
+		CreatedAt: now,
+		UpdatedAt: now,
+	}
+	res, err := Endpoints.
+		INSERT(
+			Endpoints.Internal,
+			Endpoints.InternalID,
+			Endpoints.Name,
+			Endpoints.AttachmentDisable,
+			Endpoints.TextDisable,
+			Endpoints.TitleTemplate,
+			Endpoints.BodyTemplate,
+			Endpoints.Kind,
+			Endpoints.Config,
+			Endpoints.UpdatedAt,
+			Endpoints.CreatedAt,
+		).
+		MODEL(m).
+		ExecContext(ctx, db)
+	if err != nil {
+		return 0, err
+	}
+	return res.LastInsertId()
+}
+
 func EndpointGet(ctx context.Context, db database.Querier, id int64) (models.Endpoint, error) {
 	var endpoint models.Endpoint
 	err := Endpoints.
@@ -40,6 +74,33 @@ func EndpointList(ctx context.Context, db database.Querier) ([]models.Endpoint,
 	return endpoints, err
 }
 
+func EndpointUpdate(ctx context.Context, db database.Querier, end models.Endpoint) error {
+	m := struct {
+		models.Endpoint
+		UpdatedAt models.Time
+	}{
+		Endpoint:  end,
+		UpdatedAt: models.NewTime(time.Now()),
+	}
+	_, err := Endpoints.
+		UPDATE(
+			Endpoints.Internal,
+			Endpoints.InternalID,
+			Endpoints.Name,
+			Endpoints.AttachmentDisable,
+			Endpoints.TextDisable,
+			Endpoints.TitleTemplate,
+			Endpoints.BodyTemplate,
+			Endpoints.Kind,
+			Endpoints.Config,
+			Endpoints.UpdatedAt,
+		).
+		MODEL(m).
+		WHERE(Endpoints.ID.EQ(Int64(end.ID))).
+		ExecContext(ctx, db)
+	return err
+}
+
 func EndpointDelete(ctx context.Context, db database.Querier, id int64) error {
 	res, err := Endpoints.
 		DELETE().