Skip to content

Commit 84691ab

Browse files
cherylEnkiducherylEnkidu
committed
Add stage API
1 parent bd0ce8f commit 84691ab

23 files changed

+839
-0
lines changed

‎Firestore/Swift/Source/SwiftAPI/Pipeline/AccumulatorExpr.swift

+15
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,15 @@
1+
// Copyright 2025 Google LLC
2+
//
3+
// Licensed under the Apache License, Version 2.0 (the "License");
4+
// you may not use this file except in compliance with the License.
5+
// You may obtain a copy of the License at
6+
//
7+
// http://www.apache.org/licenses/LICENSE-2.0
8+
//
9+
// Unless required by applicable law or agreed to in writing, software
10+
// distributed under the License is distributed on an "AS IS" BASIS,
11+
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
// See the License for the specific language governing permissions and
13+
// limitations under the License.
14+
15+
public protocol AccumulatorExpr: Sendable {}

‎Firestore/Swift/Source/SwiftAPI/Pipeline/AccumulatorWithAlias.swift

+17
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,17 @@
1+
// Copyright 2025 Google LLC
2+
//
3+
// Licensed under the Apache License, Version 2.0 (the "License");
4+
// you may not use this file except in compliance with the License.
5+
// You may obtain a copy of the License at
6+
//
7+
// http://www.apache.org/licenses/LICENSE-2.0
8+
//
9+
// Unless required by applicable law or agreed to in writing, software
10+
// distributed under the License is distributed on an "AS IS" BASIS,
11+
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
// See the License for the specific language governing permissions and
13+
// limitations under the License.
14+
15+
public struct AccumulatorWithAlias {
16+
init(_ expr: some AccumulatorExpr, _ alias: String) {}
17+
}

‎Firestore/Swift/Source/SwiftAPI/Pipeline/Add.swift

+15
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,15 @@
1+
// Copyright 2025 Google LLC
2+
//
3+
// Licensed under the Apache License, Version 2.0 (the "License");
4+
// you may not use this file except in compliance with the License.
5+
// You may obtain a copy of the License at
6+
//
7+
// http://www.apache.org/licenses/LICENSE-2.0
8+
//
9+
// Unless required by applicable law or agreed to in writing, software
10+
// distributed under the License is distributed on an "AS IS" BASIS,
11+
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
// See the License for the specific language governing permissions and
13+
// limitations under the License.
14+
15+
public struct Add {}

‎Firestore/Swift/Source/SwiftAPI/Pipeline/AggregateOption.swift

+23
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
// Copyright 2025 Google LLC
2+
//
3+
// Licensed under the Apache License, Version 2.0 (the "License");
4+
// you may not use this file except in compliance with the License.
5+
// You may obtain a copy of the License at
6+
//
7+
// http://www.apache.org/licenses/LICENSE-2.0
8+
//
9+
// Unless required by applicable law or agreed to in writing, software
10+
// distributed under the License is distributed on an "AS IS" BASIS,
11+
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
// See the License for the specific language governing permissions and
13+
// limitations under the License.
14+
15+
public struct AggregateOption {
16+
public let accumulators: [AccumulatorWithAlias]
17+
public let groups: [SelectableOrFieldName]?
18+
19+
public init(accumulators: [AccumulatorWithAlias], groups: [SelectableOrFieldName]? = nil) {
20+
self.accumulators = accumulators
21+
self.groups = groups
22+
}
23+
}

‎Firestore/Swift/Source/SwiftAPI/Pipeline/BooleanExpr.swift

+15
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,15 @@
1+
// Copyright 2025 Google LLC
2+
//
3+
// Licensed under the Apache License, Version 2.0 (the "License");
4+
// you may not use this file except in compliance with the License.
5+
// You may obtain a copy of the License at
6+
//
7+
// http://www.apache.org/licenses/LICENSE-2.0
8+
//
9+
// Unless required by applicable law or agreed to in writing, software
10+
// distributed under the License is distributed on an "AS IS" BASIS,
11+
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
// See the License for the specific language governing permissions and
13+
// limitations under the License.
14+
15+
public struct BooleanExpr: FunctionExpr, Sendable {}

‎Firestore/Swift/Source/SwiftAPI/Pipeline/Constant.swift

+15
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,15 @@
1+
// Copyright 2025 Google LLC
2+
//
3+
// Licensed under the Apache License, Version 2.0 (the "License");
4+
// you may not use this file except in compliance with the License.
5+
// You may obtain a copy of the License at
6+
//
7+
// http://www.apache.org/licenses/LICENSE-2.0
8+
//
9+
// Unless required by applicable law or agreed to in writing, software
10+
// distributed under the License is distributed on an "AS IS" BASIS,
11+
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
// See the License for the specific language governing permissions and
13+
// limitations under the License.
14+
15+
public struct Constant: Expr, Sendable {}

