mirror of
https://github.com/phishingclub/phishingclub.git
synced 2026-02-12 16:12:44 +00:00
109 lines
3.2 KiB
Go
109 lines
3.2 KiB
Go
package model
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/go-errors/errors"
|
|
"github.com/google/uuid"
|
|
"github.com/oapi-codegen/nullable"
|
|
"github.com/phishingclub/phishingclub/validate"
|
|
"github.com/phishingclub/phishingclub/vo"
|
|
)
|
|
|
|
// Proxy is a proxy configuration
|
|
type Proxy struct {
|
|
ID nullable.Nullable[uuid.UUID] `json:"id"`
|
|
CreatedAt *time.Time `json:"createdAt"`
|
|
UpdatedAt *time.Time `json:"updatedAt"`
|
|
CompanyID nullable.Nullable[uuid.UUID] `json:"companyID"`
|
|
Name nullable.Nullable[vo.String64] `json:"name"`
|
|
Description nullable.Nullable[vo.OptionalString1024] `json:"description"`
|
|
StartURL nullable.Nullable[vo.String1024] `json:"startURL"`
|
|
ProxyConfig nullable.Nullable[vo.String1MB] `json:"proxyConfig"`
|
|
|
|
Company *Company `json:"-"`
|
|
}
|
|
|
|
// Validate checks if the Proxy has a valid state
|
|
func (m *Proxy) Validate() error {
|
|
if err := validate.NullableFieldRequired("name", m.Name); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := validate.NullableFieldRequired("startURL", m.StartURL); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := validate.NullableFieldRequired("proxyConfig", m.ProxyConfig); err != nil {
|
|
return err
|
|
}
|
|
|
|
// validate start URL format
|
|
startURL, err := m.StartURL.Get()
|
|
if err != nil {
|
|
return validate.WrapErrorWithField(errors.New("start URL is required"), "startURL")
|
|
}
|
|
|
|
startURLStr := startURL.String()
|
|
if startURLStr == "" {
|
|
return validate.WrapErrorWithField(errors.New("start URL cannot be empty"), "startURL")
|
|
}
|
|
|
|
// validate that start URL is a valid, full URL
|
|
if err := validate.ErrorIfInvalidURL(startURLStr); err != nil {
|
|
return validate.WrapErrorWithField(err, "startURL")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// ToDBMap converts the fields that can be stored or updated to a map
|
|
// if the value is nullable and not set, it is not included
|
|
// if the value is nullable and set, it is included, if it is null, it is set to nil
|
|
func (m *Proxy) ToDBMap() map[string]any {
|
|
dbMap := map[string]any{}
|
|
if m.Name.IsSpecified() {
|
|
dbMap["name"] = nil
|
|
if name, err := m.Name.Get(); err == nil {
|
|
dbMap["name"] = name.String()
|
|
}
|
|
}
|
|
if m.Description.IsSpecified() {
|
|
dbMap["description"] = nil
|
|
if description, err := m.Description.Get(); err == nil {
|
|
dbMap["description"] = description.String()
|
|
}
|
|
}
|
|
if m.StartURL.IsSpecified() {
|
|
dbMap["start_url"] = nil
|
|
if startURL, err := m.StartURL.Get(); err == nil {
|
|
dbMap["start_url"] = startURL.String()
|
|
}
|
|
}
|
|
if m.ProxyConfig.IsSpecified() {
|
|
dbMap["proxy_config"] = nil
|
|
if proxyConfig, err := m.ProxyConfig.Get(); err == nil {
|
|
dbMap["proxy_config"] = proxyConfig.String()
|
|
}
|
|
}
|
|
if m.CompanyID.IsSpecified() {
|
|
if m.CompanyID.IsNull() {
|
|
dbMap["company_id"] = nil
|
|
} else {
|
|
dbMap["company_id"] = m.CompanyID.MustGet()
|
|
}
|
|
}
|
|
return dbMap
|
|
}
|
|
|
|
// ProxyOverview is a subset of the Proxy as used as read-only
|
|
type ProxyOverview struct {
|
|
ID uuid.UUID `json:"id,omitempty"`
|
|
CreatedAt *time.Time `json:"createdAt"`
|
|
UpdatedAt *time.Time `json:"updatedAt"`
|
|
Name string `json:"name"`
|
|
Description string `json:"description"`
|
|
StartURL string `json:"startURL"`
|
|
CompanyID *uuid.UUID `json:"companyID"`
|
|
}
|