feat: 插件更新管理

This commit is contained in:
yokowu
2025-07-04 20:17:49 +08:00
parent 01245897fe
commit 3a1affd88e
29 changed files with 3053 additions and 35 deletions

View File

@@ -12,15 +12,17 @@ import (
"github.com/chaitin/MonkeyCode/backend/db"
"github.com/chaitin/MonkeyCode/backend/domain"
v1_5 "github.com/chaitin/MonkeyCode/backend/internal/billing/handler/http/v1"
repo6 "github.com/chaitin/MonkeyCode/backend/internal/billing/repo"
usecase5 "github.com/chaitin/MonkeyCode/backend/internal/billing/usecase"
repo7 "github.com/chaitin/MonkeyCode/backend/internal/billing/repo"
usecase6 "github.com/chaitin/MonkeyCode/backend/internal/billing/usecase"
v1_4 "github.com/chaitin/MonkeyCode/backend/internal/dashboard/handler/v1"
repo5 "github.com/chaitin/MonkeyCode/backend/internal/dashboard/repo"
usecase4 "github.com/chaitin/MonkeyCode/backend/internal/dashboard/usecase"
repo6 "github.com/chaitin/MonkeyCode/backend/internal/dashboard/repo"
usecase5 "github.com/chaitin/MonkeyCode/backend/internal/dashboard/usecase"
repo3 "github.com/chaitin/MonkeyCode/backend/internal/extension/repo"
usecase2 "github.com/chaitin/MonkeyCode/backend/internal/extension/usecase"
"github.com/chaitin/MonkeyCode/backend/internal/middleware"
v1_2 "github.com/chaitin/MonkeyCode/backend/internal/model/handler/http/v1"
repo3 "github.com/chaitin/MonkeyCode/backend/internal/model/repo"
usecase2 "github.com/chaitin/MonkeyCode/backend/internal/model/usecase"
repo4 "github.com/chaitin/MonkeyCode/backend/internal/model/repo"
usecase3 "github.com/chaitin/MonkeyCode/backend/internal/model/usecase"
"github.com/chaitin/MonkeyCode/backend/internal/openai/handler/v1"
repo2 "github.com/chaitin/MonkeyCode/backend/internal/openai/repo"
"github.com/chaitin/MonkeyCode/backend/internal/openai/usecase"
@@ -28,8 +30,8 @@ import (
"github.com/chaitin/MonkeyCode/backend/internal/proxy/repo"
"github.com/chaitin/MonkeyCode/backend/internal/proxy/usecase"
v1_3 "github.com/chaitin/MonkeyCode/backend/internal/user/handler/v1"
repo4 "github.com/chaitin/MonkeyCode/backend/internal/user/repo"
usecase3 "github.com/chaitin/MonkeyCode/backend/internal/user/usecase"
repo5 "github.com/chaitin/MonkeyCode/backend/internal/user/repo"
usecase4 "github.com/chaitin/MonkeyCode/backend/internal/user/usecase"
"github.com/chaitin/MonkeyCode/backend/pkg"
"github.com/chaitin/MonkeyCode/backend/pkg/logger"
"github.com/chaitin/MonkeyCode/backend/pkg/session"
@@ -56,23 +58,25 @@ func newServer(dir string) (*Server, error) {
domainProxy := proxy.NewLLMProxy(proxyUsecase, configConfig, slogLogger)
openAIRepo := repo2.NewOpenAIRepo(client)
openAIUsecase := openai.NewOpenAIUsecase(configConfig, openAIRepo, slogLogger)
extensionRepo := repo3.NewExtensionRepo(client)
extensionUsecase := usecase2.NewExtensionUsecase(extensionRepo, configConfig, slogLogger)
proxyMiddleware := middleware.NewProxyMiddleware(proxyUsecase)
redisClient := store.NewRedisCli(configConfig)
activeMiddleware := middleware.NewActiveMiddleware(redisClient, slogLogger)
v1Handler := v1.NewV1Handler(slogLogger, web, domainProxy, openAIUsecase, proxyMiddleware, activeMiddleware)
modelRepo := repo3.NewModelRepo(client)
modelUsecase := usecase2.NewModelUsecase(slogLogger, modelRepo, configConfig)
v1Handler := v1.NewV1Handler(slogLogger, web, domainProxy, openAIUsecase, extensionUsecase, proxyMiddleware, activeMiddleware, configConfig)
modelRepo := repo4.NewModelRepo(client)
modelUsecase := usecase3.NewModelUsecase(slogLogger, modelRepo, configConfig)
sessionSession := session.NewSession(configConfig)
authMiddleware := middleware.NewAuthMiddleware(sessionSession, slogLogger)
modelHandler := v1_2.NewModelHandler(web, modelUsecase, authMiddleware, slogLogger)
userRepo := repo4.NewUserRepo(client)
userUsecase := usecase3.NewUserUsecase(configConfig, redisClient, userRepo, slogLogger)
userHandler := v1_3.NewUserHandler(web, userUsecase, authMiddleware, sessionSession, slogLogger, configConfig)
dashboardRepo := repo5.NewDashboardRepo(client)
dashboardUsecase := usecase4.NewDashboardUsecase(dashboardRepo)
userRepo := repo5.NewUserRepo(client)
userUsecase := usecase4.NewUserUsecase(configConfig, redisClient, userRepo, slogLogger)
userHandler := v1_3.NewUserHandler(web, userUsecase, extensionUsecase, authMiddleware, sessionSession, slogLogger, configConfig)
dashboardRepo := repo6.NewDashboardRepo(client)
dashboardUsecase := usecase5.NewDashboardUsecase(dashboardRepo)
dashboardHandler := v1_4.NewDashboardHandler(web, dashboardUsecase, authMiddleware)
billingRepo := repo6.NewBillingRepo(client)
billingUsecase := usecase5.NewBillingUsecase(billingRepo)
billingRepo := repo7.NewBillingRepo(client)
billingUsecase := usecase6.NewBillingUsecase(billingRepo)
billingHandler := v1_5.NewBillingHandler(web, billingUsecase, authMiddleware)
server := &Server{
config: configConfig,

View File

@@ -57,15 +57,15 @@ type Config struct {
RequestLogPath string `mapstructure:"request_log_path"`
} `mapstructure:"llm_proxy"`
VSCode struct {
VSIXFile string `mapstructure:"vsix_file"`
} `mapstructure:"vscode"`
InitModel struct {
ModelName string `mapstructure:"model_name"`
ModelKey string `mapstructure:"model_key"`
ModelURL string `mapstructure:"model_url"`
} `mapstructure:"init_model"`
Extension struct {
Baseurl string `mapstructure:"baseurl"`
} `mapstructure:"extension"`
}
func Init(dir string) (*Config, error) {

View File

@@ -23,6 +23,7 @@ import (
"github.com/chaitin/MonkeyCode/backend/db/billingquota"
"github.com/chaitin/MonkeyCode/backend/db/billingrecord"
"github.com/chaitin/MonkeyCode/backend/db/billingusage"
"github.com/chaitin/MonkeyCode/backend/db/extension"
"github.com/chaitin/MonkeyCode/backend/db/invitecode"
"github.com/chaitin/MonkeyCode/backend/db/model"
"github.com/chaitin/MonkeyCode/backend/db/modelprovider"
@@ -56,6 +57,8 @@ type Client struct {
BillingRecord *BillingRecordClient
// BillingUsage is the client for interacting with the BillingUsage builders.
BillingUsage *BillingUsageClient
// Extension is the client for interacting with the Extension builders.
Extension *ExtensionClient
// InviteCode is the client for interacting with the InviteCode builders.
InviteCode *InviteCodeClient
// Model is the client for interacting with the Model builders.
@@ -94,6 +97,7 @@ func (c *Client) init() {
c.BillingQuota = NewBillingQuotaClient(c.config)
c.BillingRecord = NewBillingRecordClient(c.config)
c.BillingUsage = NewBillingUsageClient(c.config)
c.Extension = NewExtensionClient(c.config)
c.InviteCode = NewInviteCodeClient(c.config)
c.Model = NewModelClient(c.config)
c.ModelProvider = NewModelProviderClient(c.config)
@@ -203,6 +207,7 @@ func (c *Client) Tx(ctx context.Context) (*Tx, error) {
BillingQuota: NewBillingQuotaClient(cfg),
BillingRecord: NewBillingRecordClient(cfg),
BillingUsage: NewBillingUsageClient(cfg),
Extension: NewExtensionClient(cfg),
InviteCode: NewInviteCodeClient(cfg),
Model: NewModelClient(cfg),
ModelProvider: NewModelProviderClient(cfg),
@@ -239,6 +244,7 @@ func (c *Client) BeginTx(ctx context.Context, opts *sql.TxOptions) (*Tx, error)
BillingQuota: NewBillingQuotaClient(cfg),
BillingRecord: NewBillingRecordClient(cfg),
BillingUsage: NewBillingUsageClient(cfg),
Extension: NewExtensionClient(cfg),
InviteCode: NewInviteCodeClient(cfg),
Model: NewModelClient(cfg),
ModelProvider: NewModelProviderClient(cfg),
@@ -279,9 +285,9 @@ func (c *Client) Close() error {
func (c *Client) Use(hooks ...Hook) {
for _, n := range []interface{ Use(...Hook) }{
c.Admin, c.AdminLoginHistory, c.ApiKey, c.BillingPlan, c.BillingQuota,
c.BillingRecord, c.BillingUsage, c.InviteCode, c.Model, c.ModelProvider,
c.ModelProviderModel, c.Setting, c.Task, c.TaskRecord, c.User, c.UserIdentity,
c.UserLoginHistory,
c.BillingRecord, c.BillingUsage, c.Extension, c.InviteCode, c.Model,
c.ModelProvider, c.ModelProviderModel, c.Setting, c.Task, c.TaskRecord, c.User,
c.UserIdentity, c.UserLoginHistory,
} {
n.Use(hooks...)
}
@@ -292,9 +298,9 @@ func (c *Client) Use(hooks ...Hook) {
func (c *Client) Intercept(interceptors ...Interceptor) {
for _, n := range []interface{ Intercept(...Interceptor) }{
c.Admin, c.AdminLoginHistory, c.ApiKey, c.BillingPlan, c.BillingQuota,
c.BillingRecord, c.BillingUsage, c.InviteCode, c.Model, c.ModelProvider,
c.ModelProviderModel, c.Setting, c.Task, c.TaskRecord, c.User, c.UserIdentity,
c.UserLoginHistory,
c.BillingRecord, c.BillingUsage, c.Extension, c.InviteCode, c.Model,
c.ModelProvider, c.ModelProviderModel, c.Setting, c.Task, c.TaskRecord, c.User,
c.UserIdentity, c.UserLoginHistory,
} {
n.Intercept(interceptors...)
}
@@ -317,6 +323,8 @@ func (c *Client) Mutate(ctx context.Context, m Mutation) (Value, error) {
return c.BillingRecord.mutate(ctx, m)
case *BillingUsageMutation:
return c.BillingUsage.mutate(ctx, m)
case *ExtensionMutation:
return c.Extension.mutate(ctx, m)
case *InviteCodeMutation:
return c.InviteCode.mutate(ctx, m)
case *ModelMutation:
@@ -1309,6 +1317,139 @@ func (c *BillingUsageClient) mutate(ctx context.Context, m *BillingUsageMutation
}
}
// ExtensionClient is a client for the Extension schema.
type ExtensionClient struct {
config
}
// NewExtensionClient returns a client for the Extension from the given config.
func NewExtensionClient(c config) *ExtensionClient {
return &ExtensionClient{config: c}
}
// Use adds a list of mutation hooks to the hooks stack.
// A call to `Use(f, g, h)` equals to `extension.Hooks(f(g(h())))`.
func (c *ExtensionClient) Use(hooks ...Hook) {
c.hooks.Extension = append(c.hooks.Extension, hooks...)
}
// Intercept adds a list of query interceptors to the interceptors stack.
// A call to `Intercept(f, g, h)` equals to `extension.Intercept(f(g(h())))`.
func (c *ExtensionClient) Intercept(interceptors ...Interceptor) {
c.inters.Extension = append(c.inters.Extension, interceptors...)
}
// Create returns a builder for creating a Extension entity.
func (c *ExtensionClient) Create() *ExtensionCreate {
mutation := newExtensionMutation(c.config, OpCreate)
return &ExtensionCreate{config: c.config, hooks: c.Hooks(), mutation: mutation}
}
// CreateBulk returns a builder for creating a bulk of Extension entities.
func (c *ExtensionClient) CreateBulk(builders ...*ExtensionCreate) *ExtensionCreateBulk {
return &ExtensionCreateBulk{config: c.config, builders: builders}
}
// MapCreateBulk creates a bulk creation builder from the given slice. For each item in the slice, the function creates
// a builder and applies setFunc on it.
func (c *ExtensionClient) MapCreateBulk(slice any, setFunc func(*ExtensionCreate, int)) *ExtensionCreateBulk {
rv := reflect.ValueOf(slice)
if rv.Kind() != reflect.Slice {
return &ExtensionCreateBulk{err: fmt.Errorf("calling to ExtensionClient.MapCreateBulk with wrong type %T, need slice", slice)}
}
builders := make([]*ExtensionCreate, rv.Len())
for i := 0; i < rv.Len(); i++ {
builders[i] = c.Create()
setFunc(builders[i], i)
}
return &ExtensionCreateBulk{config: c.config, builders: builders}
}
// Update returns an update builder for Extension.
func (c *ExtensionClient) Update() *ExtensionUpdate {
mutation := newExtensionMutation(c.config, OpUpdate)
return &ExtensionUpdate{config: c.config, hooks: c.Hooks(), mutation: mutation}
}
// UpdateOne returns an update builder for the given entity.
func (c *ExtensionClient) UpdateOne(e *Extension) *ExtensionUpdateOne {
mutation := newExtensionMutation(c.config, OpUpdateOne, withExtension(e))
return &ExtensionUpdateOne{config: c.config, hooks: c.Hooks(), mutation: mutation}
}
// UpdateOneID returns an update builder for the given id.
func (c *ExtensionClient) UpdateOneID(id uuid.UUID) *ExtensionUpdateOne {
mutation := newExtensionMutation(c.config, OpUpdateOne, withExtensionID(id))
return &ExtensionUpdateOne{config: c.config, hooks: c.Hooks(), mutation: mutation}
}
// Delete returns a delete builder for Extension.
func (c *ExtensionClient) Delete() *ExtensionDelete {
mutation := newExtensionMutation(c.config, OpDelete)
return &ExtensionDelete{config: c.config, hooks: c.Hooks(), mutation: mutation}
}
// DeleteOne returns a builder for deleting the given entity.
func (c *ExtensionClient) DeleteOne(e *Extension) *ExtensionDeleteOne {
return c.DeleteOneID(e.ID)
}
// DeleteOneID returns a builder for deleting the given entity by its id.
func (c *ExtensionClient) DeleteOneID(id uuid.UUID) *ExtensionDeleteOne {
builder := c.Delete().Where(extension.ID(id))
builder.mutation.id = &id
builder.mutation.op = OpDeleteOne
return &ExtensionDeleteOne{builder}
}
// Query returns a query builder for Extension.
func (c *ExtensionClient) Query() *ExtensionQuery {
return &ExtensionQuery{
config: c.config,
ctx: &QueryContext{Type: TypeExtension},
inters: c.Interceptors(),
}
}
// Get returns a Extension entity by its id.
func (c *ExtensionClient) Get(ctx context.Context, id uuid.UUID) (*Extension, error) {
return c.Query().Where(extension.ID(id)).Only(ctx)
}
// GetX is like Get, but panics if an error occurs.
func (c *ExtensionClient) GetX(ctx context.Context, id uuid.UUID) *Extension {
obj, err := c.Get(ctx, id)
if err != nil {
panic(err)
}
return obj
}
// Hooks returns the client hooks.
func (c *ExtensionClient) Hooks() []Hook {
return c.hooks.Extension
}
// Interceptors returns the client interceptors.
func (c *ExtensionClient) Interceptors() []Interceptor {
return c.inters.Extension
}
func (c *ExtensionClient) mutate(ctx context.Context, m *ExtensionMutation) (Value, error) {
switch m.Op() {
case OpCreate:
return (&ExtensionCreate{config: c.config, hooks: c.Hooks(), mutation: m}).Save(ctx)
case OpUpdate:
return (&ExtensionUpdate{config: c.config, hooks: c.Hooks(), mutation: m}).Save(ctx)
case OpUpdateOne:
return (&ExtensionUpdateOne{config: c.config, hooks: c.Hooks(), mutation: m}).Save(ctx)
case OpDelete, OpDeleteOne:
return (&ExtensionDelete{config: c.config, hooks: c.Hooks(), mutation: m}).Exec(ctx)
default:
return nil, fmt.Errorf("db: unknown Extension mutation op: %q", m.Op())
}
}
// InviteCodeClient is a client for the InviteCode schema.
type InviteCodeClient struct {
config
@@ -2871,13 +3012,14 @@ func (c *UserLoginHistoryClient) mutate(ctx context.Context, m *UserLoginHistory
type (
hooks struct {
Admin, AdminLoginHistory, ApiKey, BillingPlan, BillingQuota, BillingRecord,
BillingUsage, InviteCode, Model, ModelProvider, ModelProviderModel, Setting,
Task, TaskRecord, User, UserIdentity, UserLoginHistory []ent.Hook
BillingUsage, Extension, InviteCode, Model, ModelProvider, ModelProviderModel,
Setting, Task, TaskRecord, User, UserIdentity, UserLoginHistory []ent.Hook
}
inters struct {
Admin, AdminLoginHistory, ApiKey, BillingPlan, BillingQuota, BillingRecord,
BillingUsage, InviteCode, Model, ModelProvider, ModelProviderModel, Setting,
Task, TaskRecord, User, UserIdentity, UserLoginHistory []ent.Interceptor
BillingUsage, Extension, InviteCode, Model, ModelProvider, ModelProviderModel,
Setting, Task, TaskRecord, User, UserIdentity,
UserLoginHistory []ent.Interceptor
}
)

View File

@@ -19,6 +19,7 @@ import (
"github.com/chaitin/MonkeyCode/backend/db/billingquota"
"github.com/chaitin/MonkeyCode/backend/db/billingrecord"
"github.com/chaitin/MonkeyCode/backend/db/billingusage"
"github.com/chaitin/MonkeyCode/backend/db/extension"
"github.com/chaitin/MonkeyCode/backend/db/invitecode"
"github.com/chaitin/MonkeyCode/backend/db/model"
"github.com/chaitin/MonkeyCode/backend/db/modelprovider"
@@ -96,6 +97,7 @@ func checkColumn(table, column string) error {
billingquota.Table: billingquota.ValidColumn,
billingrecord.Table: billingrecord.ValidColumn,
billingusage.Table: billingusage.ValidColumn,
extension.Table: extension.ValidColumn,
invitecode.Table: invitecode.ValidColumn,
model.Table: model.ValidColumn,
modelprovider.Table: modelprovider.ValidColumn,

129
backend/db/extension.go Normal file
View File

@@ -0,0 +1,129 @@
// Code generated by ent, DO NOT EDIT.
package db
import (
"fmt"
"strings"
"time"
"entgo.io/ent"
"entgo.io/ent/dialect/sql"
"github.com/chaitin/MonkeyCode/backend/db/extension"
"github.com/google/uuid"
)
// Extension is the model entity for the Extension schema.
type Extension struct {
config `json:"-"`
// ID of the ent.
ID uuid.UUID `json:"id,omitempty"`
// Version holds the value of the "version" field.
Version string `json:"version,omitempty"`
// Path holds the value of the "path" field.
Path string `json:"path,omitempty"`
// CreatedAt holds the value of the "created_at" field.
CreatedAt time.Time `json:"created_at,omitempty"`
selectValues sql.SelectValues
}
// scanValues returns the types for scanning values from sql.Rows.
func (*Extension) scanValues(columns []string) ([]any, error) {
values := make([]any, len(columns))
for i := range columns {
switch columns[i] {
case extension.FieldVersion, extension.FieldPath:
values[i] = new(sql.NullString)
case extension.FieldCreatedAt:
values[i] = new(sql.NullTime)
case extension.FieldID:
values[i] = new(uuid.UUID)
default:
values[i] = new(sql.UnknownType)
}
}
return values, nil
}
// assignValues assigns the values that were returned from sql.Rows (after scanning)
// to the Extension fields.
func (e *Extension) assignValues(columns []string, values []any) error {
if m, n := len(values), len(columns); m < n {
return fmt.Errorf("mismatch number of scan values: %d != %d", m, n)
}
for i := range columns {
switch columns[i] {
case extension.FieldID:
if value, ok := values[i].(*uuid.UUID); !ok {
return fmt.Errorf("unexpected type %T for field id", values[i])
} else if value != nil {
e.ID = *value
}
case extension.FieldVersion:
if value, ok := values[i].(*sql.NullString); !ok {
return fmt.Errorf("unexpected type %T for field version", values[i])
} else if value.Valid {
e.Version = value.String
}
case extension.FieldPath:
if value, ok := values[i].(*sql.NullString); !ok {
return fmt.Errorf("unexpected type %T for field path", values[i])
} else if value.Valid {
e.Path = value.String
}
case extension.FieldCreatedAt:
if value, ok := values[i].(*sql.NullTime); !ok {
return fmt.Errorf("unexpected type %T for field created_at", values[i])
} else if value.Valid {
e.CreatedAt = value.Time
}
default:
e.selectValues.Set(columns[i], values[i])
}
}
return nil
}
// Value returns the ent.Value that was dynamically selected and assigned to the Extension.
// This includes values selected through modifiers, order, etc.
func (e *Extension) Value(name string) (ent.Value, error) {
return e.selectValues.Get(name)
}
// Update returns a builder for updating this Extension.
// Note that you need to call Extension.Unwrap() before calling this method if this Extension
// was returned from a transaction, and the transaction was committed or rolled back.
func (e *Extension) Update() *ExtensionUpdateOne {
return NewExtensionClient(e.config).UpdateOne(e)
}
// Unwrap unwraps the Extension entity that was returned from a transaction after it was closed,
// so that all future queries will be executed through the driver which created the transaction.
func (e *Extension) Unwrap() *Extension {
_tx, ok := e.config.driver.(*txDriver)
if !ok {
panic("db: Extension is not a transactional entity")
}
e.config.driver = _tx.drv
return e
}
// String implements the fmt.Stringer.
func (e *Extension) String() string {
var builder strings.Builder
builder.WriteString("Extension(")
builder.WriteString(fmt.Sprintf("id=%v, ", e.ID))
builder.WriteString("version=")
builder.WriteString(e.Version)
builder.WriteString(", ")
builder.WriteString("path=")
builder.WriteString(e.Path)
builder.WriteString(", ")
builder.WriteString("created_at=")
builder.WriteString(e.CreatedAt.Format(time.ANSIC))
builder.WriteByte(')')
return builder.String()
}
// Extensions is a parsable slice of Extension.
type Extensions []*Extension

View File

@@ -0,0 +1,70 @@
// Code generated by ent, DO NOT EDIT.
package extension
import (
"time"
"entgo.io/ent/dialect/sql"
)
const (
// Label holds the string label denoting the extension type in the database.
Label = "extension"
// FieldID holds the string denoting the id field in the database.
FieldID = "id"
// FieldVersion holds the string denoting the version field in the database.
FieldVersion = "version"
// FieldPath holds the string denoting the path field in the database.
FieldPath = "path"
// FieldCreatedAt holds the string denoting the created_at field in the database.
FieldCreatedAt = "created_at"
// Table holds the table name of the extension in the database.
Table = "extensions"
)
// Columns holds all SQL columns for extension fields.
var Columns = []string{
FieldID,
FieldVersion,
FieldPath,
FieldCreatedAt,
}
// ValidColumn reports if the column name is valid (part of the table columns).
func ValidColumn(column string) bool {
for i := range Columns {
if column == Columns[i] {
return true
}
}
return false
}
var (
// DefaultCreatedAt holds the default value on creation for the "created_at" field.
DefaultCreatedAt func() time.Time
)
// OrderOption defines the ordering options for the Extension queries.
type OrderOption func(*sql.Selector)
// ByID orders the results by the id field.
func ByID(opts ...sql.OrderTermOption) OrderOption {
return sql.OrderByField(FieldID, opts...).ToFunc()
}
// ByVersion orders the results by the version field.
func ByVersion(opts ...sql.OrderTermOption) OrderOption {
return sql.OrderByField(FieldVersion, opts...).ToFunc()
}
// ByPath orders the results by the path field.
func ByPath(opts ...sql.OrderTermOption) OrderOption {
return sql.OrderByField(FieldPath, opts...).ToFunc()
}
// ByCreatedAt orders the results by the created_at field.
func ByCreatedAt(opts ...sql.OrderTermOption) OrderOption {
return sql.OrderByField(FieldCreatedAt, opts...).ToFunc()
}

View File

@@ -0,0 +1,256 @@
// Code generated by ent, DO NOT EDIT.
package extension
import (
"time"
"entgo.io/ent/dialect/sql"
"github.com/chaitin/MonkeyCode/backend/db/predicate"
"github.com/google/uuid"
)
// ID filters vertices based on their ID field.
func ID(id uuid.UUID) predicate.Extension {
return predicate.Extension(sql.FieldEQ(FieldID, id))
}
// IDEQ applies the EQ predicate on the ID field.
func IDEQ(id uuid.UUID) predicate.Extension {
return predicate.Extension(sql.FieldEQ(FieldID, id))
}
// IDNEQ applies the NEQ predicate on the ID field.
func IDNEQ(id uuid.UUID) predicate.Extension {
return predicate.Extension(sql.FieldNEQ(FieldID, id))
}
// IDIn applies the In predicate on the ID field.
func IDIn(ids ...uuid.UUID) predicate.Extension {
return predicate.Extension(sql.FieldIn(FieldID, ids...))
}
// IDNotIn applies the NotIn predicate on the ID field.
func IDNotIn(ids ...uuid.UUID) predicate.Extension {
return predicate.Extension(sql.FieldNotIn(FieldID, ids...))
}
// IDGT applies the GT predicate on the ID field.
func IDGT(id uuid.UUID) predicate.Extension {
return predicate.Extension(sql.FieldGT(FieldID, id))
}
// IDGTE applies the GTE predicate on the ID field.
func IDGTE(id uuid.UUID) predicate.Extension {
return predicate.Extension(sql.FieldGTE(FieldID, id))
}
// IDLT applies the LT predicate on the ID field.
func IDLT(id uuid.UUID) predicate.Extension {
return predicate.Extension(sql.FieldLT(FieldID, id))
}
// IDLTE applies the LTE predicate on the ID field.
func IDLTE(id uuid.UUID) predicate.Extension {
return predicate.Extension(sql.FieldLTE(FieldID, id))
}
// Version applies equality check predicate on the "version" field. It's identical to VersionEQ.
func Version(v string) predicate.Extension {
return predicate.Extension(sql.FieldEQ(FieldVersion, v))
}
// Path applies equality check predicate on the "path" field. It's identical to PathEQ.
func Path(v string) predicate.Extension {
return predicate.Extension(sql.FieldEQ(FieldPath, v))
}
// CreatedAt applies equality check predicate on the "created_at" field. It's identical to CreatedAtEQ.
func CreatedAt(v time.Time) predicate.Extension {
return predicate.Extension(sql.FieldEQ(FieldCreatedAt, v))
}
// VersionEQ applies the EQ predicate on the "version" field.
func VersionEQ(v string) predicate.Extension {
return predicate.Extension(sql.FieldEQ(FieldVersion, v))
}
// VersionNEQ applies the NEQ predicate on the "version" field.
func VersionNEQ(v string) predicate.Extension {
return predicate.Extension(sql.FieldNEQ(FieldVersion, v))
}
// VersionIn applies the In predicate on the "version" field.
func VersionIn(vs ...string) predicate.Extension {
return predicate.Extension(sql.FieldIn(FieldVersion, vs...))
}
// VersionNotIn applies the NotIn predicate on the "version" field.
func VersionNotIn(vs ...string) predicate.Extension {
return predicate.Extension(sql.FieldNotIn(FieldVersion, vs...))
}
// VersionGT applies the GT predicate on the "version" field.
func VersionGT(v string) predicate.Extension {
return predicate.Extension(sql.FieldGT(FieldVersion, v))
}
// VersionGTE applies the GTE predicate on the "version" field.
func VersionGTE(v string) predicate.Extension {
return predicate.Extension(sql.FieldGTE(FieldVersion, v))
}
// VersionLT applies the LT predicate on the "version" field.
func VersionLT(v string) predicate.Extension {
return predicate.Extension(sql.FieldLT(FieldVersion, v))
}
// VersionLTE applies the LTE predicate on the "version" field.
func VersionLTE(v string) predicate.Extension {
return predicate.Extension(sql.FieldLTE(FieldVersion, v))
}
// VersionContains applies the Contains predicate on the "version" field.
func VersionContains(v string) predicate.Extension {
return predicate.Extension(sql.FieldContains(FieldVersion, v))
}
// VersionHasPrefix applies the HasPrefix predicate on the "version" field.
func VersionHasPrefix(v string) predicate.Extension {
return predicate.Extension(sql.FieldHasPrefix(FieldVersion, v))
}
// VersionHasSuffix applies the HasSuffix predicate on the "version" field.
func VersionHasSuffix(v string) predicate.Extension {
return predicate.Extension(sql.FieldHasSuffix(FieldVersion, v))
}
// VersionEqualFold applies the EqualFold predicate on the "version" field.
func VersionEqualFold(v string) predicate.Extension {
return predicate.Extension(sql.FieldEqualFold(FieldVersion, v))
}
// VersionContainsFold applies the ContainsFold predicate on the "version" field.
func VersionContainsFold(v string) predicate.Extension {
return predicate.Extension(sql.FieldContainsFold(FieldVersion, v))
}
// PathEQ applies the EQ predicate on the "path" field.
func PathEQ(v string) predicate.Extension {
return predicate.Extension(sql.FieldEQ(FieldPath, v))
}
// PathNEQ applies the NEQ predicate on the "path" field.
func PathNEQ(v string) predicate.Extension {
return predicate.Extension(sql.FieldNEQ(FieldPath, v))
}
// PathIn applies the In predicate on the "path" field.
func PathIn(vs ...string) predicate.Extension {
return predicate.Extension(sql.FieldIn(FieldPath, vs...))
}
// PathNotIn applies the NotIn predicate on the "path" field.
func PathNotIn(vs ...string) predicate.Extension {
return predicate.Extension(sql.FieldNotIn(FieldPath, vs...))
}
// PathGT applies the GT predicate on the "path" field.
func PathGT(v string) predicate.Extension {
return predicate.Extension(sql.FieldGT(FieldPath, v))
}
// PathGTE applies the GTE predicate on the "path" field.
func PathGTE(v string) predicate.Extension {
return predicate.Extension(sql.FieldGTE(FieldPath, v))
}
// PathLT applies the LT predicate on the "path" field.
func PathLT(v string) predicate.Extension {
return predicate.Extension(sql.FieldLT(FieldPath, v))
}
// PathLTE applies the LTE predicate on the "path" field.
func PathLTE(v string) predicate.Extension {
return predicate.Extension(sql.FieldLTE(FieldPath, v))
}
// PathContains applies the Contains predicate on the "path" field.
func PathContains(v string) predicate.Extension {
return predicate.Extension(sql.FieldContains(FieldPath, v))
}
// PathHasPrefix applies the HasPrefix predicate on the "path" field.
func PathHasPrefix(v string) predicate.Extension {
return predicate.Extension(sql.FieldHasPrefix(FieldPath, v))
}
// PathHasSuffix applies the HasSuffix predicate on the "path" field.
func PathHasSuffix(v string) predicate.Extension {
return predicate.Extension(sql.FieldHasSuffix(FieldPath, v))
}
// PathEqualFold applies the EqualFold predicate on the "path" field.
func PathEqualFold(v string) predicate.Extension {
return predicate.Extension(sql.FieldEqualFold(FieldPath, v))
}
// PathContainsFold applies the ContainsFold predicate on the "path" field.
func PathContainsFold(v string) predicate.Extension {
return predicate.Extension(sql.FieldContainsFold(FieldPath, v))
}
// CreatedAtEQ applies the EQ predicate on the "created_at" field.
func CreatedAtEQ(v time.Time) predicate.Extension {
return predicate.Extension(sql.FieldEQ(FieldCreatedAt, v))
}
// CreatedAtNEQ applies the NEQ predicate on the "created_at" field.
func CreatedAtNEQ(v time.Time) predicate.Extension {
return predicate.Extension(sql.FieldNEQ(FieldCreatedAt, v))
}
// CreatedAtIn applies the In predicate on the "created_at" field.
func CreatedAtIn(vs ...time.Time) predicate.Extension {
return predicate.Extension(sql.FieldIn(FieldCreatedAt, vs...))
}
// CreatedAtNotIn applies the NotIn predicate on the "created_at" field.
func CreatedAtNotIn(vs ...time.Time) predicate.Extension {
return predicate.Extension(sql.FieldNotIn(FieldCreatedAt, vs...))
}
// CreatedAtGT applies the GT predicate on the "created_at" field.
func CreatedAtGT(v time.Time) predicate.Extension {
return predicate.Extension(sql.FieldGT(FieldCreatedAt, v))
}
// CreatedAtGTE applies the GTE predicate on the "created_at" field.
func CreatedAtGTE(v time.Time) predicate.Extension {
return predicate.Extension(sql.FieldGTE(FieldCreatedAt, v))
}
// CreatedAtLT applies the LT predicate on the "created_at" field.
func CreatedAtLT(v time.Time) predicate.Extension {
return predicate.Extension(sql.FieldLT(FieldCreatedAt, v))
}
// CreatedAtLTE applies the LTE predicate on the "created_at" field.
func CreatedAtLTE(v time.Time) predicate.Extension {
return predicate.Extension(sql.FieldLTE(FieldCreatedAt, v))
}
// And groups predicates with the AND operator between them.
func And(predicates ...predicate.Extension) predicate.Extension {
return predicate.Extension(sql.AndPredicates(predicates...))
}
// Or groups predicates with the OR operator between them.
func Or(predicates ...predicate.Extension) predicate.Extension {
return predicate.Extension(sql.OrPredicates(predicates...))
}
// Not applies the not operator on the given predicate.
func Not(p predicate.Extension) predicate.Extension {
return predicate.Extension(sql.NotPredicates(p))
}

View File

@@ -0,0 +1,615 @@
// Code generated by ent, DO NOT EDIT.
package db
import (
"context"
"errors"
"fmt"
"time"
"entgo.io/ent/dialect"
"entgo.io/ent/dialect/sql"
"entgo.io/ent/dialect/sql/sqlgraph"
"entgo.io/ent/schema/field"
"github.com/chaitin/MonkeyCode/backend/db/extension"
"github.com/google/uuid"
)
// ExtensionCreate is the builder for creating a Extension entity.
type ExtensionCreate struct {
config
mutation *ExtensionMutation
hooks []Hook
conflict []sql.ConflictOption
}
// SetVersion sets the "version" field.
func (ec *ExtensionCreate) SetVersion(s string) *ExtensionCreate {
ec.mutation.SetVersion(s)
return ec
}
// SetPath sets the "path" field.
func (ec *ExtensionCreate) SetPath(s string) *ExtensionCreate {
ec.mutation.SetPath(s)
return ec
}
// SetCreatedAt sets the "created_at" field.
func (ec *ExtensionCreate) SetCreatedAt(t time.Time) *ExtensionCreate {
ec.mutation.SetCreatedAt(t)
return ec
}
// SetNillableCreatedAt sets the "created_at" field if the given value is not nil.
func (ec *ExtensionCreate) SetNillableCreatedAt(t *time.Time) *ExtensionCreate {
if t != nil {
ec.SetCreatedAt(*t)
}
return ec
}
// SetID sets the "id" field.
func (ec *ExtensionCreate) SetID(u uuid.UUID) *ExtensionCreate {
ec.mutation.SetID(u)
return ec
}
// Mutation returns the ExtensionMutation object of the builder.
func (ec *ExtensionCreate) Mutation() *ExtensionMutation {
return ec.mutation
}
// Save creates the Extension in the database.
func (ec *ExtensionCreate) Save(ctx context.Context) (*Extension, error) {
ec.defaults()
return withHooks(ctx, ec.sqlSave, ec.mutation, ec.hooks)
}
// SaveX calls Save and panics if Save returns an error.
func (ec *ExtensionCreate) SaveX(ctx context.Context) *Extension {
v, err := ec.Save(ctx)
if err != nil {
panic(err)
}
return v
}
// Exec executes the query.
func (ec *ExtensionCreate) Exec(ctx context.Context) error {
_, err := ec.Save(ctx)
return err
}
// ExecX is like Exec, but panics if an error occurs.
func (ec *ExtensionCreate) ExecX(ctx context.Context) {
if err := ec.Exec(ctx); err != nil {
panic(err)
}
}
// defaults sets the default values of the builder before save.
func (ec *ExtensionCreate) defaults() {
if _, ok := ec.mutation.CreatedAt(); !ok {
v := extension.DefaultCreatedAt()
ec.mutation.SetCreatedAt(v)
}
}
// check runs all checks and user-defined validators on the builder.
func (ec *ExtensionCreate) check() error {
if _, ok := ec.mutation.Version(); !ok {
return &ValidationError{Name: "version", err: errors.New(`db: missing required field "Extension.version"`)}
}
if _, ok := ec.mutation.Path(); !ok {
return &ValidationError{Name: "path", err: errors.New(`db: missing required field "Extension.path"`)}
}
if _, ok := ec.mutation.CreatedAt(); !ok {
return &ValidationError{Name: "created_at", err: errors.New(`db: missing required field "Extension.created_at"`)}
}
return nil
}
func (ec *ExtensionCreate) sqlSave(ctx context.Context) (*Extension, error) {
if err := ec.check(); err != nil {
return nil, err
}
_node, _spec := ec.createSpec()
if err := sqlgraph.CreateNode(ctx, ec.driver, _spec); err != nil {
if sqlgraph.IsConstraintError(err) {
err = &ConstraintError{msg: err.Error(), wrap: err}
}
return nil, err
}
if _spec.ID.Value != nil {
if id, ok := _spec.ID.Value.(*uuid.UUID); ok {
_node.ID = *id
} else if err := _node.ID.Scan(_spec.ID.Value); err != nil {
return nil, err
}
}
ec.mutation.id = &_node.ID
ec.mutation.done = true
return _node, nil
}
func (ec *ExtensionCreate) createSpec() (*Extension, *sqlgraph.CreateSpec) {
var (
_node = &Extension{config: ec.config}
_spec = sqlgraph.NewCreateSpec(extension.Table, sqlgraph.NewFieldSpec(extension.FieldID, field.TypeUUID))
)
_spec.OnConflict = ec.conflict
if id, ok := ec.mutation.ID(); ok {
_node.ID = id
_spec.ID.Value = &id
}
if value, ok := ec.mutation.Version(); ok {
_spec.SetField(extension.FieldVersion, field.TypeString, value)
_node.Version = value
}
if value, ok := ec.mutation.Path(); ok {
_spec.SetField(extension.FieldPath, field.TypeString, value)
_node.Path = value
}
if value, ok := ec.mutation.CreatedAt(); ok {
_spec.SetField(extension.FieldCreatedAt, field.TypeTime, value)
_node.CreatedAt = value
}
return _node, _spec
}
// OnConflict allows configuring the `ON CONFLICT` / `ON DUPLICATE KEY` clause
// of the `INSERT` statement. For example:
//
// client.Extension.Create().
// SetVersion(v).
// OnConflict(
// // Update the row with the new values
// // the was proposed for insertion.
// sql.ResolveWithNewValues(),
// ).
// // Override some of the fields with custom
// // update values.
// Update(func(u *ent.ExtensionUpsert) {
// SetVersion(v+v).
// }).
// Exec(ctx)
func (ec *ExtensionCreate) OnConflict(opts ...sql.ConflictOption) *ExtensionUpsertOne {
ec.conflict = opts
return &ExtensionUpsertOne{
create: ec,
}
}
// OnConflictColumns calls `OnConflict` and configures the columns
// as conflict target. Using this option is equivalent to using:
//
// client.Extension.Create().
// OnConflict(sql.ConflictColumns(columns...)).
// Exec(ctx)
func (ec *ExtensionCreate) OnConflictColumns(columns ...string) *ExtensionUpsertOne {
ec.conflict = append(ec.conflict, sql.ConflictColumns(columns...))
return &ExtensionUpsertOne{
create: ec,
}
}
type (
// ExtensionUpsertOne is the builder for "upsert"-ing
// one Extension node.
ExtensionUpsertOne struct {
create *ExtensionCreate
}
// ExtensionUpsert is the "OnConflict" setter.
ExtensionUpsert struct {
*sql.UpdateSet
}
)
// SetVersion sets the "version" field.
func (u *ExtensionUpsert) SetVersion(v string) *ExtensionUpsert {
u.Set(extension.FieldVersion, v)
return u
}
// UpdateVersion sets the "version" field to the value that was provided on create.
func (u *ExtensionUpsert) UpdateVersion() *ExtensionUpsert {
u.SetExcluded(extension.FieldVersion)
return u
}
// SetPath sets the "path" field.
func (u *ExtensionUpsert) SetPath(v string) *ExtensionUpsert {
u.Set(extension.FieldPath, v)
return u
}
// UpdatePath sets the "path" field to the value that was provided on create.
func (u *ExtensionUpsert) UpdatePath() *ExtensionUpsert {
u.SetExcluded(extension.FieldPath)
return u
}
// SetCreatedAt sets the "created_at" field.
func (u *ExtensionUpsert) SetCreatedAt(v time.Time) *ExtensionUpsert {
u.Set(extension.FieldCreatedAt, v)
return u
}
// UpdateCreatedAt sets the "created_at" field to the value that was provided on create.
func (u *ExtensionUpsert) UpdateCreatedAt() *ExtensionUpsert {
u.SetExcluded(extension.FieldCreatedAt)
return u
}
// UpdateNewValues updates the mutable fields using the new values that were set on create except the ID field.
// Using this option is equivalent to using:
//
// client.Extension.Create().
// OnConflict(
// sql.ResolveWithNewValues(),
// sql.ResolveWith(func(u *sql.UpdateSet) {
// u.SetIgnore(extension.FieldID)
// }),
// ).
// Exec(ctx)
func (u *ExtensionUpsertOne) UpdateNewValues() *ExtensionUpsertOne {
u.create.conflict = append(u.create.conflict, sql.ResolveWithNewValues())
u.create.conflict = append(u.create.conflict, sql.ResolveWith(func(s *sql.UpdateSet) {
if _, exists := u.create.mutation.ID(); exists {
s.SetIgnore(extension.FieldID)
}
}))
return u
}
// Ignore sets each column to itself in case of conflict.
// Using this option is equivalent to using:
//
// client.Extension.Create().
// OnConflict(sql.ResolveWithIgnore()).
// Exec(ctx)
func (u *ExtensionUpsertOne) Ignore() *ExtensionUpsertOne {
u.create.conflict = append(u.create.conflict, sql.ResolveWithIgnore())
return u
}
// DoNothing configures the conflict_action to `DO NOTHING`.
// Supported only by SQLite and PostgreSQL.
func (u *ExtensionUpsertOne) DoNothing() *ExtensionUpsertOne {
u.create.conflict = append(u.create.conflict, sql.DoNothing())
return u
}
// Update allows overriding fields `UPDATE` values. See the ExtensionCreate.OnConflict
// documentation for more info.
func (u *ExtensionUpsertOne) Update(set func(*ExtensionUpsert)) *ExtensionUpsertOne {
u.create.conflict = append(u.create.conflict, sql.ResolveWith(func(update *sql.UpdateSet) {
set(&ExtensionUpsert{UpdateSet: update})
}))
return u
}
// SetVersion sets the "version" field.
func (u *ExtensionUpsertOne) SetVersion(v string) *ExtensionUpsertOne {
return u.Update(func(s *ExtensionUpsert) {
s.SetVersion(v)
})
}
// UpdateVersion sets the "version" field to the value that was provided on create.
func (u *ExtensionUpsertOne) UpdateVersion() *ExtensionUpsertOne {
return u.Update(func(s *ExtensionUpsert) {
s.UpdateVersion()
})
}
// SetPath sets the "path" field.
func (u *ExtensionUpsertOne) SetPath(v string) *ExtensionUpsertOne {
return u.Update(func(s *ExtensionUpsert) {
s.SetPath(v)
})
}
// UpdatePath sets the "path" field to the value that was provided on create.
func (u *ExtensionUpsertOne) UpdatePath() *ExtensionUpsertOne {
return u.Update(func(s *ExtensionUpsert) {
s.UpdatePath()
})
}
// SetCreatedAt sets the "created_at" field.
func (u *ExtensionUpsertOne) SetCreatedAt(v time.Time) *ExtensionUpsertOne {
return u.Update(func(s *ExtensionUpsert) {
s.SetCreatedAt(v)
})
}
// UpdateCreatedAt sets the "created_at" field to the value that was provided on create.
func (u *ExtensionUpsertOne) UpdateCreatedAt() *ExtensionUpsertOne {
return u.Update(func(s *ExtensionUpsert) {
s.UpdateCreatedAt()
})
}
// Exec executes the query.
func (u *ExtensionUpsertOne) Exec(ctx context.Context) error {
if len(u.create.conflict) == 0 {
return errors.New("db: missing options for ExtensionCreate.OnConflict")
}
return u.create.Exec(ctx)
}
// ExecX is like Exec, but panics if an error occurs.
func (u *ExtensionUpsertOne) ExecX(ctx context.Context) {
if err := u.create.Exec(ctx); err != nil {
panic(err)
}
}
// Exec executes the UPSERT query and returns the inserted/updated ID.
func (u *ExtensionUpsertOne) ID(ctx context.Context) (id uuid.UUID, err error) {
if u.create.driver.Dialect() == dialect.MySQL {
// In case of "ON CONFLICT", there is no way to get back non-numeric ID
// fields from the database since MySQL does not support the RETURNING clause.
return id, errors.New("db: ExtensionUpsertOne.ID is not supported by MySQL driver. Use ExtensionUpsertOne.Exec instead")
}
node, err := u.create.Save(ctx)
if err != nil {
return id, err
}
return node.ID, nil
}
// IDX is like ID, but panics if an error occurs.
func (u *ExtensionUpsertOne) IDX(ctx context.Context) uuid.UUID {
id, err := u.ID(ctx)
if err != nil {
panic(err)
}
return id
}
// ExtensionCreateBulk is the builder for creating many Extension entities in bulk.
type ExtensionCreateBulk struct {
config
err error
builders []*ExtensionCreate
conflict []sql.ConflictOption
}
// Save creates the Extension entities in the database.
func (ecb *ExtensionCreateBulk) Save(ctx context.Context) ([]*Extension, error) {
if ecb.err != nil {
return nil, ecb.err
}
specs := make([]*sqlgraph.CreateSpec, len(ecb.builders))
nodes := make([]*Extension, len(ecb.builders))
mutators := make([]Mutator, len(ecb.builders))
for i := range ecb.builders {
func(i int, root context.Context) {
builder := ecb.builders[i]
builder.defaults()
var mut Mutator = MutateFunc(func(ctx context.Context, m Mutation) (Value, error) {
mutation, ok := m.(*ExtensionMutation)
if !ok {
return nil, fmt.Errorf("unexpected mutation type %T", m)
}
if err := builder.check(); err != nil {
return nil, err
}
builder.mutation = mutation
var err error
nodes[i], specs[i] = builder.createSpec()
if i < len(mutators)-1 {
_, err = mutators[i+1].Mutate(root, ecb.builders[i+1].mutation)
} else {
spec := &sqlgraph.BatchCreateSpec{Nodes: specs}
spec.OnConflict = ecb.conflict
// Invoke the actual operation on the latest mutation in the chain.
if err = sqlgraph.BatchCreate(ctx, ecb.driver, spec); err != nil {
if sqlgraph.IsConstraintError(err) {
err = &ConstraintError{msg: err.Error(), wrap: err}
}
}
}
if err != nil {
return nil, err
}
mutation.id = &nodes[i].ID
mutation.done = true
return nodes[i], nil
})
for i := len(builder.hooks) - 1; i >= 0; i-- {
mut = builder.hooks[i](mut)
}
mutators[i] = mut
}(i, ctx)
}
if len(mutators) > 0 {
if _, err := mutators[0].Mutate(ctx, ecb.builders[0].mutation); err != nil {
return nil, err
}
}
return nodes, nil
}
// SaveX is like Save, but panics if an error occurs.
func (ecb *ExtensionCreateBulk) SaveX(ctx context.Context) []*Extension {
v, err := ecb.Save(ctx)
if err != nil {
panic(err)
}
return v
}
// Exec executes the query.
func (ecb *ExtensionCreateBulk) Exec(ctx context.Context) error {
_, err := ecb.Save(ctx)
return err
}
// ExecX is like Exec, but panics if an error occurs.
func (ecb *ExtensionCreateBulk) ExecX(ctx context.Context) {
if err := ecb.Exec(ctx); err != nil {
panic(err)
}
}
// OnConflict allows configuring the `ON CONFLICT` / `ON DUPLICATE KEY` clause
// of the `INSERT` statement. For example:
//
// client.Extension.CreateBulk(builders...).
// OnConflict(
// // Update the row with the new values
// // the was proposed for insertion.
// sql.ResolveWithNewValues(),
// ).
// // Override some of the fields with custom
// // update values.
// Update(func(u *ent.ExtensionUpsert) {
// SetVersion(v+v).
// }).
// Exec(ctx)
func (ecb *ExtensionCreateBulk) OnConflict(opts ...sql.ConflictOption) *ExtensionUpsertBulk {
ecb.conflict = opts
return &ExtensionUpsertBulk{
create: ecb,
}
}
// OnConflictColumns calls `OnConflict` and configures the columns
// as conflict target. Using this option is equivalent to using:
//
// client.Extension.Create().
// OnConflict(sql.ConflictColumns(columns...)).
// Exec(ctx)
func (ecb *ExtensionCreateBulk) OnConflictColumns(columns ...string) *ExtensionUpsertBulk {
ecb.conflict = append(ecb.conflict, sql.ConflictColumns(columns...))
return &ExtensionUpsertBulk{
create: ecb,
}
}
// ExtensionUpsertBulk is the builder for "upsert"-ing
// a bulk of Extension nodes.
type ExtensionUpsertBulk struct {
create *ExtensionCreateBulk
}
// UpdateNewValues updates the mutable fields using the new values that
// were set on create. Using this option is equivalent to using:
//
// client.Extension.Create().
// OnConflict(
// sql.ResolveWithNewValues(),
// sql.ResolveWith(func(u *sql.UpdateSet) {
// u.SetIgnore(extension.FieldID)
// }),
// ).
// Exec(ctx)
func (u *ExtensionUpsertBulk) UpdateNewValues() *ExtensionUpsertBulk {
u.create.conflict = append(u.create.conflict, sql.ResolveWithNewValues())
u.create.conflict = append(u.create.conflict, sql.ResolveWith(func(s *sql.UpdateSet) {
for _, b := range u.create.builders {
if _, exists := b.mutation.ID(); exists {
s.SetIgnore(extension.FieldID)
}
}
}))
return u
}
// Ignore sets each column to itself in case of conflict.
// Using this option is equivalent to using:
//
// client.Extension.Create().
// OnConflict(sql.ResolveWithIgnore()).
// Exec(ctx)
func (u *ExtensionUpsertBulk) Ignore() *ExtensionUpsertBulk {
u.create.conflict = append(u.create.conflict, sql.ResolveWithIgnore())
return u
}
// DoNothing configures the conflict_action to `DO NOTHING`.
// Supported only by SQLite and PostgreSQL.
func (u *ExtensionUpsertBulk) DoNothing() *ExtensionUpsertBulk {
u.create.conflict = append(u.create.conflict, sql.DoNothing())
return u
}
// Update allows overriding fields `UPDATE` values. See the ExtensionCreateBulk.OnConflict
// documentation for more info.
func (u *ExtensionUpsertBulk) Update(set func(*ExtensionUpsert)) *ExtensionUpsertBulk {
u.create.conflict = append(u.create.conflict, sql.ResolveWith(func(update *sql.UpdateSet) {
set(&ExtensionUpsert{UpdateSet: update})
}))
return u
}
// SetVersion sets the "version" field.
func (u *ExtensionUpsertBulk) SetVersion(v string) *ExtensionUpsertBulk {
return u.Update(func(s *ExtensionUpsert) {
s.SetVersion(v)
})
}
// UpdateVersion sets the "version" field to the value that was provided on create.
func (u *ExtensionUpsertBulk) UpdateVersion() *ExtensionUpsertBulk {
return u.Update(func(s *ExtensionUpsert) {
s.UpdateVersion()
})
}
// SetPath sets the "path" field.
func (u *ExtensionUpsertBulk) SetPath(v string) *ExtensionUpsertBulk {
return u.Update(func(s *ExtensionUpsert) {
s.SetPath(v)
})
}
// UpdatePath sets the "path" field to the value that was provided on create.
func (u *ExtensionUpsertBulk) UpdatePath() *ExtensionUpsertBulk {
return u.Update(func(s *ExtensionUpsert) {
s.UpdatePath()
})
}
// SetCreatedAt sets the "created_at" field.
func (u *ExtensionUpsertBulk) SetCreatedAt(v time.Time) *ExtensionUpsertBulk {
return u.Update(func(s *ExtensionUpsert) {
s.SetCreatedAt(v)
})
}
// UpdateCreatedAt sets the "created_at" field to the value that was provided on create.
func (u *ExtensionUpsertBulk) UpdateCreatedAt() *ExtensionUpsertBulk {
return u.Update(func(s *ExtensionUpsert) {
s.UpdateCreatedAt()
})
}
// Exec executes the query.
func (u *ExtensionUpsertBulk) Exec(ctx context.Context) error {
if u.create.err != nil {
return u.create.err
}
for i, b := range u.create.builders {
if len(b.conflict) != 0 {
return fmt.Errorf("db: OnConflict was set for builder %d. Set it on the ExtensionCreateBulk instead", i)
}
}
if len(u.create.conflict) == 0 {
return errors.New("db: missing options for ExtensionCreateBulk.OnConflict")
}
return u.create.Exec(ctx)
}
// ExecX is like Exec, but panics if an error occurs.
func (u *ExtensionUpsertBulk) ExecX(ctx context.Context) {
if err := u.create.Exec(ctx); err != nil {
panic(err)
}
}

View File

@@ -0,0 +1,88 @@
// Code generated by ent, DO NOT EDIT.
package db
import (
"context"
"entgo.io/ent/dialect/sql"
"entgo.io/ent/dialect/sql/sqlgraph"
"entgo.io/ent/schema/field"
"github.com/chaitin/MonkeyCode/backend/db/extension"
"github.com/chaitin/MonkeyCode/backend/db/predicate"
)
// ExtensionDelete is the builder for deleting a Extension entity.
type ExtensionDelete struct {
config
hooks []Hook
mutation *ExtensionMutation
}
// Where appends a list predicates to the ExtensionDelete builder.
func (ed *ExtensionDelete) Where(ps ...predicate.Extension) *ExtensionDelete {
ed.mutation.Where(ps...)
return ed
}
// Exec executes the deletion query and returns how many vertices were deleted.
func (ed *ExtensionDelete) Exec(ctx context.Context) (int, error) {
return withHooks(ctx, ed.sqlExec, ed.mutation, ed.hooks)
}
// ExecX is like Exec, but panics if an error occurs.
func (ed *ExtensionDelete) ExecX(ctx context.Context) int {
n, err := ed.Exec(ctx)
if err != nil {
panic(err)
}
return n
}
func (ed *ExtensionDelete) sqlExec(ctx context.Context) (int, error) {
_spec := sqlgraph.NewDeleteSpec(extension.Table, sqlgraph.NewFieldSpec(extension.FieldID, field.TypeUUID))
if ps := ed.mutation.predicates; len(ps) > 0 {
_spec.Predicate = func(selector *sql.Selector) {
for i := range ps {
ps[i](selector)
}
}
}
affected, err := sqlgraph.DeleteNodes(ctx, ed.driver, _spec)
if err != nil && sqlgraph.IsConstraintError(err) {
err = &ConstraintError{msg: err.Error(), wrap: err}
}
ed.mutation.done = true
return affected, err
}
// ExtensionDeleteOne is the builder for deleting a single Extension entity.
type ExtensionDeleteOne struct {
ed *ExtensionDelete
}
// Where appends a list predicates to the ExtensionDelete builder.
func (edo *ExtensionDeleteOne) Where(ps ...predicate.Extension) *ExtensionDeleteOne {
edo.ed.mutation.Where(ps...)
return edo
}
// Exec executes the deletion query.
func (edo *ExtensionDeleteOne) Exec(ctx context.Context) error {
n, err := edo.ed.Exec(ctx)
switch {
case err != nil:
return err
case n == 0:
return &NotFoundError{extension.Label}
default:
return nil
}
}
// ExecX is like Exec, but panics if an error occurs.
func (edo *ExtensionDeleteOne) ExecX(ctx context.Context) {
if err := edo.Exec(ctx); err != nil {
panic(err)
}
}

View File

@@ -0,0 +1,578 @@
// Code generated by ent, DO NOT EDIT.
package db
import (
"context"
"fmt"
"math"
"entgo.io/ent"
"entgo.io/ent/dialect"
"entgo.io/ent/dialect/sql"
"entgo.io/ent/dialect/sql/sqlgraph"
"entgo.io/ent/schema/field"
"github.com/chaitin/MonkeyCode/backend/db/extension"
"github.com/chaitin/MonkeyCode/backend/db/predicate"
"github.com/google/uuid"
)
// ExtensionQuery is the builder for querying Extension entities.
type ExtensionQuery struct {
config
ctx *QueryContext
order []extension.OrderOption
inters []Interceptor
predicates []predicate.Extension
modifiers []func(*sql.Selector)
// intermediate query (i.e. traversal path).
sql *sql.Selector
path func(context.Context) (*sql.Selector, error)
}
// Where adds a new predicate for the ExtensionQuery builder.
func (eq *ExtensionQuery) Where(ps ...predicate.Extension) *ExtensionQuery {
eq.predicates = append(eq.predicates, ps...)
return eq
}
// Limit the number of records to be returned by this query.
func (eq *ExtensionQuery) Limit(limit int) *ExtensionQuery {
eq.ctx.Limit = &limit
return eq
}
// Offset to start from.
func (eq *ExtensionQuery) Offset(offset int) *ExtensionQuery {
eq.ctx.Offset = &offset
return eq
}
// Unique configures the query builder to filter duplicate records on query.
// By default, unique is set to true, and can be disabled using this method.
func (eq *ExtensionQuery) Unique(unique bool) *ExtensionQuery {
eq.ctx.Unique = &unique
return eq
}
// Order specifies how the records should be ordered.
func (eq *ExtensionQuery) Order(o ...extension.OrderOption) *ExtensionQuery {
eq.order = append(eq.order, o...)
return eq
}
// First returns the first Extension entity from the query.
// Returns a *NotFoundError when no Extension was found.
func (eq *ExtensionQuery) First(ctx context.Context) (*Extension, error) {
nodes, err := eq.Limit(1).All(setContextOp(ctx, eq.ctx, ent.OpQueryFirst))
if err != nil {
return nil, err
}
if len(nodes) == 0 {
return nil, &NotFoundError{extension.Label}
}
return nodes[0], nil
}
// FirstX is like First, but panics if an error occurs.
func (eq *ExtensionQuery) FirstX(ctx context.Context) *Extension {
node, err := eq.First(ctx)
if err != nil && !IsNotFound(err) {
panic(err)
}
return node
}
// FirstID returns the first Extension ID from the query.
// Returns a *NotFoundError when no Extension ID was found.
func (eq *ExtensionQuery) FirstID(ctx context.Context) (id uuid.UUID, err error) {
var ids []uuid.UUID
if ids, err = eq.Limit(1).IDs(setContextOp(ctx, eq.ctx, ent.OpQueryFirstID)); err != nil {
return
}
if len(ids) == 0 {
err = &NotFoundError{extension.Label}
return
}
return ids[0], nil
}
// FirstIDX is like FirstID, but panics if an error occurs.
func (eq *ExtensionQuery) FirstIDX(ctx context.Context) uuid.UUID {
id, err := eq.FirstID(ctx)
if err != nil && !IsNotFound(err) {
panic(err)
}
return id
}
// Only returns a single Extension entity found by the query, ensuring it only returns one.
// Returns a *NotSingularError when more than one Extension entity is found.
// Returns a *NotFoundError when no Extension entities are found.
func (eq *ExtensionQuery) Only(ctx context.Context) (*Extension, error) {
nodes, err := eq.Limit(2).All(setContextOp(ctx, eq.ctx, ent.OpQueryOnly))
if err != nil {
return nil, err
}
switch len(nodes) {
case 1:
return nodes[0], nil
case 0:
return nil, &NotFoundError{extension.Label}
default:
return nil, &NotSingularError{extension.Label}
}
}
// OnlyX is like Only, but panics if an error occurs.
func (eq *ExtensionQuery) OnlyX(ctx context.Context) *Extension {
node, err := eq.Only(ctx)
if err != nil {
panic(err)
}
return node
}
// OnlyID is like Only, but returns the only Extension ID in the query.
// Returns a *NotSingularError when more than one Extension ID is found.
// Returns a *NotFoundError when no entities are found.
func (eq *ExtensionQuery) OnlyID(ctx context.Context) (id uuid.UUID, err error) {
var ids []uuid.UUID
if ids, err = eq.Limit(2).IDs(setContextOp(ctx, eq.ctx, ent.OpQueryOnlyID)); err != nil {
return
}
switch len(ids) {
case 1:
id = ids[0]
case 0:
err = &NotFoundError{extension.Label}
default:
err = &NotSingularError{extension.Label}
}
return
}
// OnlyIDX is like OnlyID, but panics if an error occurs.
func (eq *ExtensionQuery) OnlyIDX(ctx context.Context) uuid.UUID {
id, err := eq.OnlyID(ctx)
if err != nil {
panic(err)
}
return id
}
// All executes the query and returns a list of Extensions.
func (eq *ExtensionQuery) All(ctx context.Context) ([]*Extension, error) {
ctx = setContextOp(ctx, eq.ctx, ent.OpQueryAll)
if err := eq.prepareQuery(ctx); err != nil {
return nil, err
}
qr := querierAll[[]*Extension, *ExtensionQuery]()
return withInterceptors[[]*Extension](ctx, eq, qr, eq.inters)
}
// AllX is like All, but panics if an error occurs.
func (eq *ExtensionQuery) AllX(ctx context.Context) []*Extension {
nodes, err := eq.All(ctx)
if err != nil {
panic(err)
}
return nodes
}
// IDs executes the query and returns a list of Extension IDs.
func (eq *ExtensionQuery) IDs(ctx context.Context) (ids []uuid.UUID, err error) {
if eq.ctx.Unique == nil && eq.path != nil {
eq.Unique(true)
}
ctx = setContextOp(ctx, eq.ctx, ent.OpQueryIDs)
if err = eq.Select(extension.FieldID).Scan(ctx, &ids); err != nil {
return nil, err
}
return ids, nil
}
// IDsX is like IDs, but panics if an error occurs.
func (eq *ExtensionQuery) IDsX(ctx context.Context) []uuid.UUID {
ids, err := eq.IDs(ctx)
if err != nil {
panic(err)
}
return ids
}
// Count returns the count of the given query.
func (eq *ExtensionQuery) Count(ctx context.Context) (int, error) {
ctx = setContextOp(ctx, eq.ctx, ent.OpQueryCount)
if err := eq.prepareQuery(ctx); err != nil {
return 0, err
}
return withInterceptors[int](ctx, eq, querierCount[*ExtensionQuery](), eq.inters)
}
// CountX is like Count, but panics if an error occurs.
func (eq *ExtensionQuery) CountX(ctx context.Context) int {
count, err := eq.Count(ctx)
if err != nil {
panic(err)
}
return count
}
// Exist returns true if the query has elements in the graph.
func (eq *ExtensionQuery) Exist(ctx context.Context) (bool, error) {
ctx = setContextOp(ctx, eq.ctx, ent.OpQueryExist)
switch _, err := eq.FirstID(ctx); {
case IsNotFound(err):
return false, nil
case err != nil:
return false, fmt.Errorf("db: check existence: %w", err)
default:
return true, nil
}
}
// ExistX is like Exist, but panics if an error occurs.
func (eq *ExtensionQuery) ExistX(ctx context.Context) bool {
exist, err := eq.Exist(ctx)
if err != nil {
panic(err)
}
return exist
}
// Clone returns a duplicate of the ExtensionQuery builder, including all associated steps. It can be
// used to prepare common query builders and use them differently after the clone is made.
func (eq *ExtensionQuery) Clone() *ExtensionQuery {
if eq == nil {
return nil
}
return &ExtensionQuery{
config: eq.config,
ctx: eq.ctx.Clone(),
order: append([]extension.OrderOption{}, eq.order...),
inters: append([]Interceptor{}, eq.inters...),
predicates: append([]predicate.Extension{}, eq.predicates...),
// clone intermediate query.
sql: eq.sql.Clone(),
path: eq.path,
modifiers: append([]func(*sql.Selector){}, eq.modifiers...),
}
}
// GroupBy is used to group vertices by one or more fields/columns.
// It is often used with aggregate functions, like: count, max, mean, min, sum.
//
// Example:
//
// var v []struct {
// Version string `json:"version,omitempty"`
// Count int `json:"count,omitempty"`
// }
//
// client.Extension.Query().
// GroupBy(extension.FieldVersion).
// Aggregate(db.Count()).
// Scan(ctx, &v)
func (eq *ExtensionQuery) GroupBy(field string, fields ...string) *ExtensionGroupBy {
eq.ctx.Fields = append([]string{field}, fields...)
grbuild := &ExtensionGroupBy{build: eq}
grbuild.flds = &eq.ctx.Fields
grbuild.label = extension.Label
grbuild.scan = grbuild.Scan
return grbuild
}
// Select allows the selection one or more fields/columns for the given query,
// instead of selecting all fields in the entity.
//
// Example:
//
// var v []struct {
// Version string `json:"version,omitempty"`
// }
//
// client.Extension.Query().
// Select(extension.FieldVersion).
// Scan(ctx, &v)
func (eq *ExtensionQuery) Select(fields ...string) *ExtensionSelect {
eq.ctx.Fields = append(eq.ctx.Fields, fields...)
sbuild := &ExtensionSelect{ExtensionQuery: eq}
sbuild.label = extension.Label
sbuild.flds, sbuild.scan = &eq.ctx.Fields, sbuild.Scan
return sbuild
}
// Aggregate returns a ExtensionSelect configured with the given aggregations.
func (eq *ExtensionQuery) Aggregate(fns ...AggregateFunc) *ExtensionSelect {
return eq.Select().Aggregate(fns...)
}
func (eq *ExtensionQuery) prepareQuery(ctx context.Context) error {
for _, inter := range eq.inters {
if inter == nil {
return fmt.Errorf("db: uninitialized interceptor (forgotten import db/runtime?)")
}
if trv, ok := inter.(Traverser); ok {
if err := trv.Traverse(ctx, eq); err != nil {
return err
}
}
}
for _, f := range eq.ctx.Fields {
if !extension.ValidColumn(f) {
return &ValidationError{Name: f, err: fmt.Errorf("db: invalid field %q for query", f)}
}
}
if eq.path != nil {
prev, err := eq.path(ctx)
if err != nil {
return err
}
eq.sql = prev
}
return nil
}
func (eq *ExtensionQuery) sqlAll(ctx context.Context, hooks ...queryHook) ([]*Extension, error) {
var (
nodes = []*Extension{}
_spec = eq.querySpec()
)
_spec.ScanValues = func(columns []string) ([]any, error) {
return (*Extension).scanValues(nil, columns)
}
_spec.Assign = func(columns []string, values []any) error {
node := &Extension{config: eq.config}
nodes = append(nodes, node)
return node.assignValues(columns, values)
}
if len(eq.modifiers) > 0 {
_spec.Modifiers = eq.modifiers
}
for i := range hooks {
hooks[i](ctx, _spec)
}
if err := sqlgraph.QueryNodes(ctx, eq.driver, _spec); err != nil {
return nil, err
}
if len(nodes) == 0 {
return nodes, nil
}
return nodes, nil
}
func (eq *ExtensionQuery) sqlCount(ctx context.Context) (int, error) {
_spec := eq.querySpec()
if len(eq.modifiers) > 0 {
_spec.Modifiers = eq.modifiers
}
_spec.Node.Columns = eq.ctx.Fields
if len(eq.ctx.Fields) > 0 {
_spec.Unique = eq.ctx.Unique != nil && *eq.ctx.Unique
}
return sqlgraph.CountNodes(ctx, eq.driver, _spec)
}
func (eq *ExtensionQuery) querySpec() *sqlgraph.QuerySpec {
_spec := sqlgraph.NewQuerySpec(extension.Table, extension.Columns, sqlgraph.NewFieldSpec(extension.FieldID, field.TypeUUID))
_spec.From = eq.sql
if unique := eq.ctx.Unique; unique != nil {
_spec.Unique = *unique
} else if eq.path != nil {
_spec.Unique = true
}
if fields := eq.ctx.Fields; len(fields) > 0 {
_spec.Node.Columns = make([]string, 0, len(fields))
_spec.Node.Columns = append(_spec.Node.Columns, extension.FieldID)
for i := range fields {
if fields[i] != extension.FieldID {
_spec.Node.Columns = append(_spec.Node.Columns, fields[i])
}
}
}
if ps := eq.predicates; len(ps) > 0 {
_spec.Predicate = func(selector *sql.Selector) {
for i := range ps {
ps[i](selector)
}
}
}
if limit := eq.ctx.Limit; limit != nil {
_spec.Limit = *limit
}
if offset := eq.ctx.Offset; offset != nil {
_spec.Offset = *offset
}
if ps := eq.order; len(ps) > 0 {
_spec.Order = func(selector *sql.Selector) {
for i := range ps {
ps[i](selector)
}
}
}
return _spec
}
func (eq *ExtensionQuery) sqlQuery(ctx context.Context) *sql.Selector {
builder := sql.Dialect(eq.driver.Dialect())
t1 := builder.Table(extension.Table)
columns := eq.ctx.Fields
if len(columns) == 0 {
columns = extension.Columns
}
selector := builder.Select(t1.Columns(columns...)...).From(t1)
if eq.sql != nil {
selector = eq.sql
selector.Select(selector.Columns(columns...)...)
}
if eq.ctx.Unique != nil && *eq.ctx.Unique {
selector.Distinct()
}
for _, m := range eq.modifiers {
m(selector)
}
for _, p := range eq.predicates {
p(selector)
}
for _, p := range eq.order {
p(selector)
}
if offset := eq.ctx.Offset; offset != nil {
// limit is mandatory for offset clause. We start
// with default value, and override it below if needed.
selector.Offset(*offset).Limit(math.MaxInt32)
}
if limit := eq.ctx.Limit; limit != nil {
selector.Limit(*limit)
}
return selector
}
// ForUpdate locks the selected rows against concurrent updates, and prevent them from being
// updated, deleted or "selected ... for update" by other sessions, until the transaction is
// either committed or rolled-back.
func (eq *ExtensionQuery) ForUpdate(opts ...sql.LockOption) *ExtensionQuery {
if eq.driver.Dialect() == dialect.Postgres {
eq.Unique(false)
}
eq.modifiers = append(eq.modifiers, func(s *sql.Selector) {
s.ForUpdate(opts...)
})
return eq
}
// ForShare behaves similarly to ForUpdate, except that it acquires a shared mode lock
// on any rows that are read. Other sessions can read the rows, but cannot modify them
// until your transaction commits.
func (eq *ExtensionQuery) ForShare(opts ...sql.LockOption) *ExtensionQuery {
if eq.driver.Dialect() == dialect.Postgres {
eq.Unique(false)
}
eq.modifiers = append(eq.modifiers, func(s *sql.Selector) {
s.ForShare(opts...)
})
return eq
}
// Modify adds a query modifier for attaching custom logic to queries.
func (eq *ExtensionQuery) Modify(modifiers ...func(s *sql.Selector)) *ExtensionSelect {
eq.modifiers = append(eq.modifiers, modifiers...)
return eq.Select()
}
// ExtensionGroupBy is the group-by builder for Extension entities.
type ExtensionGroupBy struct {
selector
build *ExtensionQuery
}
// Aggregate adds the given aggregation functions to the group-by query.
func (egb *ExtensionGroupBy) Aggregate(fns ...AggregateFunc) *ExtensionGroupBy {
egb.fns = append(egb.fns, fns...)
return egb
}
// Scan applies the selector query and scans the result into the given value.
func (egb *ExtensionGroupBy) Scan(ctx context.Context, v any) error {
ctx = setContextOp(ctx, egb.build.ctx, ent.OpQueryGroupBy)
if err := egb.build.prepareQuery(ctx); err != nil {
return err
}
return scanWithInterceptors[*ExtensionQuery, *ExtensionGroupBy](ctx, egb.build, egb, egb.build.inters, v)
}
func (egb *ExtensionGroupBy) sqlScan(ctx context.Context, root *ExtensionQuery, v any) error {
selector := root.sqlQuery(ctx).Select()
aggregation := make([]string, 0, len(egb.fns))
for _, fn := range egb.fns {
aggregation = append(aggregation, fn(selector))
}
if len(selector.SelectedColumns()) == 0 {
columns := make([]string, 0, len(*egb.flds)+len(egb.fns))
for _, f := range *egb.flds {
columns = append(columns, selector.C(f))
}
columns = append(columns, aggregation...)
selector.Select(columns...)
}
selector.GroupBy(selector.Columns(*egb.flds...)...)
if err := selector.Err(); err != nil {
return err
}
rows := &sql.Rows{}
query, args := selector.Query()
if err := egb.build.driver.Query(ctx, query, args, rows); err != nil {
return err
}
defer rows.Close()
return sql.ScanSlice(rows, v)
}
// ExtensionSelect is the builder for selecting fields of Extension entities.
type ExtensionSelect struct {
*ExtensionQuery
selector
}
// Aggregate adds the given aggregation functions to the selector query.
func (es *ExtensionSelect) Aggregate(fns ...AggregateFunc) *ExtensionSelect {
es.fns = append(es.fns, fns...)
return es
}
// Scan applies the selector query and scans the result into the given value.
func (es *ExtensionSelect) Scan(ctx context.Context, v any) error {
ctx = setContextOp(ctx, es.ctx, ent.OpQuerySelect)
if err := es.prepareQuery(ctx); err != nil {
return err
}
return scanWithInterceptors[*ExtensionQuery, *ExtensionSelect](ctx, es.ExtensionQuery, es, es.inters, v)
}
func (es *ExtensionSelect) sqlScan(ctx context.Context, root *ExtensionQuery, v any) error {
selector := root.sqlQuery(ctx)
aggregation := make([]string, 0, len(es.fns))
for _, fn := range es.fns {
aggregation = append(aggregation, fn(selector))
}
switch n := len(*es.selector.flds); {
case n == 0 && len(aggregation) > 0:
selector.Select(aggregation...)
case n != 0 && len(aggregation) > 0:
selector.AppendSelect(aggregation...)
}
rows := &sql.Rows{}
query, args := selector.Query()
if err := es.driver.Query(ctx, query, args, rows); err != nil {
return err
}
defer rows.Close()
return sql.ScanSlice(rows, v)
}
// Modify adds a query modifier for attaching custom logic to queries.
func (es *ExtensionSelect) Modify(modifiers ...func(s *sql.Selector)) *ExtensionSelect {
es.modifiers = append(es.modifiers, modifiers...)
return es
}

View File

@@ -0,0 +1,294 @@
// Code generated by ent, DO NOT EDIT.
package db
import (
"context"
"errors"
"fmt"
"time"
"entgo.io/ent/dialect/sql"
"entgo.io/ent/dialect/sql/sqlgraph"
"entgo.io/ent/schema/field"
"github.com/chaitin/MonkeyCode/backend/db/extension"
"github.com/chaitin/MonkeyCode/backend/db/predicate"
)
// ExtensionUpdate is the builder for updating Extension entities.
type ExtensionUpdate struct {
config
hooks []Hook
mutation *ExtensionMutation
modifiers []func(*sql.UpdateBuilder)
}
// Where appends a list predicates to the ExtensionUpdate builder.
func (eu *ExtensionUpdate) Where(ps ...predicate.Extension) *ExtensionUpdate {
eu.mutation.Where(ps...)
return eu
}
// SetVersion sets the "version" field.
func (eu *ExtensionUpdate) SetVersion(s string) *ExtensionUpdate {
eu.mutation.SetVersion(s)
return eu
}
// SetNillableVersion sets the "version" field if the given value is not nil.
func (eu *ExtensionUpdate) SetNillableVersion(s *string) *ExtensionUpdate {
if s != nil {
eu.SetVersion(*s)
}
return eu
}
// SetPath sets the "path" field.
func (eu *ExtensionUpdate) SetPath(s string) *ExtensionUpdate {
eu.mutation.SetPath(s)
return eu
}
// SetNillablePath sets the "path" field if the given value is not nil.
func (eu *ExtensionUpdate) SetNillablePath(s *string) *ExtensionUpdate {
if s != nil {
eu.SetPath(*s)
}
return eu
}
// SetCreatedAt sets the "created_at" field.
func (eu *ExtensionUpdate) SetCreatedAt(t time.Time) *ExtensionUpdate {
eu.mutation.SetCreatedAt(t)
return eu
}
// SetNillableCreatedAt sets the "created_at" field if the given value is not nil.
func (eu *ExtensionUpdate) SetNillableCreatedAt(t *time.Time) *ExtensionUpdate {
if t != nil {
eu.SetCreatedAt(*t)
}
return eu
}
// Mutation returns the ExtensionMutation object of the builder.
func (eu *ExtensionUpdate) Mutation() *ExtensionMutation {
return eu.mutation
}
// Save executes the query and returns the number of nodes affected by the update operation.
func (eu *ExtensionUpdate) Save(ctx context.Context) (int, error) {
return withHooks(ctx, eu.sqlSave, eu.mutation, eu.hooks)
}
// SaveX is like Save, but panics if an error occurs.
func (eu *ExtensionUpdate) SaveX(ctx context.Context) int {
affected, err := eu.Save(ctx)
if err != nil {
panic(err)
}
return affected
}
// Exec executes the query.
func (eu *ExtensionUpdate) Exec(ctx context.Context) error {
_, err := eu.Save(ctx)
return err
}
// ExecX is like Exec, but panics if an error occurs.
func (eu *ExtensionUpdate) ExecX(ctx context.Context) {
if err := eu.Exec(ctx); err != nil {
panic(err)
}
}
// Modify adds a statement modifier for attaching custom logic to the UPDATE statement.
func (eu *ExtensionUpdate) Modify(modifiers ...func(u *sql.UpdateBuilder)) *ExtensionUpdate {
eu.modifiers = append(eu.modifiers, modifiers...)
return eu
}
func (eu *ExtensionUpdate) sqlSave(ctx context.Context) (n int, err error) {
_spec := sqlgraph.NewUpdateSpec(extension.Table, extension.Columns, sqlgraph.NewFieldSpec(extension.FieldID, field.TypeUUID))
if ps := eu.mutation.predicates; len(ps) > 0 {
_spec.Predicate = func(selector *sql.Selector) {
for i := range ps {
ps[i](selector)
}
}
}
if value, ok := eu.mutation.Version(); ok {
_spec.SetField(extension.FieldVersion, field.TypeString, value)
}
if value, ok := eu.mutation.Path(); ok {
_spec.SetField(extension.FieldPath, field.TypeString, value)
}
if value, ok := eu.mutation.CreatedAt(); ok {
_spec.SetField(extension.FieldCreatedAt, field.TypeTime, value)
}
_spec.AddModifiers(eu.modifiers...)
if n, err = sqlgraph.UpdateNodes(ctx, eu.driver, _spec); err != nil {
if _, ok := err.(*sqlgraph.NotFoundError); ok {
err = &NotFoundError{extension.Label}
} else if sqlgraph.IsConstraintError(err) {
err = &ConstraintError{msg: err.Error(), wrap: err}
}
return 0, err
}
eu.mutation.done = true
return n, nil
}
// ExtensionUpdateOne is the builder for updating a single Extension entity.
type ExtensionUpdateOne struct {
config
fields []string
hooks []Hook
mutation *ExtensionMutation
modifiers []func(*sql.UpdateBuilder)
}
// SetVersion sets the "version" field.
func (euo *ExtensionUpdateOne) SetVersion(s string) *ExtensionUpdateOne {
euo.mutation.SetVersion(s)
return euo
}
// SetNillableVersion sets the "version" field if the given value is not nil.
func (euo *ExtensionUpdateOne) SetNillableVersion(s *string) *ExtensionUpdateOne {
if s != nil {
euo.SetVersion(*s)
}
return euo
}
// SetPath sets the "path" field.
func (euo *ExtensionUpdateOne) SetPath(s string) *ExtensionUpdateOne {
euo.mutation.SetPath(s)
return euo
}
// SetNillablePath sets the "path" field if the given value is not nil.
func (euo *ExtensionUpdateOne) SetNillablePath(s *string) *ExtensionUpdateOne {
if s != nil {
euo.SetPath(*s)
}
return euo
}
// SetCreatedAt sets the "created_at" field.
func (euo *ExtensionUpdateOne) SetCreatedAt(t time.Time) *ExtensionUpdateOne {
euo.mutation.SetCreatedAt(t)
return euo
}
// SetNillableCreatedAt sets the "created_at" field if the given value is not nil.
func (euo *ExtensionUpdateOne) SetNillableCreatedAt(t *time.Time) *ExtensionUpdateOne {
if t != nil {
euo.SetCreatedAt(*t)
}
return euo
}
// Mutation returns the ExtensionMutation object of the builder.
func (euo *ExtensionUpdateOne) Mutation() *ExtensionMutation {
return euo.mutation
}
// Where appends a list predicates to the ExtensionUpdate builder.
func (euo *ExtensionUpdateOne) Where(ps ...predicate.Extension) *ExtensionUpdateOne {
euo.mutation.Where(ps...)
return euo
}
// Select allows selecting one or more fields (columns) of the returned entity.
// The default is selecting all fields defined in the entity schema.
func (euo *ExtensionUpdateOne) Select(field string, fields ...string) *ExtensionUpdateOne {
euo.fields = append([]string{field}, fields...)
return euo
}
// Save executes the query and returns the updated Extension entity.
func (euo *ExtensionUpdateOne) Save(ctx context.Context) (*Extension, error) {
return withHooks(ctx, euo.sqlSave, euo.mutation, euo.hooks)
}
// SaveX is like Save, but panics if an error occurs.
func (euo *ExtensionUpdateOne) SaveX(ctx context.Context) *Extension {
node, err := euo.Save(ctx)
if err != nil {
panic(err)
}
return node
}
// Exec executes the query on the entity.
func (euo *ExtensionUpdateOne) Exec(ctx context.Context) error {
_, err := euo.Save(ctx)
return err
}
// ExecX is like Exec, but panics if an error occurs.
func (euo *ExtensionUpdateOne) ExecX(ctx context.Context) {
if err := euo.Exec(ctx); err != nil {
panic(err)
}
}
// Modify adds a statement modifier for attaching custom logic to the UPDATE statement.
func (euo *ExtensionUpdateOne) Modify(modifiers ...func(u *sql.UpdateBuilder)) *ExtensionUpdateOne {
euo.modifiers = append(euo.modifiers, modifiers...)
return euo
}
func (euo *ExtensionUpdateOne) sqlSave(ctx context.Context) (_node *Extension, err error) {
_spec := sqlgraph.NewUpdateSpec(extension.Table, extension.Columns, sqlgraph.NewFieldSpec(extension.FieldID, field.TypeUUID))
id, ok := euo.mutation.ID()
if !ok {
return nil, &ValidationError{Name: "id", err: errors.New(`db: missing "Extension.id" for update`)}
}
_spec.Node.ID.Value = id
if fields := euo.fields; len(fields) > 0 {
_spec.Node.Columns = make([]string, 0, len(fields))
_spec.Node.Columns = append(_spec.Node.Columns, extension.FieldID)
for _, f := range fields {
if !extension.ValidColumn(f) {
return nil, &ValidationError{Name: f, err: fmt.Errorf("db: invalid field %q for query", f)}
}
if f != extension.FieldID {
_spec.Node.Columns = append(_spec.Node.Columns, f)
}
}
}
if ps := euo.mutation.predicates; len(ps) > 0 {
_spec.Predicate = func(selector *sql.Selector) {
for i := range ps {
ps[i](selector)
}
}
}
if value, ok := euo.mutation.Version(); ok {
_spec.SetField(extension.FieldVersion, field.TypeString, value)
}
if value, ok := euo.mutation.Path(); ok {
_spec.SetField(extension.FieldPath, field.TypeString, value)
}
if value, ok := euo.mutation.CreatedAt(); ok {
_spec.SetField(extension.FieldCreatedAt, field.TypeTime, value)
}
_spec.AddModifiers(euo.modifiers...)
_node = &Extension{config: euo.config}
_spec.Assign = _node.assignValues
_spec.ScanValues = _node.scanValues
if err = sqlgraph.UpdateNode(ctx, euo.driver, _spec); err != nil {
if _, ok := err.(*sqlgraph.NotFoundError); ok {
err = &NotFoundError{extension.Label}
} else if sqlgraph.IsConstraintError(err) {
err = &ConstraintError{msg: err.Error(), wrap: err}
}
return nil, err
}
euo.mutation.done = true
return _node, nil
}

View File

@@ -93,6 +93,18 @@ func (f BillingUsageFunc) Mutate(ctx context.Context, m db.Mutation) (db.Value,
return nil, fmt.Errorf("unexpected mutation type %T. expect *db.BillingUsageMutation", m)
}
// The ExtensionFunc type is an adapter to allow the use of ordinary
// function as Extension mutator.
type ExtensionFunc func(context.Context, *db.ExtensionMutation) (db.Value, error)
// Mutate calls f(ctx, m).
func (f ExtensionFunc) Mutate(ctx context.Context, m db.Mutation) (db.Value, error) {
if mv, ok := m.(*db.ExtensionMutation); ok {
return f(ctx, mv)
}
return nil, fmt.Errorf("unexpected mutation type %T. expect *db.ExtensionMutation", m)
}
// The InviteCodeFunc type is an adapter to allow the use of ordinary
// function as InviteCode mutator.
type InviteCodeFunc func(context.Context, *db.InviteCodeMutation) (db.Value, error)

View File

@@ -15,6 +15,7 @@ import (
"github.com/chaitin/MonkeyCode/backend/db/billingquota"
"github.com/chaitin/MonkeyCode/backend/db/billingrecord"
"github.com/chaitin/MonkeyCode/backend/db/billingusage"
"github.com/chaitin/MonkeyCode/backend/db/extension"
"github.com/chaitin/MonkeyCode/backend/db/invitecode"
"github.com/chaitin/MonkeyCode/backend/db/model"
"github.com/chaitin/MonkeyCode/backend/db/modelprovider"
@@ -273,6 +274,33 @@ func (f TraverseBillingUsage) Traverse(ctx context.Context, q db.Query) error {
return fmt.Errorf("unexpected query type %T. expect *db.BillingUsageQuery", q)
}
// The ExtensionFunc type is an adapter to allow the use of ordinary function as a Querier.
type ExtensionFunc func(context.Context, *db.ExtensionQuery) (db.Value, error)
// Query calls f(ctx, q).
func (f ExtensionFunc) Query(ctx context.Context, q db.Query) (db.Value, error) {
if q, ok := q.(*db.ExtensionQuery); ok {
return f(ctx, q)
}
return nil, fmt.Errorf("unexpected query type %T. expect *db.ExtensionQuery", q)
}
// The TraverseExtension type is an adapter to allow the use of ordinary function as Traverser.
type TraverseExtension func(context.Context, *db.ExtensionQuery) error
// Intercept is a dummy implementation of Intercept that returns the next Querier in the pipeline.
func (f TraverseExtension) Intercept(next db.Querier) db.Querier {
return next
}
// Traverse calls f(ctx, q).
func (f TraverseExtension) Traverse(ctx context.Context, q db.Query) error {
if q, ok := q.(*db.ExtensionQuery); ok {
return f(ctx, q)
}
return fmt.Errorf("unexpected query type %T. expect *db.ExtensionQuery", q)
}
// The InviteCodeFunc type is an adapter to allow the use of ordinary function as a Querier.
type InviteCodeFunc func(context.Context, *db.InviteCodeQuery) (db.Value, error)
@@ -560,6 +588,8 @@ func NewQuery(q db.Query) (Query, error) {
return &query[*db.BillingRecordQuery, predicate.BillingRecord, billingrecord.OrderOption]{typ: db.TypeBillingRecord, tq: q}, nil
case *db.BillingUsageQuery:
return &query[*db.BillingUsageQuery, predicate.BillingUsage, billingusage.OrderOption]{typ: db.TypeBillingUsage, tq: q}, nil
case *db.ExtensionQuery:
return &query[*db.ExtensionQuery, predicate.Extension, extension.OrderOption]{typ: db.TypeExtension, tq: q}, nil
case *db.InviteCodeQuery:
return &query[*db.InviteCodeQuery, predicate.InviteCode, invitecode.OrderOption]{typ: db.TypeInviteCode, tq: q}, nil
case *db.ModelQuery:

View File

@@ -141,6 +141,19 @@ var (
Columns: BillingUsagesColumns,
PrimaryKey: []*schema.Column{BillingUsagesColumns[0]},
}
// ExtensionsColumns holds the columns for the "extensions" table.
ExtensionsColumns = []*schema.Column{
{Name: "id", Type: field.TypeUUID},
{Name: "version", Type: field.TypeString},
{Name: "path", Type: field.TypeString},
{Name: "created_at", Type: field.TypeTime},
}
// ExtensionsTable holds the schema information for the "extensions" table.
ExtensionsTable = &schema.Table{
Name: "extensions",
Columns: ExtensionsColumns,
PrimaryKey: []*schema.Column{ExtensionsColumns[0]},
}
// InviteCodesColumns holds the columns for the "invite_codes" table.
InviteCodesColumns = []*schema.Column{
{Name: "id", Type: field.TypeUUID},
@@ -387,6 +400,7 @@ var (
BillingQuotasTable,
BillingRecordsTable,
BillingUsagesTable,
ExtensionsTable,
InviteCodesTable,
ModelsTable,
ModelProvidersTable,
@@ -423,6 +437,9 @@ func init() {
BillingUsagesTable.Annotation = &entsql.Annotation{
Table: "billing_usages",
}
ExtensionsTable.Annotation = &entsql.Annotation{
Table: "extensions",
}
InviteCodesTable.Annotation = &entsql.Annotation{
Table: "invite_codes",
}

View File

@@ -19,6 +19,7 @@ import (
"github.com/chaitin/MonkeyCode/backend/db/billingquota"
"github.com/chaitin/MonkeyCode/backend/db/billingrecord"
"github.com/chaitin/MonkeyCode/backend/db/billingusage"
"github.com/chaitin/MonkeyCode/backend/db/extension"
"github.com/chaitin/MonkeyCode/backend/db/invitecode"
"github.com/chaitin/MonkeyCode/backend/db/model"
"github.com/chaitin/MonkeyCode/backend/db/modelprovider"
@@ -49,6 +50,7 @@ const (
TypeBillingQuota = "BillingQuota"
TypeBillingRecord = "BillingRecord"
TypeBillingUsage = "BillingUsage"
TypeExtension = "Extension"
TypeInviteCode = "InviteCode"
TypeModel = "Model"
TypeModelProvider = "ModelProvider"
@@ -5335,6 +5337,446 @@ func (m *BillingUsageMutation) ResetEdge(name string) error {
return fmt.Errorf("unknown BillingUsage edge %s", name)
}
// ExtensionMutation represents an operation that mutates the Extension nodes in the graph.
type ExtensionMutation struct {
config
op Op
typ string
id *uuid.UUID
version *string
_path *string
created_at *time.Time
clearedFields map[string]struct{}
done bool
oldValue func(context.Context) (*Extension, error)
predicates []predicate.Extension
}
var _ ent.Mutation = (*ExtensionMutation)(nil)
// extensionOption allows management of the mutation configuration using functional options.
type extensionOption func(*ExtensionMutation)
// newExtensionMutation creates new mutation for the Extension entity.
func newExtensionMutation(c config, op Op, opts ...extensionOption) *ExtensionMutation {
m := &ExtensionMutation{
config: c,
op: op,
typ: TypeExtension,
clearedFields: make(map[string]struct{}),
}
for _, opt := range opts {
opt(m)
}
return m
}
// withExtensionID sets the ID field of the mutation.
func withExtensionID(id uuid.UUID) extensionOption {
return func(m *ExtensionMutation) {
var (
err error
once sync.Once
value *Extension
)
m.oldValue = func(ctx context.Context) (*Extension, error) {
once.Do(func() {
if m.done {
err = errors.New("querying old values post mutation is not allowed")
} else {
value, err = m.Client().Extension.Get(ctx, id)
}
})
return value, err
}
m.id = &id
}
}
// withExtension sets the old Extension of the mutation.
func withExtension(node *Extension) extensionOption {
return func(m *ExtensionMutation) {
m.oldValue = func(context.Context) (*Extension, error) {
return node, nil
}
m.id = &node.ID
}
}
// Client returns a new `ent.Client` from the mutation. If the mutation was
// executed in a transaction (ent.Tx), a transactional client is returned.
func (m ExtensionMutation) Client() *Client {
client := &Client{config: m.config}
client.init()
return client
}
// Tx returns an `ent.Tx` for mutations that were executed in transactions;
// it returns an error otherwise.
func (m ExtensionMutation) Tx() (*Tx, error) {
if _, ok := m.driver.(*txDriver); !ok {
return nil, errors.New("db: mutation is not running in a transaction")
}
tx := &Tx{config: m.config}
tx.init()
return tx, nil
}
// SetID sets the value of the id field. Note that this
// operation is only accepted on creation of Extension entities.
func (m *ExtensionMutation) SetID(id uuid.UUID) {
m.id = &id
}
// ID returns the ID value in the mutation. Note that the ID is only available
// if it was provided to the builder or after it was returned from the database.
func (m *ExtensionMutation) ID() (id uuid.UUID, exists bool) {
if m.id == nil {
return
}
return *m.id, true
}
// IDs queries the database and returns the entity ids that match the mutation's predicate.
// That means, if the mutation is applied within a transaction with an isolation level such
// as sql.LevelSerializable, the returned ids match the ids of the rows that will be updated
// or updated by the mutation.
func (m *ExtensionMutation) IDs(ctx context.Context) ([]uuid.UUID, error) {
switch {
case m.op.Is(OpUpdateOne | OpDeleteOne):
id, exists := m.ID()
if exists {
return []uuid.UUID{id}, nil
}
fallthrough
case m.op.Is(OpUpdate | OpDelete):
return m.Client().Extension.Query().Where(m.predicates...).IDs(ctx)
default:
return nil, fmt.Errorf("IDs is not allowed on %s operations", m.op)
}
}
// SetVersion sets the "version" field.
func (m *ExtensionMutation) SetVersion(s string) {
m.version = &s
}
// Version returns the value of the "version" field in the mutation.
func (m *ExtensionMutation) Version() (r string, exists bool) {
v := m.version
if v == nil {
return
}
return *v, true
}
// OldVersion returns the old "version" field's value of the Extension entity.
// If the Extension object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *ExtensionMutation) OldVersion(ctx context.Context) (v string, err error) {
if !m.op.Is(OpUpdateOne) {
return v, errors.New("OldVersion is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, errors.New("OldVersion requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldVersion: %w", err)
}
return oldValue.Version, nil
}
// ResetVersion resets all changes to the "version" field.
func (m *ExtensionMutation) ResetVersion() {
m.version = nil
}
// SetPath sets the "path" field.
func (m *ExtensionMutation) SetPath(s string) {
m._path = &s
}
// Path returns the value of the "path" field in the mutation.
func (m *ExtensionMutation) Path() (r string, exists bool) {
v := m._path
if v == nil {
return
}
return *v, true
}
// OldPath returns the old "path" field's value of the Extension entity.
// If the Extension object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *ExtensionMutation) OldPath(ctx context.Context) (v string, err error) {
if !m.op.Is(OpUpdateOne) {
return v, errors.New("OldPath is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, errors.New("OldPath requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldPath: %w", err)
}
return oldValue.Path, nil
}
// ResetPath resets all changes to the "path" field.
func (m *ExtensionMutation) ResetPath() {
m._path = nil
}
// SetCreatedAt sets the "created_at" field.
func (m *ExtensionMutation) SetCreatedAt(t time.Time) {
m.created_at = &t
}
// CreatedAt returns the value of the "created_at" field in the mutation.
func (m *ExtensionMutation) CreatedAt() (r time.Time, exists bool) {
v := m.created_at
if v == nil {
return
}
return *v, true
}
// OldCreatedAt returns the old "created_at" field's value of the Extension entity.
// If the Extension object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *ExtensionMutation) OldCreatedAt(ctx context.Context) (v time.Time, err error) {
if !m.op.Is(OpUpdateOne) {
return v, errors.New("OldCreatedAt is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, errors.New("OldCreatedAt requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldCreatedAt: %w", err)
}
return oldValue.CreatedAt, nil
}
// ResetCreatedAt resets all changes to the "created_at" field.
func (m *ExtensionMutation) ResetCreatedAt() {
m.created_at = nil
}
// Where appends a list predicates to the ExtensionMutation builder.
func (m *ExtensionMutation) Where(ps ...predicate.Extension) {
m.predicates = append(m.predicates, ps...)
}
// WhereP appends storage-level predicates to the ExtensionMutation builder. Using this method,
// users can use type-assertion to append predicates that do not depend on any generated package.
func (m *ExtensionMutation) WhereP(ps ...func(*sql.Selector)) {
p := make([]predicate.Extension, len(ps))
for i := range ps {
p[i] = ps[i]
}
m.Where(p...)
}
// Op returns the operation name.
func (m *ExtensionMutation) Op() Op {
return m.op
}
// SetOp allows setting the mutation operation.
func (m *ExtensionMutation) SetOp(op Op) {
m.op = op
}
// Type returns the node type of this mutation (Extension).
func (m *ExtensionMutation) Type() string {
return m.typ
}
// Fields returns all fields that were changed during this mutation. Note that in
// order to get all numeric fields that were incremented/decremented, call
// AddedFields().
func (m *ExtensionMutation) Fields() []string {
fields := make([]string, 0, 3)
if m.version != nil {
fields = append(fields, extension.FieldVersion)
}
if m._path != nil {
fields = append(fields, extension.FieldPath)
}
if m.created_at != nil {
fields = append(fields, extension.FieldCreatedAt)
}
return fields
}
// Field returns the value of a field with the given name. The second boolean
// return value indicates that this field was not set, or was not defined in the
// schema.
func (m *ExtensionMutation) Field(name string) (ent.Value, bool) {
switch name {
case extension.FieldVersion:
return m.Version()
case extension.FieldPath:
return m.Path()
case extension.FieldCreatedAt:
return m.CreatedAt()
}
return nil, false
}
// OldField returns the old value of the field from the database. An error is
// returned if the mutation operation is not UpdateOne, or the query to the
// database failed.
func (m *ExtensionMutation) OldField(ctx context.Context, name string) (ent.Value, error) {
switch name {
case extension.FieldVersion:
return m.OldVersion(ctx)
case extension.FieldPath:
return m.OldPath(ctx)
case extension.FieldCreatedAt:
return m.OldCreatedAt(ctx)
}
return nil, fmt.Errorf("unknown Extension field %s", name)
}
// SetField sets the value of a field with the given name. It returns an error if
// the field is not defined in the schema, or if the type mismatched the field
// type.
func (m *ExtensionMutation) SetField(name string, value ent.Value) error {
switch name {
case extension.FieldVersion:
v, ok := value.(string)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetVersion(v)
return nil
case extension.FieldPath:
v, ok := value.(string)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetPath(v)
return nil
case extension.FieldCreatedAt:
v, ok := value.(time.Time)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetCreatedAt(v)
return nil
}
return fmt.Errorf("unknown Extension field %s", name)
}
// AddedFields returns all numeric fields that were incremented/decremented during
// this mutation.
func (m *ExtensionMutation) AddedFields() []string {
return nil
}
// AddedField returns the numeric value that was incremented/decremented on a field
// with the given name. The second boolean return value indicates that this field
// was not set, or was not defined in the schema.
func (m *ExtensionMutation) AddedField(name string) (ent.Value, bool) {
return nil, false
}
// AddField adds the value to the field with the given name. It returns an error if
// the field is not defined in the schema, or if the type mismatched the field
// type.
func (m *ExtensionMutation) AddField(name string, value ent.Value) error {
switch name {
}
return fmt.Errorf("unknown Extension numeric field %s", name)
}
// ClearedFields returns all nullable fields that were cleared during this
// mutation.
func (m *ExtensionMutation) ClearedFields() []string {
return nil
}
// FieldCleared returns a boolean indicating if a field with the given name was
// cleared in this mutation.
func (m *ExtensionMutation) FieldCleared(name string) bool {
_, ok := m.clearedFields[name]
return ok
}
// ClearField clears the value of the field with the given name. It returns an
// error if the field is not defined in the schema.
func (m *ExtensionMutation) ClearField(name string) error {
return fmt.Errorf("unknown Extension nullable field %s", name)
}
// ResetField resets all changes in the mutation for the field with the given name.
// It returns an error if the field is not defined in the schema.
func (m *ExtensionMutation) ResetField(name string) error {
switch name {
case extension.FieldVersion:
m.ResetVersion()
return nil
case extension.FieldPath:
m.ResetPath()
return nil
case extension.FieldCreatedAt:
m.ResetCreatedAt()
return nil
}
return fmt.Errorf("unknown Extension field %s", name)
}
// AddedEdges returns all edge names that were set/added in this mutation.
func (m *ExtensionMutation) AddedEdges() []string {
edges := make([]string, 0, 0)
return edges
}
// AddedIDs returns all IDs (to other nodes) that were added for the given edge
// name in this mutation.
func (m *ExtensionMutation) AddedIDs(name string) []ent.Value {
return nil
}
// RemovedEdges returns all edge names that were removed in this mutation.
func (m *ExtensionMutation) RemovedEdges() []string {
edges := make([]string, 0, 0)
return edges
}
// RemovedIDs returns all IDs (to other nodes) that were removed for the edge with
// the given name in this mutation.
func (m *ExtensionMutation) RemovedIDs(name string) []ent.Value {
return nil
}
// ClearedEdges returns all edge names that were cleared in this mutation.
func (m *ExtensionMutation) ClearedEdges() []string {
edges := make([]string, 0, 0)
return edges
}
// EdgeCleared returns a boolean which indicates if the edge with the given name
// was cleared in this mutation.
func (m *ExtensionMutation) EdgeCleared(name string) bool {
return false
}
// ClearEdge clears the value of the edge with the given name. It returns an error
// if that edge is not defined in the schema.
func (m *ExtensionMutation) ClearEdge(name string) error {
return fmt.Errorf("unknown Extension unique edge %s", name)
}
// ResetEdge resets all changes to the edge with the given name in this mutation.
// It returns an error if the edge is not defined in the schema.
func (m *ExtensionMutation) ResetEdge(name string) error {
return fmt.Errorf("unknown Extension edge %s", name)
}
// InviteCodeMutation represents an operation that mutates the InviteCode nodes in the graph.
type InviteCodeMutation struct {
config

View File

@@ -109,6 +109,20 @@ func (bu *BillingUsageQuery) Page(ctx context.Context, page, size int) ([]*Billi
return rs, &PageInfo{HasNextPage: has, TotalCount: int64(cnt)}, nil
}
func (e *ExtensionQuery) Page(ctx context.Context, page, size int) ([]*Extension, *PageInfo, error) {
cnt, err := e.Count(ctx)
if err != nil {
return nil, nil, err
}
offset := size * (page - 1)
rs, err := e.Offset(offset).Limit(size).All(ctx)
if err != nil {
return nil, nil, err
}
has := (page * size) < cnt
return rs, &PageInfo{HasNextPage: has, TotalCount: int64(cnt)}, nil
}
func (ic *InviteCodeQuery) Page(ctx context.Context, page, size int) ([]*InviteCode, *PageInfo, error) {
cnt, err := ic.Count(ctx)
if err != nil {

View File

@@ -27,6 +27,9 @@ type BillingRecord func(*sql.Selector)
// BillingUsage is the predicate function for billingusage builders.
type BillingUsage func(*sql.Selector)
// Extension is the predicate function for extension builders.
type Extension func(*sql.Selector)
// InviteCode is the predicate function for invitecode builders.
type InviteCode func(*sql.Selector)

View File

@@ -13,6 +13,7 @@ import (
"github.com/chaitin/MonkeyCode/backend/db/billingquota"
"github.com/chaitin/MonkeyCode/backend/db/billingrecord"
"github.com/chaitin/MonkeyCode/backend/db/billingusage"
"github.com/chaitin/MonkeyCode/backend/db/extension"
"github.com/chaitin/MonkeyCode/backend/db/invitecode"
"github.com/chaitin/MonkeyCode/backend/db/model"
"github.com/chaitin/MonkeyCode/backend/db/modelprovider"
@@ -131,6 +132,12 @@ func init() {
billingusage.DefaultUpdatedAt = billingusageDescUpdatedAt.Default.(func() time.Time)
// billingusage.UpdateDefaultUpdatedAt holds the default value on update for the updated_at field.
billingusage.UpdateDefaultUpdatedAt = billingusageDescUpdatedAt.UpdateDefault.(func() time.Time)
extensionFields := schema.Extension{}.Fields()
_ = extensionFields
// extensionDescCreatedAt is the schema descriptor for created_at field.
extensionDescCreatedAt := extensionFields[3].Descriptor()
// extension.DefaultCreatedAt holds the default value on creation for the created_at field.
extension.DefaultCreatedAt = extensionDescCreatedAt.Default.(func() time.Time)
invitecodeFields := schema.InviteCode{}.Fields()
_ = invitecodeFields
// invitecodeDescCreatedAt is the schema descriptor for created_at field.

View File

@@ -28,6 +28,8 @@ type Tx struct {
BillingRecord *BillingRecordClient
// BillingUsage is the client for interacting with the BillingUsage builders.
BillingUsage *BillingUsageClient
// Extension is the client for interacting with the Extension builders.
Extension *ExtensionClient
// InviteCode is the client for interacting with the InviteCode builders.
InviteCode *InviteCodeClient
// Model is the client for interacting with the Model builders.
@@ -186,6 +188,7 @@ func (tx *Tx) init() {
tx.BillingQuota = NewBillingQuotaClient(tx.config)
tx.BillingRecord = NewBillingRecordClient(tx.config)
tx.BillingUsage = NewBillingUsageClient(tx.config)
tx.Extension = NewExtensionClient(tx.config)
tx.InviteCode = NewInviteCodeClient(tx.config)
tx.Model = NewModelClient(tx.config)
tx.ModelProvider = NewModelProviderClient(tx.config)

View File

@@ -0,0 +1,38 @@
package domain
import (
"context"
"github.com/google/uuid"
"github.com/chaitin/MonkeyCode/backend/db"
)
type ExtensionUsecase interface {
Latest(ctx context.Context) (*Extension, error)
GetByVersion(ctx context.Context, version string) (*Extension, error)
}
type ExtensionRepo interface {
Latest(ctx context.Context) (*db.Extension, error)
Save(ctx context.Context, ext *db.Extension) (*db.Extension, error)
GetByVersion(ctx context.Context, version string) (*db.Extension, error)
}
type Extension struct {
ID uuid.UUID `json:"id"`
Version string `json:"version"`
Path string `json:"path"`
CreatedAt int64 `json:"created_at"`
}
func (et *Extension) From(e *db.Extension) *Extension {
if e == nil {
return et
}
et.ID = e.ID
et.Version = e.Version
et.Path = e.Path
et.CreatedAt = e.CreatedAt.Unix()
return et
}

View File

@@ -32,6 +32,11 @@ type ProxyRepo interface {
ValidateApiKey(ctx context.Context, key string) (*db.ApiKey, error)
}
type VersionInfo struct {
Version string `json:"version"`
URL string `json:"url"`
}
type AcceptCompletionReq struct {
ID string `json:"id"` // 记录ID
Completion string `json:"completion"` // 补全内容

View File

@@ -0,0 +1,39 @@
package schema
import (
"time"
"entgo.io/ent"
"entgo.io/ent/dialect/entsql"
"entgo.io/ent/schema"
"entgo.io/ent/schema/field"
"github.com/google/uuid"
)
// Extension holds the schema definition for the Extension entity.
type Extension struct {
ent.Schema
}
func (Extension) Annotations() []schema.Annotation {
return []schema.Annotation{
entsql.Annotation{
Table: "extensions",
},
}
}
// Fields of the Extension.
func (Extension) Fields() []ent.Field {
return []ent.Field{
field.UUID("id", uuid.UUID{}),
field.String("version"),
field.String("path"),
field.Time("created_at").Default(time.Now),
}
}
// Edges of the Extension.
func (Extension) Edges() []ent.Edge {
return nil
}

View File

@@ -0,0 +1,42 @@
package repo
import (
"context"
"entgo.io/ent/dialect/sql"
"github.com/chaitin/MonkeyCode/backend/db"
"github.com/chaitin/MonkeyCode/backend/db/extension"
"github.com/chaitin/MonkeyCode/backend/domain"
)
type ExtensionRepo struct {
db *db.Client
}
func NewExtensionRepo(db *db.Client) domain.ExtensionRepo {
return &ExtensionRepo{
db: db,
}
}
// Latest implements domain.ExtensionRepo.
func (e *ExtensionRepo) Latest(ctx context.Context) (*db.Extension, error) {
return e.db.Extension.
Query().
Order(extension.ByCreatedAt(sql.OrderDesc())).
Limit(1).
Only(ctx)
}
// Save implements domain.ExtensionRepo.
func (e *ExtensionRepo) Save(ctx context.Context, ext *db.Extension) (*db.Extension, error) {
return e.db.Extension.Create().
SetVersion(ext.Version).
SetPath(ext.Path).
Save(ctx)
}
func (e *ExtensionRepo) GetByVersion(ctx context.Context, version string) (*db.Extension, error) {
return e.db.Extension.Query().Where(extension.Version(version)).Only(ctx)
}

View File

@@ -0,0 +1,146 @@
package usecase
import (
"context"
"fmt"
"io"
"log/slog"
"net/http"
"os"
"sync"
"time"
"github.com/chaitin/MonkeyCode/backend/config"
"github.com/chaitin/MonkeyCode/backend/db"
"github.com/chaitin/MonkeyCode/backend/domain"
"github.com/chaitin/MonkeyCode/backend/pkg/cvt"
)
type ExtensionUsecase struct {
logger *slog.Logger
repo domain.ExtensionRepo
config *config.Config
version string
mu sync.Mutex
}
func NewExtensionUsecase(
repo domain.ExtensionRepo,
config *config.Config,
logger *slog.Logger,
) domain.ExtensionUsecase {
e := &ExtensionUsecase{
repo: repo,
config: config,
mu: sync.Mutex{},
logger: logger,
}
go e.syncLatest()
return e
}
// GetByVersion implements domain.ExtensionUsecase.
func (e *ExtensionUsecase) GetByVersion(ctx context.Context, version string) (*domain.Extension, error) {
ee, err := e.repo.GetByVersion(ctx, version)
if err != nil {
return nil, err
}
return cvt.From(ee, &domain.Extension{}), nil
}
// Latest implements domain.ExtensionUsecase.
func (e *ExtensionUsecase) Latest(ctx context.Context) (*domain.Extension, error) {
ee, err := e.repo.Latest(ctx)
if err != nil {
return nil, err
}
return cvt.From(ee, &domain.Extension{}), nil
}
func (e *ExtensionUsecase) syncLatest() {
latest, err := e.repo.Latest(context.Background())
if err == nil {
e.version = latest.Version
}
logger := e.logger.With("fn", "syncLatest")
logger.With("version", e.version).Debug("开始同步插件版本信息")
t := time.NewTicker(1 * time.Minute)
e.innerSync()
for range t.C {
e.innerSync()
}
}
func (e *ExtensionUsecase) innerSync() {
logger := e.logger.With("fn", "syncLatest")
v, err := e.getRemoteVersion()
if err != nil {
logger.With("error", err).Error("获取远程插件版本信息失败")
return
}
if v == e.version {
return
}
e.download(v)
}
func (e *ExtensionUsecase) download(version string) {
logger := e.logger.With("fn", "download")
url := fmt.Sprintf("%s/monkeycode/vsixs/monkeycode-%s.vsix", e.config.Extension.Baseurl, version)
logger.With("url", url).With("version", version).Debug("发现新版本,开始下载")
resp, err := http.Get(url)
if err != nil {
logger.With("error", err).Error("下载插件失败")
return
}
defer resp.Body.Close()
filename := fmt.Sprintf("/app/static/monkeycode-%s.vsix", version)
f, err := os.OpenFile(filename, os.O_WRONLY|os.O_CREATE, 0644)
if err != nil {
logger.With("error", err).Error("创建文件失败")
return
}
defer f.Close()
_, err = io.Copy(f, resp.Body)
if err != nil {
logger.With("error", err).Error("复制文件内容失败")
return
}
logger.Debug("下载插件成功")
if _, err := e.repo.Save(context.Background(), &db.Extension{
Version: version,
Path: filename,
}); err != nil {
logger.With("error", err).Error("保存插件版本信息失败")
os.Remove(filename)
return
}
e.version = version
}
func (e *ExtensionUsecase) getRemoteVersion() (string, error) {
logger := e.logger.With("fn", "getRemoteVersion")
u := fmt.Sprintf("%s/monkeycode/vsix_version.txt", e.config.Extension.Baseurl)
logger.With("url", u).Debug("开始获取远程插件版本信息")
resp, err := http.Get(u)
if err != nil {
logger.With("error", err).Error("获取插件版本信息失败")
return "", err
}
defer resp.Body.Close()
version, err := io.ReadAll(resp.Body)
if err != nil {
logger.With("error", err).Error("读取插件body信息失败")
return "", err
}
logger.With("version", string(version)).Debug("读取到的插件版本信息")
return string(version), nil
}

View File

@@ -1,6 +1,7 @@
package v1
import (
"fmt"
"log/slog"
"net/http"
@@ -9,6 +10,7 @@ import (
"github.com/GoYoko/web"
"github.com/chaitin/MonkeyCode/backend/config"
"github.com/chaitin/MonkeyCode/backend/domain"
"github.com/chaitin/MonkeyCode/backend/internal/middleware"
)
@@ -17,6 +19,8 @@ type V1Handler struct {
logger *slog.Logger
proxy domain.Proxy
usecase domain.OpenAIUsecase
euse domain.ExtensionUsecase
config *config.Config
}
func NewV1Handler(
@@ -24,15 +28,20 @@ func NewV1Handler(
w *web.Web,
proxy domain.Proxy,
usecase domain.OpenAIUsecase,
euse domain.ExtensionUsecase,
middleware *middleware.ProxyMiddleware,
active *middleware.ActiveMiddleware,
config *config.Config,
) *V1Handler {
h := &V1Handler{
logger: logger.With(slog.String("handler", "openai")),
proxy: proxy,
usecase: usecase,
euse: euse,
config: config,
}
w.GET("/api/config", web.BindHandler(h.GetConfig), middleware.Auth())
w.GET("/v1/version", web.BaseHandler(h.Version), middleware.Auth())
g := w.Group("/v1", middleware.Auth())
g.GET("/models", web.BaseHandler(h.ModelList))
@@ -53,6 +62,18 @@ func BadRequest(c *web.Context, msg string) error {
return nil
}
func (h *V1Handler) Version(c *web.Context) error {
v, err := h.euse.Latest(c.Request().Context())
if err != nil {
return err
}
return c.JSON(http.StatusOK, domain.VersionInfo{
Version: v.Version,
URL: fmt.Sprintf("%s/api/v1/static/vsix/%s", h.config.BaseUrl, v.Version),
})
}
// AcceptCompletion 接受补全请求
//
// @Tags OpenAIV1

View File

@@ -9,6 +9,8 @@ import (
dashv1 "github.com/chaitin/MonkeyCode/backend/internal/dashboard/handler/v1"
dashrepo "github.com/chaitin/MonkeyCode/backend/internal/dashboard/repo"
dashusecase "github.com/chaitin/MonkeyCode/backend/internal/dashboard/usecase"
erepo "github.com/chaitin/MonkeyCode/backend/internal/extension/repo"
eusecase "github.com/chaitin/MonkeyCode/backend/internal/extension/usecase"
"github.com/chaitin/MonkeyCode/backend/internal/middleware"
modelv1 "github.com/chaitin/MonkeyCode/backend/internal/model/handler/http/v1"
modelrepo "github.com/chaitin/MonkeyCode/backend/internal/model/repo"
@@ -46,4 +48,6 @@ var Provider = wire.NewSet(
billingv1.NewBillingHandler,
billingrepo.NewBillingRepo,
billingusecase.NewBillingUsecase,
erepo.NewExtensionRepo,
eusecase.NewExtensionUsecase,
)

View File

@@ -2,6 +2,7 @@ package v1
import (
"context"
"fmt"
"log/slog"
"net/http"
@@ -18,6 +19,7 @@ import (
type UserHandler struct {
usecase domain.UserUsecase
euse domain.ExtensionUsecase
session *session.Session
logger *slog.Logger
cfg *config.Config
@@ -26,6 +28,7 @@ type UserHandler struct {
func NewUserHandler(
w *web.Web,
usecase domain.UserUsecase,
euse domain.ExtensionUsecase,
auth *middleware.AuthMiddleware,
session *session.Session,
logger *slog.Logger,
@@ -36,9 +39,10 @@ func NewUserHandler(
session: session,
logger: logger,
cfg: cfg,
euse: euse,
}
w.GET("/api/v1/static/vsix", web.BaseHandler(u.VSIXDownload))
w.GET("/api/v1/static/vsix/:version", web.BaseHandler(u.VSIXDownload))
w.POST("/api/v1/vscode/init-auth", web.BindHandler(u.VSCodeAuthInit))
// admin
@@ -88,9 +92,14 @@ func (h *UserHandler) VSCodeAuthInit(c *web.Context, req domain.VSCodeAuthInitRe
// @Produce octet-stream
// @Router /api/v1/static/vsix [get]
func (h *UserHandler) VSIXDownload(c *web.Context) error {
v, err := h.euse.GetByVersion(c.Request().Context(), c.Param("version"))
if err != nil {
return err
}
disposition := fmt.Sprintf("attachment; filename=monkeycode-%s.vsix", v.Version)
c.Response().Header().Set("Content-Type", "application/octet-stream")
c.Response().Header().Set("Content-Disposition", "attachment; filename=monkeycode.vsix")
if err := vsix.ChangeVsixEndpoint(h.cfg.VSCode.VSIXFile, "extension/package.json", h.cfg.BaseUrl, c.Response().Writer); err != nil {
c.Response().Header().Set("Content-Disposition", disposition)
if err := vsix.ChangeVsixEndpoint(v.Path, "extension/package.json", h.cfg.BaseUrl, c.Response().Writer); err != nil {
return err
}
return nil

View File

@@ -0,0 +1,8 @@
CREATE TABLE extensions (
id UUID PRIMARY KEY DEFAULT uuid_generate_v1(),
version VARCHAR(255) NOT NULL,
path VARCHAR(1024) NOT NULL,
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
);
CREATE UNIQUE INDEX unique_idx_extensions_version ON extensions(version);