‎Firestore/Swift/Source/SwiftAPI/Pipeline/Count.swift

+13
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
// Copyright 2025 Google LLC
2+
//
3+
// Licensed under the Apache License, Version 2.0 (the "License");
4+
// you may not use this file except in compliance with the License.
5+
// You may obtain a copy of the License at
6+
//
7+
// http://www.apache.org/licenses/LICENSE-2.0
8+
//
9+
// Unless required by applicable law or agreed to in writing, software
10+
// distributed under the License is distributed on an "AS IS" BASIS,
11+
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
// See the License for the specific language governing permissions and
13+
// limitations under the License.

‎Firestore/Swift/Source/SwiftAPI/Pipeline/Eq.swift

+15
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,15 @@
1+
// Copyright 2025 Google LLC
2+
//
3+
// Licensed under the Apache License, Version 2.0 (the "License");
4+
// you may not use this file except in compliance with the License.
5+
// You may obtain a copy of the License at
6+
//
7+
// http://www.apache.org/licenses/LICENSE-2.0
8+
//
9+
// Unless required by applicable law or agreed to in writing, software
10+
// distributed under the License is distributed on an "AS IS" BASIS,
11+
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
// See the License for the specific language governing permissions and
13+
// limitations under the License.
14+
15+
public struct Eq {}

‎Firestore/Swift/Source/SwiftAPI/Pipeline/Expr.swift

+35
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,35 @@
1+
// Copyright 2025 Google LLC
2+
//
3+
// Licensed under the Apache License, Version 2.0 (the "License");
4+
// you may not use this file except in compliance with the License.
5+
// You may obtain a copy of the License at
6+
//
7+
// http://www.apache.org/licenses/LICENSE-2.0
8+
//
9+
// Unless required by applicable law or agreed to in writing, software
10+
// distributed under the License is distributed on an "AS IS" BASIS,
11+
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
// See the License for the specific language governing permissions and
13+
// limitations under the License.
14+
15+
public protocol Expr: Sendable, Equatable {
16+
func alias(_ alias: String) -> ExprWithAlias
17+
18+
func eq(_ expr: BooleanExpr)
19+
20+
func add(_ expr: some FunctionExpr)
21+
22+
func add(_ expr: Constant)
23+
}
24+
25+
public extension Expr {
26+
func alias(_ alias: String) -> ExprWithAlias {
27+
return ExprWithAlias(self, alias)
28+
}
29+
30+
func eq(_ expr: some Expr) {}
31+
32+
func add(_ expr: some Expr) {}
33+
34+
func add(_ expr: Constant) {}
35+
}

‎Firestore/Swift/Source/SwiftAPI/Pipeline/ExprWithAlias.swift

+17
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,17 @@
1+
// Copyright 2025 Google LLC
2+
//
3+
// Licensed under the Apache License, Version 2.0 (the "License");
4+
// you may not use this file except in compliance with the License.
5+
// You may obtain a copy of the License at
6+
//
7+
// http://www.apache.org/licenses/LICENSE-2.0
8+
//
9+
// Unless required by applicable law or agreed to in writing, software
10+
// distributed under the License is distributed on an "AS IS" BASIS,
11+
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
// See the License for the specific language governing permissions and
13+
// limitations under the License.
14+
15+
public struct ExprWithAlias {
16+
init(_ expr: some Expr, _ alias: String) {}
17+
}

‎Firestore/Swift/Source/SwiftAPI/Pipeline/Field.swift

+18
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
1+
// Copyright 2025 Google LLC
2+
//
3+
// Licensed under the Apache License, Version 2.0 (the "License");
4+
// you may not use this file except in compliance with the License.
5+
// You may obtain a copy of the License at
6+
//
7+
// http://www.apache.org/licenses/LICENSE-2.0
8+
//
9+
// Unless required by applicable law or agreed to in writing, software
10+
// distributed under the License is distributed on an "AS IS" BASIS,
11+
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
// See the License for the specific language governing permissions and
13+
// limitations under the License.
14+
15+
public struct Field: Expr, Sendable {
16+
public var alias: String
17+
public let fieldName: String
18+
}

‎Firestore/Swift/Source/SwiftAPI/Pipeline/FindNearestOptions.swift

