Re-sync with internal repository (facebook#9577)

The internal and external repositories are out of sync. This Pull Request attempts to brings them back in sync by patching the GitHub repository. Please carefully review this patch. You must disable ShipIt for your project in order to merge this pull request. DO NOT IMPORT this pull request. Instead, merge it directly on GitHub using the MERGE BUTTON. Re-enable ShipIt after merging.

Author: facebook-github-bot

third-party/thrift/src/thrift/lib/hack/src/TClientAsyncHandler.php

@@ -34,7 +34,7 @@ class TClientAsyncHandler {
     string $service_name,
     string $func_name,
     ?IThriftStruct $args = null,
-  )[zoned_local]: Awaitable<void> {
+  ): Awaitable<void> {
     // Do nothing
   }
 

third-party/thrift/src/thrift/lib/hack/src/TClientMultiAsyncHandler.php

@@ -52,7 +52,7 @@ public function getHandlers()[]: readonly dict<string, TClientAsyncHandler> {
     string $service_name,
     string $func_name,
     ?IThriftStruct $args = null,
-  )[zoned_shallow]: Awaitable<void> {
+  ): Awaitable<void> {
     await Vec\map_async(
       $this->handlers,
       async $handler ==>

third-party/thrift/src/thrift/lib/hack/src/TContextPropV2ServerHandler.php

@@ -41,6 +41,29 @@ public function preWrite(
     string $fn_name,
     mixed $result,
   ): void {
+    // keeping this logic in tact from v1 and not moving it to handlers
+    // because we write to ARTILLERY_TRACING_HEADERS and not TFMR
+    if (
+      JustKnobs::eval('artillery/sdk_www:response_header_propagation_rollout')
+    ) {
+      $context_manager = ContextManager::get();
+      if ($context_manager->getCoreContext() !== null) {
+        $ctx_prop_headers = $context_manager->processOutgoingResponse(shape());
+
+        if (
+          C\contains_key(
+            $ctx_prop_headers,
+            HTTPResponseHeader::ARTILLERY_TRACING_HEADERS,
+          )
+        ) {
+          $this->thriftServer->addHTTPHeader(
+            HTTPResponseHeader::ARTILLERY_TRACING_HEADERS,
+            $ctx_prop_headers[HTTPResponseHeader::ARTILLERY_TRACING_HEADERS],
+          );
+        }
+      }
+    }
+
     $full_params = $this->params;
     $full_params['fn_name'] = $fn_name;
     $full_params['fn_result'] = $result;
@@ -68,29 +91,6 @@ public function preWrite(
       HTTPResponseHeader::THRIFT_FRAMEWORK_METADATA_RESPONSE,
       $encoded_response_tfm,
     );
-
-    // keeping this logic in tact from v1 and not moving it to handlers
-    // because we write to ARTILLERY_TRACING_HEADERS and not TFMR
-    if (
-      JustKnobs::eval('artillery/sdk_www:response_header_propagation_rollout')
-    ) {
-      $context_manager = ContextManager::get();
-      if ($context_manager->getCoreContext() !== null) {
-        $ctx_prop_headers = $context_manager->processOutgoingResponse(shape());
-
-        if (
-          C\contains_key(
-            $ctx_prop_headers,
-            HTTPResponseHeader::ARTILLERY_TRACING_HEADERS,
-          )
-        ) {
-          $this->thriftServer->addHTTPHeader(
-            HTTPResponseHeader::ARTILLERY_TRACING_HEADERS,
-            $ctx_prop_headers[HTTPResponseHeader::ARTILLERY_TRACING_HEADERS],
-          );
-        }
-      }
-    }
   }
 
 }

third-party/thrift/src/thrift/lib/hack/src/ThriftContextPropState.php

@@ -233,6 +233,53 @@ public function getModelTypeId()[]: ?int {
     return $this->getModelInfo()?->get_model_type_id();
   }
 
+  // user id getters
+  public function getUserIds()[]: ?ContextProp\UserIds {
+    return $this->getBaggage()?->user_ids;
+  }
+
+  public function getFBUserId()[]: ?int {
+    return $this->getUserIds()?->fb_user_id;
+  }
+
+  public function getIGUserId()[]: ?int {
+    return $this->getUserIds()?->ig_user_id;
+  }
+
+  // user id setters
+  public function setUserIds(
+    ?ContextProp\UserIds $user_ids,
+  )[write_props]: void {
+    $this->storage->baggage =
+      $this->storage->baggage ?? ContextProp\Baggage::withDefaultValues();
+
+    $baggage = $this->storage->baggage as nonnull;
+    $baggage->user_ids = $user_ids;
+    $this->dirty();
+  }
+
+  public function setFBUserId(int $fb_user_id)[write_props]: void {
+    $this->storage->baggage =
+      $this->storage->baggage ?? ContextProp\Baggage::withDefaultValues();
+    $baggage = $this->storage->baggage as nonnull;
+
+    $baggage->user_ids =
+      $baggage->user_ids ?? ContextProp\UserIds::withDefaultValues();
+    $baggage->user_ids->fb_user_id = $fb_user_id;
+    $this->dirty();
+  }
+
+  public function setIGUserId(int $ig_user_id)[write_props]: void {
+    $this->storage->baggage =
+      $this->storage->baggage ?? ContextProp\Baggage::withDefaultValues();
+    $baggage = $this->storage->baggage as nonnull;
+
+    $baggage->user_ids =
+      $baggage->user_ids ?? ContextProp\UserIds::withDefaultValues();
+    $baggage->user_ids->ig_user_id = $ig_user_id;
+    $this->dirty();
+  }
+
   public function getTraceContext()[]: ?ContextProp\TraceContext {
     if ($this->storage->baggage is nonnull) {
       $trace_context = $this->storage->baggage?->trace_context;

third-party/thrift/src/thrift/lib/hack/src/__tests__/ThriftBacktraceEventHandlerTest.php

@@ -69,6 +69,7 @@ final class BacktraceClientEventHandlerTest extends WWWTest {
 
 <<Oncalls('xdc_artillery')>>
 final class ThriftBacktraceServerEventHandlerTest extends WWWTest {
+  use ClassLevelTest;
   public async function testSetsHeaderWhenBacktraceActive(): Awaitable<void> {
     $thrift_server = mock(ThriftServer::class);
     $handler = new ThriftBacktraceServerEventHandler($thrift_server);

third-party/thrift/src/thrift/lib/hack/src/__tests__/ThriftContextPropHandlerTest.php

@@ -3,6 +3,7 @@
 
 <<Oncalls('thrift')>>
 final class ThriftContextPropHandlerTest extends WWWTest {
+  use ClassLevelTest;
   private static function readTFMForTest(string $v): ThriftFrameworkMetadata {
     $transport = Base64::decode($v);
     $buf = new TMemoryBuffer($transport);

third-party/thrift/src/thrift/lib/hack/src/__tests__/ThriftContextPropServerEventHandlerTest.php

@@ -3,6 +3,7 @@
 
 <<Oncalls('xdc_artillery')>>
 final class ThriftContextPropServerEventHandlerTest extends WWWTest {
+  use ClassLevelTest;
   public async function testThriftContextPropResponsePath(): Awaitable<void> {
     ThriftContextPropState::get()->addExperimentId(1);
 

third-party/thrift/src/thrift/lib/hack/src/__tests__/ThriftContextPropStateScopeGuardTest.php

@@ -4,6 +4,8 @@
 <<Oncalls('xdc_artillery')>>
 final class ThriftContextPropStateScopeGuardTest extends WWWTest {
 
+  use ClassLevelTest;
+
   private function getSerializedTFMHeaders(
     string $overriden_request_id,
     int $overriden_origin_id,

third-party/thrift/src/thrift/lib/hack/src/__tests__/ThriftContextPropStateTest.php

@@ -3,6 +3,7 @@
 
 <<Oncalls('thrift')>>
 final class ThriftContextPropStateTest extends WWWTest {
+  use ClassLevelTest;
   public function testAccess(): void {
     $tcps = ThriftContextPropState::get();
     expect($tcps->getRequestId())->toEqual("");
@@ -37,6 +38,39 @@ public function testRegionalizationEntityNullable(): void {
     expect($tcps->getRegionalizationEntity())->toBeNull();
   }
 
+  public function testUserIdsNullable(): void {
+    $tcps = ThriftContextPropState::get();
+    expect($tcps->getUserIds())->toBeNull();
+
+    // 0 is different from null
+    $tcps->setUserIds(ContextProp\UserIds::withDefaultValues());
+    expect($tcps->getUserIds())->toNotBeNull();
+    expect($tcps->getUserIds()?->fb_user_id)->toEqual(0);
+    expect($tcps->getUserIds()?->ig_user_id)->toEqual(0);
+
+    // override existing value
+    $tcps->setUserIds(new ContextProp\UserIds(1, 2));
+    expect($tcps->getUserIds()?->fb_user_id)->toEqual(1);
+    expect($tcps->getUserIds()?->ig_user_id)->toEqual(2);
+
+    // back to null
+    $tcps->setUserIds(null);
+    expect($tcps->getUserIds())->toBeNull();
+
+    // set FB Id only
+    $tcps->setFBUserId(3);
+    expect($tcps->getUserIds()?->fb_user_id)->toEqual(3);
+    expect($tcps->getFBUserId())->toEqual(3);
+    expect($tcps->getUserIds()?->ig_user_id)->toEqual(0);
+
+    // set IG Id only
+    $tcps->setIGUserId(4);
+    expect($tcps->getUserIds()?->fb_user_id)->toEqual(3);
+    expect($tcps->getUserIds()?->ig_user_id)->toEqual(4);
+    expect($tcps->getIGUserId())->toEqual(4);
+
+  }
+
   public function testBaggage(): void {
     $tcps = ThriftContextPropState::get();
     $tcps->clear();

third-party/thrift/src/thrift/lib/hack/src/__tests__/ThriftImmutableWrapperTest.php

@@ -4,6 +4,8 @@
 <<Oncalls('signals_infra')>>
 final class ThriftImmutableWrapperTest extends WWWTest {
 
+  use ClassLevelTest;
+
   public async function testGetSerializedThrift(): Awaitable<void> {
     $underlying = AdsConsentPlatformParams::fromShape(shape(
       'traceID' => '123',