Make range filters generic

Reviewed-on: #58
Reviewed-by: Ivan Malison <ivanmalison@gmail.com>

src/operations/shots.gql

@@ -116,6 +116,7 @@ query GetShotsWithMetadata(
       serializedShotPaths {
         b64EncodedBuffer
       }
+
       createdAt @include(if: $includeCreatedAt)
       updatedAt @include(if: $includeUpdatedAt)
       cueObjectFeatures @include(if: $includeCueObjectFeatures) {

src/schema.gql

@@ -54,6 +54,7 @@ type TargetMetricsGQL {
   makePercentage: Float
   averageDifficulty: Float
   spinTypeCounts: SpinTypeCountsGQL
+  shotDirectionCounts: ShotDirectionCountsGQL
 }
 
 type SpinTypeCountsGQL {
@@ -63,6 +64,12 @@ type SpinTypeCountsGQL {
   unknown: Int!
 }
 
+type ShotDirectionCountsGQL {
+  left: Int!
+  right: Int!
+  straight: Int!
+}
+
 input AggregateInputGQL {
   aggregations: [AggregationInput!]!
   filterInput: FilterInput
@@ -124,11 +131,11 @@ input FilterInput @oneOf {
   andFilters: [FilterInput!]
   orFilters: [FilterInput!]
   notFilter: FilterInput
-  cueObjectDistance: RangeFilter
-  targetPocketDistance: RangeFilter
-  cueObjectAngle: RangeFilter
-  cueBallSpeed: RangeFilter
-  difficulty: RangeFilter
+  cueObjectDistance: FloatRangeFilter
+  targetPocketDistance: FloatRangeFilter
+  cueObjectAngle: FloatRangeFilter
+  cueBallSpeed: FloatRangeFilter
+  difficulty: FloatRangeFilter
   intendedPocketType: [PocketEnum!]
   shotDirection: [ShotDirectionEnum!]
   videoId: [Int!]
@@ -142,18 +149,23 @@ input FilterInput @oneOf {
   isLeftMiss: [Boolean!]
   isRightMiss: [Boolean!]
   isDirect: [Boolean!]
-  tableSize: RangeFilter
-  bankAngle: RangeFilter
-  bankDistance: RangeFilter
-  kickAngle: RangeFilter
-  kickDistance: RangeFilter
-  cueAngleAfterObject: RangeFilter
+  tableSize: FloatRangeFilter
+  bankAngle: FloatRangeFilter
+  bankDistance: FloatRangeFilter
+  kickAngle: FloatRangeFilter
+  kickDistance: FloatRangeFilter
+  cueAngleAfterObject: FloatRangeFilter
   spinType: [SpinTypeEnum!]
-  cueSpeedAfterObject: RangeFilter
-  falsePositiveScore: RangeFilter
+  cueSpeedAfterObject: FloatRangeFilter
+  falsePositiveScore: FloatRangeFilter
+  backcut: [Boolean!]
+  targetPocketAngleDirection: [ShotDirectionEnum!]
+  targetPocketAngle: FloatRangeFilter
+  missAngleInDegrees: FloatRangeFilter
+  createdAt: DateRangeFilter
 }
 
-input RangeFilter {
+input FloatRangeFilter {
   lessThan: Float = null
   greaterThanEqualTo: Float = null
   greaterThan: Float = null
@@ -193,6 +205,20 @@ enum SpinTypeEnum {
   UNKNOWN
 }
 
+input DateRangeFilter {
+  lessThan: Date = null
+  greaterThanEqualTo: Date = null
+  greaterThan: Date = null
+  includeOnNone: Boolean! = false
+  lessThanInclusive: Boolean! = false
+  greaterThanInclusive: Boolean! = true
+}
+
+"""
+Date (isoformat)
+"""
+scalar Date
+
 type BucketSetGQL {
   keyName: String!
   feature: String!
@@ -231,6 +257,7 @@ type ShotGQL {
   updatedAt: DateTime
   cueObjectFeatures: CueObjectFeaturesGQL
   pocketingIntentionFeatures: PocketingIntentionFeaturesGQL
+  pocketingIntentionInfo: PocketingIntentionInfoGQL
   bankFeatures: BankFeaturesGQL
   serializedShotPaths: SerializedShotPathsGQL
   user: UserGQL
@@ -252,6 +279,24 @@ type PocketingIntentionFeaturesGQL {
   make: Boolean
   intendedPocketType: PocketEnum
   difficulty: Float
+  targetPocketAngle: Float
+  targetPocketAngleDirection: ShotDirectionEnum
+  backcut: Boolean
+}
+
+type PocketingIntentionInfoGQL {
+  ballId: Int!
+  pocketId: PocketIdentifier!
+  pathMetadataIndex: Int!
+}
+
+enum PocketIdentifier {
+  TOP_LEFT
+  TOP_SIDE
+  TOP_RIGHT
+  BOTTOM_LEFT
+  BOTTOM_SIDE
+  BOTTOM_RIGHT
 }
 
 type BankFeaturesGQL {