Add VirusTotal Results to API

conversion/ent_to_graphql.go

@@ -99,7 +99,7 @@ type Mod interface {
 // goverter:extend TimeToString UIntToInt Int64ToInt
 type Version interface {
 	// goverter:map Edges.Targets Targets
-	// goverter:ignore Link Mod Dependencies SmlVersion
+	// goverter:ignore Link Mod Dependencies SmlVersion VirustotalResults
 	Convert(source *ent.Version) *generated.Version
 	ConvertSlice(source []*ent.Version) []*generated.Version
 
@@ -118,6 +118,15 @@ type VersionDependency interface {
 	ConvertSlice(source []*ent.VersionDependency) []*generated.VersionDependency
 }
 
+// goverter:converter
+// goverter:output:file ../generated/conv/virustotal_result.go
+// goverter:output:package conv
+// goverter:extend TimeToString UIntToInt Int64ToInt
+type VirustotalResult interface {
+	Convert(source *ent.VirustotalResult) *generated.VirustotalResult
+	ConvertSlice(source []*ent.VirustotalResult) []*generated.VirustotalResult
+}
+
 func TimeToString(i time.Time) string {
 	return i.Format(time.RFC3339)
 }

db/schema/version.go

@@ -61,5 +61,6 @@
 		edge.To("dependencies", Mod.Type).
 			Through("version_dependencies", VersionDependency.Type),
 		edge.To("targets", VersionTarget.Type),
+		edge.To("virustotalResults", VirustotalResult.Type),
 	}
 }

db/schema/virustotal_result.go

@@ -0,0 +1,48 @@
+package schema
+
+import (
+	"entgo.io/ent"
+	"entgo.io/ent/dialect/entsql"
+	"entgo.io/ent/schema"
+	"entgo.io/ent/schema/edge"
+	"entgo.io/ent/schema/field"
+)
+
+// YourTableName holds the schema definition for the YourTableName entity.
+type VirustotalResult struct {
+	ent.Schema
+}
+
+func (VirustotalResult) Mixin() []ent.Mixin {
+	return []ent.Mixin{
+		IDMixin{},
+		TimeMixin{},
+	}
+}
+
+func (VirustotalResult) Fields() []ent.Field {
+	return []ent.Field{
+		field.Bool("safe").Default(false),
+		field.String("url").
+			NotEmpty(),
+		field.String("hash").Unique().NotEmpty(),
+		field.String("file_name").NotEmpty(),
+		field.String("version_id").NotEmpty(),
+	}
+}
+
+func (VirustotalResult) Annotations() []schema.Annotation {
+	return []schema.Annotation{
+		entsql.Annotation{Table: "virustotal_results"},
+	}
+}
+
+func (VirustotalResult) Edges() []ent.Edge {
+	return []ent.Edge{
+		edge.From("version", Version.Type).
+			Ref("virustotalResults").
+			Field("version_id").
+			Unique().
+			Required(),
+	}
+}