chore: 初始化 Cloudreve 源码仓库

2026-09-13 11:28:59 +08:00
commit 2a7b03ab1d
489 changed files with 163867 additions and 0 deletions
@@ -0,0 +1,101 @@
+package schema
+
+import (
+	"context"
+	"time"
+
+	"entgo.io/ent"
+	"entgo.io/ent/dialect"
+	"entgo.io/ent/schema/edge"
+	"entgo.io/ent/schema/field"
+	"entgo.io/ent/schema/index"
+	"github.com/cloudreve/Cloudreve/v4/ent/hook"
+	"github.com/cloudreve/Cloudreve/v4/inventory/types"
+)
+
+// File holds the schema definition for the File entity.
+type File struct {
+	ent.Schema
+}
+
+// Fields of the File.
+func (File) Fields() []ent.Field {
+	return []ent.Field{
+		field.Time("created_at").
+			Immutable().
+			Default(time.Now).
+			SchemaType(map[string]string{
+				dialect.MySQL: "datetime",
+			}),
+		field.Time("updated_at").
+			Default(time.Now).
+			SchemaType(map[string]string{
+				dialect.MySQL: "datetime",
+			}),
+		field.Int("type"),
+		field.String("name"),
+		field.Int("owner_id"),
+		field.Int64("size").
+			Default(0),
+		field.Int("primary_entity").
+			Optional(),
+		field.Int("file_children").
+			Optional(),
+		field.Bool("is_symbolic").
+			Default(false),
+		field.JSON("props", &types.FileProps{}).Optional(),
+		field.Int("storage_policy_files").
+			Optional(),
+	}
+}
+
+// Edges of the File.
+func (File) Edges() []ent.Edge {
+	return []ent.Edge{
+		edge.From("owner", User.Type).
+			Ref("files").
+			Field("owner_id").
+			Unique().
+			Required(),
+		edge.From("storage_policies", StoragePolicy.Type).
+			Ref("files").
+			Field("storage_policy_files").
+			Unique(),
+		edge.To("children", File.Type).
+			From("parent").
+			Field("file_children").
+			Unique(),
+		edge.To("metadata", Metadata.Type),
+		edge.To("entities", Entity.Type),
+		edge.To("shares", Share.Type),
+		edge.To("direct_links", DirectLink.Type),
+	}
+}
+
+// Indexes of the File.
+func (File) Indexes() []ent.Index {
+	return []ent.Index{
+		index.Fields("file_children", "name").
+			Unique(),
+		index.Fields("file_children", "type", "updated_at"),
+		index.Fields("file_children", "type", "created_at"),
+		index.Fields("file_children", "type", "size"),
+	}
+}
+
+func (f File) Hooks() []ent.Hook {
+	return []ent.Hook{
+		hook.On(func(next ent.Mutator) ent.Mutator {
+			return ent.MutateFunc(func(ctx context.Context, m ent.Mutation) (ent.Value, error) {
+				if s, ok := m.(interface{ SetUpdatedAt(time.Time) }); ok {
+					_, set := m.Field("updated_at")
+					if !set {
+						s.SetUpdatedAt(time.Now())
+					}
+				}
+				v, err := next.Mutate(ctx, m)
+				return v, err
+			})
+		}, ent.OpUpdate|ent.OpUpdateOne),
+	}
+}