Added team and alliances to gql_vex.go

gql.go

@@ -147,6 +147,129 @@ func GQLEventChildren(p graphql.ResolveParams) (interface{}, error) {
   })
 }
 
+var gql_list_resource * graphql.List = nil
+func GQLListResource() * graphql.List {
+  if gql_list_resource == nil {
+    gql_list_resource = graphql.NewList(GQLInterfaceResource())
+  }
+
+  return gql_list_resource
+}
+
+var gql_interface_resource * graphql.Interface = nil
+func GQLInterfaceResource() * graphql.Interface {
+  if gql_interface_resource == nil {
+    gql_interface_resource = graphql.NewInterface(graphql.InterfaceConfig{
+      Name: "Resource",
+      ResolveType: func(p graphql.ResolveTypeParams) *graphql.Object {
+        valid_resources, ok := p.Context.Value("valid_resources").(map[reflect.Type]*graphql.Object)
+        if ok == false {
+          return nil
+        }
+        for key, value := range(valid_resources) {
+          if reflect.TypeOf(p.Value) == key {
+            return value
+          }
+        }
+        return nil
+      },
+      Fields: graphql.Fields{},
+    })
+
+    if gql_list_resource == nil {
+      gql_list_resource = graphql.NewList(gql_interface_resource)
+    }
+
+    gql_interface_resource.AddFieldConfig("ID", &graphql.Field{
+      Type: graphql.String,
+    })
+
+    gql_interface_resource.AddFieldConfig("Name", &graphql.Field{
+      Type: graphql.String,
+    })
+
+    gql_interface_resource.AddFieldConfig("Description", &graphql.Field{
+      Type: graphql.String,
+    })
+
+    gql_interface_resource.AddFieldConfig("Parents", &graphql.Field{
+      Type: GQLListResource(),
+    })
+
+  }
+
+  return gql_interface_resource
+}
+
+func GQLResourceFn(p graphql.ResolveParams, fn func(Resource, graphql.ResolveParams)(interface{}, error))(interface{}, error) {
+    if resource, ok := p.Source.(Resource); ok {
+      return fn(resource, p)
+    }
+    return nil, errors.New(fmt.Sprintf("Failed to cast source to resource, %+v", p.Source))
+}
+
+func GQLResourceID(p graphql.ResolveParams) (interface{}, error) {
+  return GQLResourceFn(p, func(resource Resource, p graphql.ResolveParams) (interface{}, error) {
+    return resource.ID(), nil
+  })
+}
+
+func GQLResourceName(p graphql.ResolveParams) (interface{}, error) {
+  return GQLResourceFn(p, func(resource Resource, p graphql.ResolveParams) (interface{}, error) {
+    return resource.Name(), nil
+  })
+}
+
+func GQLResourceDescription(p graphql.ResolveParams) (interface{}, error) {
+  return GQLResourceFn(p, func(resource Resource, p graphql.ResolveParams) (interface{}, error) {
+    return resource.Description(), nil
+  })
+}
+
+func GQLResourceParents(p graphql.ResolveParams) (interface{}, error) {
+  return GQLResourceFn(p, func(resource Resource, p graphql.ResolveParams) (interface{}, error) {
+    return resource.Parents(), nil
+  })
+}
+
+var gql_type_base_resource *graphql.Object = nil
+func GQLTypeBaseResource() * graphql.Object {
+  if gql_type_base_resource == nil {
+    gql_type_base_resource = graphql.NewObject(graphql.ObjectConfig{
+      Name: "BaseResource",
+      Interfaces: []*graphql.Interface{
+        GQLInterfaceResource(),
+      },
+      IsTypeOf: func(p graphql.IsTypeOfParams) bool {
+        _, ok := p.Value.(*BaseResource)
+        return ok
+      },
+      Fields: graphql.Fields{},
+    })
+
+    gql_type_base_resource.AddFieldConfig("ID", &graphql.Field{
+      Type: graphql.String,
+      Resolve: GQLResourceID,
+    })
+
+    gql_type_base_resource.AddFieldConfig("Name", &graphql.Field{
+      Type: graphql.String,
+      Resolve: GQLResourceName,
+    })
+
+    gql_type_base_resource.AddFieldConfig("Description", &graphql.Field{
+      Type: graphql.String,
+      Resolve: GQLResourceDescription,
+    })
+
+    gql_type_base_resource.AddFieldConfig("Parents", &graphql.Field{
+      Type: GQLListResource(),
+      Resolve: GQLResourceParents,
+    })
+  }
+
+  return gql_type_base_resource
+}
 
 var gql_list_event * graphql.List = nil
 func GQLListEvent() * graphql.List {
@@ -302,6 +425,9 @@ func (server * GQLServer) Handler() func(http.ResponseWriter, *http.Request) {
   valid_events[reflect.TypeOf((*BaseEvent)(nil))] = GQLTypeBaseEvent()
   valid_events[reflect.TypeOf((*EventQueue)(nil))] = GQLTypeEventQueue()
 
+  valid_resources := map[reflect.Type]*graphql.Object{}
+  valid_resources[reflect.TypeOf((*BaseResource)(nil))] = GQLTypeBaseResource()
+
   gql_types := []graphql.Type{GQLTypeBaseEvent(), GQLTypeEventQueue()}
   for go_t, gql_t := range(server.extended_types) {
     valid_events[go_t] = gql_t
@@ -332,8 +458,10 @@ func (server * GQLServer) Handler() func(http.ResponseWriter, *http.Request) {
   if err != nil{
     panic(err)
   }
-
+  ctx := context.Background()
-  return GQLHandler(schema, context.WithValue(context.Background(), "valid_events", valid_events))
+  ctx = context.WithValue(ctx, "valid_events", valid_events)
+  ctx = context.WithValue(ctx, "valid_resources", valid_resources)
+  return GQLHandler(schema, ctx)
 }
 
 func (server * GQLServer) Init(abort chan error) bool {

gql_vex.go

@@ -12,6 +12,29 @@ func GQLVexTypes() map[reflect.Type]*graphql.Object {
   return types
 }
 
+var gql_vex_list_team * graphql.List = nil
+func GQLVexListTeam() * graphql.List {
+  if gql_vex_list_team == nil {
+    gql_vex_list_team = graphql.NewList(GQLVexTypeTeam())
+  }
+  return gql_vex_list_team
+}
+
+var gql_vex_list_alliance * graphql.List = nil
+func GQLVexListAlliance() * graphql.List {
+  if gql_vex_list_alliance == nil {
+    gql_vex_list_alliance = graphql.NewList(GQLVexTypeAlliance())
+  }
+  return gql_vex_list_alliance
+}
+
+func GQLVexMatchAlliances(p graphql.ResolveParams) (interface{}, error) {
+  return GQLEventFn(p, func(event Event, p graphql.ResolveParams) (interface{}, error) {
+    //TODO improve
+    return event.(*Match).alliances, nil
+  })
+}
+
 var gql_vex_type_match * graphql.Object = nil
 func GQLVexTypeMatch() * graphql.Object {
   if gql_vex_type_match == nil {
@@ -46,7 +69,90 @@ func GQLVexTypeMatch() * graphql.Object {
       Type: GQLListEvent(),
       Resolve: GQLEventChildren,
     })
+
+    gql_vex_type_match.AddFieldConfig("Alliances", &graphql.Field{
+      Type: GQLVexListAlliance(),
+      Resolve: GQLVexMatchAlliances,
+    })
   }
 
   return gql_vex_type_match
 }
+
+var gql_vex_type_alliance * graphql.Object = nil
+func GQLVexTypeAlliance() * graphql.Object {
+  if gql_vex_type_alliance == nil {
+    gql_vex_type_alliance = graphql.NewObject(graphql.ObjectConfig{
+      Name: "Alliance",
+      Interfaces: []*graphql.Interface{
+        GQLInterfaceResource(),
+      },
+      IsTypeOf: func(p graphql.IsTypeOfParams) bool {
+        _, ok := p.Value.(*Alliance)
+        return ok
+      },
+      Fields: graphql.Fields{},
+    })
+  }
+
+  gql_vex_type_alliance.AddFieldConfig("ID", &graphql.Field{
+    Type: graphql.String,
+    Resolve: GQLResourceID,
+  })
+
+  gql_vex_type_alliance.AddFieldConfig("Name", &graphql.Field{
+    Type: graphql.String,
+    Resolve: GQLResourceName,
+  })
+
+  gql_vex_type_alliance.AddFieldConfig("Description", &graphql.Field{
+    Type: graphql.String,
+    Resolve: GQLResourceDescription,
+  })
+
+  gql_vex_type_alliance.AddFieldConfig("Parents", &graphql.Field{
+    Type: GQLListResource(),
+    Resolve: GQLResourceParents,
+  })
+
+  return gql_vex_type_alliance
+}
+
+var gql_vex_type_team * graphql.Object = nil
+func GQLVexTypeTeam() * graphql.Object {
+  if gql_vex_type_team == nil {
+    gql_vex_type_team = graphql.NewObject(graphql.ObjectConfig{
+      Name: "Team",
+      Interfaces: []*graphql.Interface{
+        GQLInterfaceResource(),
+      },
+      IsTypeOf: func(p graphql.IsTypeOfParams) bool {
+        _, ok := p.Value.(*Team)
+        return ok
+      },
+      Fields: graphql.Fields{},
+    })
+  }
+
+  gql_vex_type_team.AddFieldConfig("ID", &graphql.Field{
+    Type: graphql.String,
+    Resolve: GQLResourceID,
+  })
+
+  gql_vex_type_team.AddFieldConfig("Name", &graphql.Field{
+    Type: graphql.String,
+    Resolve: GQLResourceName,
+  })
+
+  gql_vex_type_team.AddFieldConfig("Description", &graphql.Field{
+    Type: graphql.String,
+    Resolve: GQLResourceDescription,
+  })
+
+  gql_vex_type_team.AddFieldConfig("Parents", &graphql.Field{
+    Type: GQLListResource(),
+    Resolve: GQLResourceParents,
+  })
+
+  return gql_vex_type_team
+}

vex.go

@@ -147,6 +147,7 @@ type Match struct {
   state string
   control string
   control_start time.Time
+  alliances []*Alliance
 }
 
 func (match * Match) update(signal GraphSignal) {
@@ -164,6 +165,7 @@ func NewMatch(alliance0 * Alliance, alliance1 * Alliance, arena Arena) * Match {
     control: "init",
     control_start: time.UnixMilli(0),
     arena: arena,
+    alliances: []*Alliance{alliance0, alliance1},
   }
 
   match.actions["wait"] = EventWait(match)