fixup! feat: jmap support - part 2

Signed-off-by: SebastianKrupinski <krupinskis05@gmail.com>

Author: SebastianKrupinski

lib/JMAP/JmapMessageAdapter.php

@@ -14,6 +14,7 @@
 use JmapClient\Responses\Mail\MailPart as MailPartResponse;
 use OCA\Mail\Address;
 use OCA\Mail\AddressList;
+use OCA\Mail\Db\LocalMessage;
 use OCA\Mail\Db\Message;
 use OCA\Mail\Db\Tag;
 use OCA\Mail\Model\IMAPMessage;
@@ -94,6 +95,7 @@ public function convertToModelMessage(MailParametersResponse $source, int $uid,
 		/** @var list<IMAPAttachment> $inlineAttachments */
 		$dispositionNotificationTo = $this->firstHeaderValue($source, 'Disposition-Notification-To') ?? '';
 		$hasDkimSignature = $this->firstHeaderValue($source, 'DKIM-Signature') !== null;
+		$hasAiGeneratedHeader = trim($this->firstHeaderValue($source, LocalMessage::HEADER_AI_GENERATED) ?? '') === '1';
 		[$unsubscribeUrl, $unsubscribeMailto] = $this->extractUnsubscribeTargets($source);
 		$isOneClickUnsubscribe = $unsubscribeUrl !== null
 			&& str_contains(strtolower($this->firstHeaderValue($source, 'List-Unsubscribe-Post') ?? ''), 'one-click');
@@ -120,6 +122,7 @@ public function convertToModelMessage(MailParametersResponse $source, int $uid,
 			$this->normalizeRawMessageIdList($source->references()),
 			$dispositionNotificationTo,
 			$hasDkimSignature,
+			$hasAiGeneratedHeader,
 			[],
 			$unsubscribeUrl,
 			$isOneClickUnsubscribe,

tests/Unit/JMAP/JmapMessageAdapterTest.php

@@ -180,4 +180,27 @@ public function testFallsBackToSenderWhenFromMissing(): void {
 		self::assertNotNull($from);
 		self::assertSame('sender@example.com', $from->getEmail());
 	}
+
+	public static function aiGeneratedHeaderProvider(): array {
+		return [
+			'absent' => [null, false],
+			'set' => ['1', true],
+			'set with whitespace' => [' 1 ', true],
+			'other value' => ['0', false],
+		];
+	}
+
+	/**
+	 * @dataProvider aiGeneratedHeaderProvider
+	 */
+	public function testModelMessageAiGeneratedHeader(?string $headerValue, bool $expected): void {
+		$source = $this->source();
+		$source->method('header')->willReturnCallback(
+			static fn (string $name): ?string => $name === 'X-AI-Generated' ? $headerValue : null,
+		);
+
+		$message = $this->adapter->convertToModelMessage($source, 1, false);
+
+		self::assertSame($expected, $message->jsonSerialize()['hasAiGeneratedHeader']);
+	}
 }