Merge pull request #42227 from Azoy/add-script-data

[stdlib] Add Unicode script data to scalar properties

Author: Azoy

stdlib/public/SwiftShims/UnicodeData.h

@@ -58,7 +58,6 @@ __swift_uint32_t _swift_stdlib_getComposition(__swift_uint32_t x,
 // Grapheme Breaking
 //===----------------------------------------------------------------------===//
 
-
 SWIFT_RUNTIME_STDLIB_INTERNAL
 __swift_uint8_t _swift_stdlib_getGraphemeBreakProperty(__swift_uint32_t scalar);
 
@@ -101,6 +100,14 @@ __swift_uint16_t _swift_stdlib_getAge(__swift_uint32_t scalar);
 SWIFT_RUNTIME_STDLIB_INTERNAL
 __swift_uint8_t _swift_stdlib_getGeneralCategory(__swift_uint32_t scalar);
 
+SWIFT_RUNTIME_STDLIB_INTERNAL
+__swift_uint8_t _swift_stdlib_getScript(__swift_uint32_t scalar);
+
+SWIFT_RUNTIME_STDLIB_INTERNAL
+const __swift_uint8_t *_swift_stdlib_getScriptExtensions(
+                                                        __swift_uint32_t scalar,
+                                                        __swift_uint8_t *count);
+
 #ifdef __cplusplus
 } // extern "C"
 #endif

stdlib/public/core/UnicodeSPI.swift

@@ -10,6 +10,8 @@
 //
 //===----------------------------------------------------------------------===//
 
+import SwiftShims
+
 //===----------------------------------------------------------------------===//
 // Unicode.NFD
 //===----------------------------------------------------------------------===//
@@ -123,3 +125,43 @@ extension Substring {
     Unicode._NFC(base: unicodeScalars)
   }
 }
+
+//===----------------------------------------------------------------------===//
+// Unicode.Script
+//===----------------------------------------------------------------------===//
+
+extension Unicode.Scalar.Properties {
+  @_spi(_Unicode)
+  @available(SwiftStdlib 5.7, *)
+  public var _script: UInt8 {
+    let rawValue = _swift_stdlib_getScript(_scalar.value)
+
+    _internalInvariant(rawValue != .max, "Unknown script rawValue")
+
+    return rawValue
+  }
+
+  @_spi(_Unicode)
+  @available(SwiftStdlib 5.7, *)
+  public var _scriptExtensions: [UInt8] {
+    var count: UInt8 = 0
+    let pointer = _swift_stdlib_getScriptExtensions(_scalar.value, &count)
+
+    guard let pointer = pointer else {
+      return [_script]
+    }
+
+    var result: [UInt8] = []
+    result.reserveCapacity(Int(count))
+
+    for i in 0 ..< count {
+      let rawValue = pointer[Int(i)]
+
+      _internalInvariant(rawValue != .max, "Unknown script rawValue")
+
+      result.append(rawValue)
+    }
+
+    return result
+  }
+}