+63
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,63 @@
1+
// Copyright 2025 Google LLC
2+
//
3+
// Licensed under the Apache License, Version 2.0 (the "License");
4+
// you may not use this file except in compliance with the License.
5+
// You may obtain a copy of the License at
6+
//
7+
// http://www.apache.org/licenses/LICENSE-2.0
8+
//
9+
// Unless required by applicable law or agreed to in writing, software
10+
// distributed under the License is distributed on an "AS IS" BASIS,
11+
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
// See the License for the specific language governing permissions and
13+
// limitations under the License.
14+
15+
#if SWIFT_PACKAGE
16+
@_exported import FirebaseFirestoreInternalWrapper
17+
#else
18+
@_exported import FirebaseFirestoreInternal
19+
#endif // SWIFT_PACKAGE
20+
21+
import Foundation
22+
23+
public struct FindNearestOptions {
24+
let field: Field
25+
let vectorValue: [VectorValue]
26+
let distanceMeasure: DistanceMeasure
27+
let limit: Int?
28+
let distanceField: String?
29+
}
30+
31+
public struct DistanceMeasure: Sendable, Equatable, Hashable {
32+
enum Kind: String {
33+
case euclidean
34+
case cosine
35+
case dotProduct = "dot_product"
36+
}
37+
38+
public static var euclidean: DistanceMeasure {
39+
return self.init(kind: .euclidean)
40+
}
41+
42+
public static var cosine: DistanceMeasure {
43+
return self.init(kind: .cosine)
44+
}
45+
46+
public static var dotProduct: DistanceMeasure {
47+
return self.init(kind: .dotProduct)
48+
}
49+
50+
/// Returns the raw string representation of the `DistanceMeasure` value.
51+
public let rawValue: String
52+
53+
init(kind: Kind) {
54+
rawValue = kind.rawValue
55+
}
56+
57+
public init(rawValue: String) {
58+
if Kind(rawValue: rawValue) == nil {
59+
// impl
60+
}
61+
self.rawValue = rawValue
62+
}
63+
}

‎Firestore/Swift/Source/SwiftAPI/Pipeline/FunctionExpr.swift

+15
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,15 @@
1+
// Copyright 2025 Google LLC
2+
//
3+
// Licensed under the Apache License, Version 2.0 (the "License");
4+
// you may not use this file except in compliance with the License.
5+
// You may obtain a copy of the License at
6+
//
7+
// http://www.apache.org/licenses/LICENSE-2.0
8+
//
9+
// Unless required by applicable law or agreed to in writing, software
10+
// distributed under the License is distributed on an "AS IS" BASIS,
11+
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
// See the License for the specific language governing permissions and
13+
// limitations under the License.
14+
15+
public protocol FunctionExpr: Expr, Sendable {}

‎Firestore/Swift/Source/SwiftAPI/Pipeline/LogicalMin.swift

+15
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,15 @@
1+
// Copyright 2025 Google LLC
2+
//
3+
// Licensed under the Apache License, Version 2.0 (the "License");
4+
// you may not use this file except in compliance with the License.
5+
// You may obtain a copy of the License at
6+
//
7+
// http://www.apache.org/licenses/LICENSE-2.0
8+
//
9+
// Unless required by applicable law or agreed to in writing, software
10+
// distributed under the License is distributed on an "AS IS" BASIS,
11+
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
// See the License for the specific language governing permissions and
13+
// limitations under the License.
14+
15+
public struct LogicalMin: FunctionExpr, Sendable, Equatable {}

‎Firestore/Swift/Source/SwiftAPI/Pipeline/Min.swift

+13
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
// Copyright 2025 Google LLC
2+
//
3+
// Licensed under the Apache License, Version 2.0 (the "License");
4+
// you may not use this file except in compliance with the License.
5+
// You may obtain a copy of the License at
6+
//
7+
// http://www.apache.org/licenses/LICENSE-2.0
8+
//
9+
// Unless required by applicable law or agreed to in writing, software
10+
// distributed under the License is distributed on an "AS IS" BASIS,
11+
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
// See the License for the specific language governing permissions and
13+
// limitations under the License.

‎Firestore/Swift/Source/SwiftAPI/Pipeline/Ordering.swift

+47
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,47 @@
1+
/*
2+
* Copyright 2024 Google LLC
3+
*
4+
* Licensed under the Apache License, Version 2.0 (the "License");
5+
* you may not use this file except in compliance with the License.
6+
* You may obtain a copy of the License at
7+
*
8+
* http://www.apache.org/licenses/LICENSE-2.0
9+
*
10+
* Unless required by applicable law or agreed to in writing, software
11+
* distributed under the License is distributed on an "AS IS" BASIS,
12+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
* See the License for the specific language governing permissions and
14+
* limitations under the License.
15+
*/
16+
17+
public struct Ordering {
18+
let direction: Direction
19+
}
20+
21+
public struct Direction: Sendable, Equatable, Hashable {
22+
enum Kind: String {
23+
case ascending
24+
case descending
25+
}
26+
27+
public static var ascending: Direction {
28+
return self.init(kind: .ascending)
29+
}
30+
31+
public static var descending: Direction {
32+
return self.init(kind: .descending)
33+
}
34+
35+
public let rawValue: String
36+
37+
init(kind: Kind) {
38+
rawValue = kind.rawValue
39+
}
40+
41+
public init(rawValue: String) {
42+
if Kind(rawValue: rawValue) == nil {
43+
// impl
44+
}
45+
self.rawValue = rawValue
46+
}
47+
}

0 commit comments

Comments
 (0)