feat: add VirusTotal Results to API (#92)

* feat: add virustotal results to api

* fix: add missing files from commit, fix migration

* fix: remove unused vars

* cleanup commented out code

* fix comments, naming of functions, bulk saving

* add newline back to gqlgrn

* fix lint errors

* fix generated.go being out of sync

* fix migration sum

* switch to virustotal_results output

* update migration to remove url

* reorganized code around virus total

* resolved conflicts from staging

* redo migrations/schema, fix formatting issues

* fix migration hash

* use devbox to generate files

Author: knightzac19

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

@@ -62,5 +62,6 @@ func (Version) Edges() []ent.Edge {
 		edge.To("dependencies", Mod.Type).
 			Through("version_dependencies", VersionDependency.Type),
 		edge.To("targets", VersionTarget.Type),
+		edge.To("virustotal_results", VirustotalResult.Type),
 	}
 }

db/schema/virustotal_result.go

@@ -0,0 +1,46 @@
+package schema
+
+import (
+	"entgo.io/ent"
+	"entgo.io/ent/schema/edge"
+	"entgo.io/ent/schema/field"
+	"entgo.io/ent/schema/index"
+)
+
+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("hash").MaxLen(64).NotEmpty(),
+		field.String("file_name").NotEmpty(),
+		field.String("version_id").MaxLen(14).NotEmpty(),
+	}
+}
+
+func (VirustotalResult) Indexes() []ent.Index {
+	return []ent.Index{
+		index.Fields("safe"),
+		index.Fields("hash", "version_id").Unique(),
+		index.Fields("file_name"),
+	}
+}
+
+func (VirustotalResult) Edges() []ent.Edge {
+	return []ent.Edge{
+		edge.From("version", Version.Type).
+			Ref("virustotal_results").
+			Field("version_id").
+			Unique().
+			Required(),
+	}
+}