api

var MutationObject = graphql.NewObject(graphql.ObjectConfig{
	Name: "Mutation",
	Fields: graphql.Fields{
		"addAsset": {
			Description: "Add a Asset.",
			Type:        graphql.NewNonNull(object.Asset),
			Args: graphql.FieldConfigArgument{
				"input": {
					Type: graphql.NewNonNull(object.AddAssetInput),
				},
			},
			Resolve: func(p graphql.ResolveParams) (interface{}, error) {
				input := db.AddAssetInput{}
				Cast(p.Args["input"], &input)

				input.ID = uuid.NewString()

				return mutation.AddAsset(apidb.Mem, input)
			},
		},
		"updateAsset": {
			Description: "Update a Asset.",
			Type:        graphql.NewNonNull(object.Asset),
			Args: graphql.FieldConfigArgument{
				"input": {
					Type: graphql.NewNonNull(object.UpdateAssetInput),
				},
			},
			Resolve: func(p graphql.ResolveParams) (interface{}, error) {
				input := db.UpdateAssetInput{}
				Cast(p.Args["input"], &input)
				return mutation.UpdateAsset(apidb.Mem, input)
			},
		},
		"deleteAsset": {
			Description: "Delete a Asset.",
			Type:        graphql.NewNonNull(object.Asset),
			Args: graphql.FieldConfigArgument{
				"input": {
					Type: graphql.NewNonNull(object.DeleteAssetInput),
				},
			},
			Resolve: func(p graphql.ResolveParams) (interface{}, error) {
				input := db.DeleteAssetInput{}
				Cast(p.Args["input"], &input)
				return mutation.DeleteAsset(apidb.Mem, input)
			},
		},
		"addCVE": {
			Description: "Add a CVE.",
			Type:        graphql.NewNonNull(object.CVE),
			Args: graphql.FieldConfigArgument{
				"input": {
					Type: graphql.NewNonNull(object.AddCVEInput),
				},
			},
			Resolve: func(p graphql.ResolveParams) (interface{}, error) {
				input := db.AddCVEInput{}
				Cast(p.Args["input"], &input)
				return mutation.AddCVE(apidb.Mem, input)
			},
		},
		"updateCVE": {
			Description: "Update a CVE.",
			Type:        graphql.NewNonNull(object.CVE),
			Args: graphql.FieldConfigArgument{
				"input": {
					Type: graphql.NewNonNull(object.UpdateCVEInput),
				},
			},
			Resolve: func(p graphql.ResolveParams) (interface{}, error) {
				input := db.UpdateCVEInput{}
				Cast(p.Args["input"], &input)
				return mutation.UpdateCVE(apidb.Mem, input)
			},
		},
		"deleteCVE": {
			Description: "Delete a CVE.",
			Type:        graphql.NewNonNull(object.CVE),
			Args: graphql.FieldConfigArgument{
				"input": {
					Type: graphql.NewNonNull(object.DeleteCVEInput),
				},
			},
			Resolve: func(p graphql.ResolveParams) (interface{}, error) {
				input := db.DeleteCVEInput{}
				Cast(p.Args["input"], &input)
				return mutation.DeleteCVE(apidb.Mem, input)
			},
		},
	},
})

var QueryObject = graphql.NewObject(graphql.ObjectConfig{
	Name: "Query",
	Fields: graphql.Fields{
		"getAsset": {
			Description: "Get a specific Asset.",
			Type:        object.Asset,
			Args: graphql.FieldConfigArgument{
				"input": {
					Type: graphql.NewNonNull(object.GetAssetInput),
				},
			},
			Resolve: func(p graphql.ResolveParams) (interface{}, error) {
				input := db.GetAssetInput{}
				Cast(p.Args["input"], &input)
				return query.GetAsset(apidb.Mem, input)
			},
		},
		"queryAssets": {
			Description: "Query a list of Asset.",
			Type: graphql.NewNonNull(&graphql.List{
				OfType: graphql.NewNonNull(object.Asset),
			}),
			Args: graphql.FieldConfigArgument{
				"input": {
					Type: graphql.NewNonNull(object.QueryAssetsInput),
				},
			},
			Resolve: func(p graphql.ResolveParams) (interface{}, error) {
				input := db.QueryAssetInput{}
				Cast(p.Args["input"], &input)
				return query.QueryAssets(apidb.Mem, input)
			},
		},
		"getCVE": {
			Description: "Get a specific CVE.",
			Type:        object.CVE,
			Args: graphql.FieldConfigArgument{
				"input": {
					Type: graphql.NewNonNull(object.GetCVEInput),
				},
			},
			Resolve: func(p graphql.ResolveParams) (interface{}, error) {
				input := db.GetCVEInput{}
				Cast(p.Args["input"], &input)
				return query.GetCVE(apidb.Mem, input)
			},
		},
		"queryCVEs": {
			Description: "Query a list of CVE.",
			Type: graphql.NewNonNull(&graphql.List{
				OfType: graphql.NewNonNull(object.CVE),
			}),
			Args: graphql.FieldConfigArgument{
				"input": {
					Type: graphql.NewNonNull(object.QueryCVEsInput),
				},
			},
			Resolve: func(p graphql.ResolveParams) (interface{}, error) {
				input := db.QueryCVEInput{}
				Cast(p.Args["input"], &input)
				return query.QueryCVEs(apidb.Mem, input)
			},
		},
	},
})

var Schema = must(graphql.NewSchema(graphql.SchemaConfig{
	Query:    QueryObject,
	Mutation: MutationObject,
}))