NeedImages

server/db/repo/need.go

@@ -33,6 +33,11 @@ func (r *NeedRepository) Get(id int64) (*model.Need, error) {
 		return nil, err
 	}
 
+	n.Images, err = getNeedImages(r.db, n)
+	if err != nil {
+		return nil, err
+	}
+
 	c, _ := r.catRepo.Get(n.CategoryID)
 	n.Category = *c
 	return n, nil

server/graphql/types.go

@@ -133,6 +133,39 @@ var (
 		Type: needStatusEnum,
 	}
 
+	needImageType = graphql.NewObject(graphql.ObjectConfig{
+		Name: "NeedImage",
+		Fields: graphql.Fields{
+			"id": &graphql.Field{
+				Type: graphql.NewNonNull(graphql.Int),
+				Resolve: func(p graphql.ResolveParams) (interface{}, error) {
+					if i, ok := p.Source.(model.NeedImage); ok {
+						return i.ID, nil
+					}
+					return nil, nil
+				},
+			},
+			"name": &graphql.Field{
+				Type: graphql.NewNonNull(graphql.String),
+				Resolve: func(p graphql.ResolveParams) (interface{}, error) {
+					if i, ok := p.Source.(model.NeedImage); ok {
+						return i.Name, nil
+					}
+					return nil, nil
+				},
+			},
+			"url": &graphql.Field{
+				Type: graphql.NewNonNull(graphql.String),
+				Resolve: func(p graphql.ResolveParams) (interface{}, error) {
+					if i, ok := p.Source.(model.NeedImage); ok {
+						return i.URL, nil
+					}
+					return nil, nil
+				},
+			},
+		},
+	})
+
 	needType = graphql.NewObject(graphql.ObjectConfig{
 		Name: "Need",
 		Fields: graphql.Fields{
@@ -154,6 +187,9 @@ var (
 			"updatedAt": &graphql.Field{
 				Type: DateTime,
 			},
+			"images": &graphql.Field{
+				Type: graphql.NewList(needImageType),
+			},
 		},
 	})