haiai 0.2.0

Rust SDK for HAI.AI agent benchmarking, designed as a JACS-delegating wrapper
Documentation
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
977
978
use std::time::Duration;

use base64::Engine;
use serde::{Deserialize, Deserializer, Serialize, Serializer};
use serde_json::Value;

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct HelloResult {
    #[serde(default)]
    pub timestamp: String,
    #[serde(default)]
    pub client_ip: String,
    #[serde(default)]
    pub hai_public_key_fingerprint: String,
    #[serde(default)]
    pub message: String,
    #[serde(default)]
    pub hai_signed_ack: String,
    #[serde(default)]
    pub hello_id: String,
    #[serde(default)]
    pub test_scenario: Option<Value>,
}

#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub struct CreateAgentOptions {
    pub name: String,
    pub password: String,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub algorithm: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub data_directory: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub key_directory: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub config_path: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub agent_type: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub description: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub domain: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub default_storage: Option<String>,
}

#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub struct CreateAgentResult {
    #[serde(default)]
    pub agent_id: String,
    #[serde(default)]
    pub name: String,
    #[serde(default)]
    pub public_key_path: String,
    #[serde(default)]
    pub config_path: String,
    #[serde(default)]
    pub version: String,
    #[serde(default)]
    pub algorithm: String,
    #[serde(default)]
    pub private_key_path: String,
    #[serde(default)]
    pub data_directory: String,
    #[serde(default)]
    pub key_directory: String,
    #[serde(default)]
    pub domain: String,
    #[serde(default)]
    pub dns_record: String,
}

/// Options for key rotation.
#[derive(Debug, Clone, Default)]
pub struct RotateKeysOptions {
    /// Whether to re-register with HAI after local rotation. Default: true.
    pub register_with_hai: Option<bool>,
}

/// Result of a key rotation operation.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct RotationResult {
    /// Agent's stable JACS ID (unchanged).
    pub jacs_id: String,
    /// Version before rotation.
    pub old_version: String,
    /// New version assigned during rotation.
    pub new_version: String,
    /// SHA-256 hash of the new public key (hex).
    pub new_public_key_hash: String,
    /// Whether re-registration with HAI succeeded.
    pub registered_with_hai: bool,
    /// Complete self-signed agent JSON string.
    pub signed_agent_json: String,
}

/// Result of an agent document update (metadata re-sign with existing key).
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct UpdateAgentResult {
    /// Agent's stable JACS ID (unchanged).
    pub jacs_id: String,
    /// Version before the update.
    pub old_version: String,
    /// New version assigned during the update.
    pub new_version: String,
    /// Complete self-signed agent JSON string.
    pub signed_agent_json: String,
    /// Whether re-registration with HAI succeeded.
    #[serde(default)]
    pub registered_with_hai: bool,
}

/// Result of a legacy agent migration.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct MigrateAgentResult {
    /// Agent's stable JACS ID (unchanged).
    pub jacs_id: String,
    /// Version before migration.
    pub old_version: String,
    /// New version assigned during migration.
    pub new_version: String,
    /// Fields that were patched in the raw JSON before loading (e.g. `["iat", "jti"]`).
    pub patched_fields: Vec<String>,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct CheckUsernameResult {
    #[serde(default)]
    pub available: bool,
    #[serde(default)]
    pub username: String,
    #[serde(default)]
    pub reason: Option<String>,
}

#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub struct RegisterAgentOptions {
    pub agent_json: String,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub public_key_pem: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub owner_email: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub domain: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub description: Option<String>,
}

#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub struct RegistrationResult {
    #[serde(default)]
    pub success: bool,
    #[serde(default)]
    pub agent_id: String,
    #[serde(default)]
    pub jacs_id: String,
    #[serde(default)]
    pub dns_verified: bool,
    #[serde(default)]
    pub registrations: Vec<RegistrationEntry>,
    #[serde(default)]
    pub registered_at: String,
    #[serde(default)]
    pub message: Option<String>,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct RegistrationEntry {
    #[serde(default)]
    pub key_id: String,
    #[serde(default)]
    pub algorithm: String,
    #[serde(default)]
    pub signature_json: String,
    #[serde(default)]
    pub signed_at: String,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct VerifyAgentResult {
    #[serde(default)]
    pub jacs_id: String,
    #[serde(default)]
    pub registered: bool,
    #[serde(default)]
    pub registrations: Vec<RegistrationEntry>,
    #[serde(default)]
    pub dns_verified: bool,
    #[serde(default)]
    pub registered_at: String,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct JobResponseResult {
    #[serde(default)]
    pub success: bool,
    #[serde(default)]
    pub job_id: String,
    #[serde(default)]
    pub message: String,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ClaimUsernameResult {
    #[serde(default)]
    pub username: String,
    #[serde(default)]
    pub email: String,
    #[serde(default)]
    pub agent_id: String,
}

#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub struct UpdateUsernameResult {
    #[serde(default)]
    pub username: String,
    #[serde(default)]
    pub email: String,
    #[serde(default)]
    pub previous_username: String,
}

#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub struct DeleteUsernameResult {
    #[serde(default)]
    pub released_username: String,
    #[serde(default)]
    pub cooldown_until: String,
    #[serde(default)]
    pub message: String,
}

#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq)]
#[serde(rename_all = "snake_case")]
pub enum FieldStatus {
    Pass,
    Modified,
    Fail,
    Unverifiable,
}

#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq)]
pub struct FieldResult {
    #[serde(default)]
    pub field: String,
    pub status: FieldStatus,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub original_hash: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub current_hash: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub original_value: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub current_value: Option<String>,
}

#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq)]
pub struct ChainEntry {
    #[serde(default)]
    pub signer: String,
    #[serde(default)]
    pub jacs_id: String,
    #[serde(default)]
    pub valid: bool,
    #[serde(default)]
    pub forwarded: bool,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct EmailVerificationResultV2 {
    pub valid: bool,
    #[serde(default)]
    pub jacs_id: String,
    #[serde(default)]
    pub algorithm: String,
    #[serde(default)]
    pub reputation_tier: String,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub dns_verified: Option<bool>,
    #[serde(default)]
    pub field_results: Vec<FieldResult>,
    #[serde(default, skip_serializing_if = "Vec::is_empty")]
    pub chain: Vec<ChainEntry>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub error: Option<String>,
    /// Agent status from registry: "active", "suspended", or "revoked".
    #[serde(skip_serializing_if = "Option::is_none")]
    pub agent_status: Option<String>,
    /// Benchmark tiers the agent has completed.
    #[serde(default, skip_serializing_if = "Vec::is_empty")]
    pub benchmarks_completed: Vec<String>,
}

impl EmailVerificationResultV2 {
    pub fn err(jacs_id: &str, reputation_tier: &str, error: &str) -> Self {
        Self {
            valid: false,
            jacs_id: jacs_id.to_string(),
            algorithm: String::new(),
            reputation_tier: reputation_tier.to_string(),
            dns_verified: None,
            field_results: Vec::new(),
            chain: Vec::new(),
            error: Some(error.to_string()),
            agent_status: None,
            benchmarks_completed: Vec::new(),
        }
    }
}

/// An email attachment with raw data.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct EmailAttachment {
    pub filename: String,
    pub content_type: String,
    /// Raw attachment bytes. Serialized as base64 `data_base64` for the API.
    #[serde(skip)]
    pub data: Vec<u8>,
    /// Base64-encoded data sent to the API.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub data_base64: Option<String>,
}

impl EmailAttachment {
    /// Create a new attachment from raw bytes.
    ///
    /// `data_base64` is left as `None` and will be computed automatically
    /// during `send_email`.
    pub fn new(filename: String, content_type: String, data: Vec<u8>) -> Self {
        Self {
            filename,
            content_type,
            data,
            data_base64: None,
        }
    }

    /// Return the raw attachment bytes, falling back to decoding `data_base64`
    /// when `data` is empty.
    pub fn effective_data(&self) -> Vec<u8> {
        if !self.data.is_empty() {
            return self.data.clone();
        }
        if let Some(ref b64) = self.data_base64 {
            if let Ok(decoded) = base64::engine::general_purpose::STANDARD.decode(b64) {
                return decoded;
            }
        }
        Vec::new()
    }
}

#[derive(Debug, Clone, Default, Serialize, Deserialize)]
pub struct SendEmailOptions {
    pub to: String,
    pub subject: String,
    pub body: String,
    /// CC recipients
    #[serde(default, skip_serializing_if = "Vec::is_empty")]
    pub cc: Vec<String>,
    /// BCC recipients (not visible to other recipients)
    #[serde(default, skip_serializing_if = "Vec::is_empty")]
    pub bcc: Vec<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub in_reply_to: Option<String>,
    #[serde(default, skip_serializing_if = "Vec::is_empty")]
    pub attachments: Vec<EmailAttachment>,
    /// Labels/tags to apply to the sent message
    #[serde(default, skip_serializing_if = "Vec::is_empty")]
    pub labels: Vec<String>,
    /// Whether to append a verification footer for external recipients.
    /// Default (None) = true: appends footer for non-@hai.ai recipients.
    /// Set to Some(false) to suppress the footer.
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub append_footer: Option<bool>,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct SendEmailResult {
    #[serde(default)]
    pub message_id: String,
    #[serde(default)]
    pub status: String,
}

#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub struct ListMessagesOptions {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub limit: Option<u32>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub offset: Option<u32>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub direction: Option<String>,
    /// Filter by read status: true=read only, false=unread only, None=all
    #[serde(skip_serializing_if = "Option::is_none")]
    pub is_read: Option<bool>,
    /// Filter by folder: "inbox", "sent", "archive", "trash"
    #[serde(skip_serializing_if = "Option::is_none")]
    pub folder: Option<String>,
    /// Filter by label
    #[serde(skip_serializing_if = "Option::is_none")]
    pub label: Option<String>,
    /// Filter by attachment presence
    #[serde(skip_serializing_if = "Option::is_none")]
    pub has_attachments: Option<bool>,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct EmailMessage {
    #[serde(default)]
    pub id: String,
    #[serde(default)]
    pub direction: String,
    #[serde(default)]
    pub from_address: String,
    #[serde(default)]
    pub to_address: String,
    #[serde(default)]
    pub subject: String,
    #[serde(default)]
    pub body_text: String,
    #[serde(default)]
    pub message_id: Option<String>,
    #[serde(default)]
    pub in_reply_to: Option<String>,
    #[serde(default)]
    pub is_read: bool,
    #[serde(default)]
    pub delivery_status: String,
    #[serde(default)]
    pub created_at: String,
    #[serde(default)]
    pub read_at: Option<String>,
    #[serde(default)]
    pub jacs_verified: Option<bool>,
    /// CC recipients on this message
    #[serde(default, skip_serializing_if = "Vec::is_empty")]
    pub cc_addresses: Vec<String>,
    /// Labels/tags applied to this message
    #[serde(default, skip_serializing_if = "Vec::is_empty")]
    pub labels: Vec<String>,
    /// MUSUBI composite trust score (0-100). Higher = safer.
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub trust_score: Option<f32>,
    /// Clean body text with quoted reply text removed (derived at response time)
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub body_text_clean: Option<String>,
    /// Quoted reply text extracted from the body (derived at response time)
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub quoted_text: Option<String>,
    /// Thread context: other messages in the same conversation (populated on get_message)
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub thread: Option<Vec<EmailMessage>>,
}

#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub struct SearchOptions {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub q: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub direction: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub from_address: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub to_address: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub since: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub until: Option<String>,
    /// Filter by read status
    #[serde(skip_serializing_if = "Option::is_none")]
    pub is_read: Option<bool>,
    /// Filter by JACS verification status
    #[serde(skip_serializing_if = "Option::is_none")]
    pub jacs_verified: Option<bool>,
    /// Filter by folder (inbox, sent, archive, trash)
    #[serde(skip_serializing_if = "Option::is_none")]
    pub folder: Option<String>,
    /// Filter by label
    #[serde(skip_serializing_if = "Option::is_none")]
    pub label: Option<String>,
    /// Filter by attachment presence
    #[serde(skip_serializing_if = "Option::is_none")]
    pub has_attachments: Option<bool>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub limit: Option<u32>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub offset: Option<u32>,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct EmailStatus {
    #[serde(default)]
    pub email: String,
    #[serde(default)]
    pub status: String,
    #[serde(default)]
    pub tier: String,
    #[serde(default)]
    pub billing_tier: String,
    #[serde(default)]
    pub messages_sent_24h: i32,
    #[serde(default)]
    pub daily_limit: i32,
    #[serde(default)]
    pub daily_used: i32,
    #[serde(default)]
    pub resets_at: String,
    #[serde(default)]
    pub messages_sent_total: i32,
    #[serde(default)]
    pub external_enabled: bool,
    #[serde(default)]
    pub external_sends_today: i32,
    #[serde(default)]
    pub last_tier_change: Option<String>,
    /// Volume statistics (from consolidated status)
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub volume: Option<EmailVolumeInfo>,
    /// Delivery metrics (from consolidated status)
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub delivery: Option<EmailDeliveryInfo>,
    /// Reputation scoring (from consolidated status)
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub reputation: Option<EmailReputationInfo>,
}

/// Volume statistics from the email status response.
#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub struct EmailVolumeInfo {
    #[serde(default)]
    pub sent_total: i64,
    #[serde(default)]
    pub received_total: i64,
    #[serde(default)]
    pub sent_24h: i64,
}

/// Delivery metrics from the email status response.
#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub struct EmailDeliveryInfo {
    #[serde(default)]
    pub bounce_count: i32,
    #[serde(default)]
    pub spam_report_count: i32,
    #[serde(default)]
    pub delivery_rate: f64,
}

/// Reputation scoring from the email status response.
#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub struct EmailReputationInfo {
    #[serde(default)]
    pub score: f64,
    #[serde(default)]
    pub tier: String,
    #[serde(default)]
    pub email_score: f64,
    #[serde(default)]
    pub hai_score: Option<f64>,
}

/// A contact (correspondent) from the email contacts endpoint.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct Contact {
    #[serde(default)]
    pub email: String,
    #[serde(default)]
    pub display_name: Option<String>,
    #[serde(default)]
    pub last_contact: String,
    #[serde(default)]
    pub jacs_verified: bool,
    #[serde(default)]
    pub reputation_tier: Option<String>,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct KeyRegistryResponse {
    #[serde(default)]
    pub email: String,
    #[serde(default)]
    pub jacs_id: String,
    #[serde(default)]
    pub public_key: String,
    #[serde(default)]
    pub algorithm: String,
    #[serde(default)]
    pub reputation_tier: String,
    #[serde(default)]
    pub registered_at: String,
    /// Agent status: "active", "suspended", or "revoked".
    #[serde(default)]
    pub agent_status: Option<String>,
    /// Benchmark tiers the agent has completed (e.g., ["free", "pro"]).
    #[serde(default)]
    pub benchmarks_completed: Option<Vec<String>>,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct PublicKeyInfo {
    #[serde(default)]
    pub jacs_id: String,
    #[serde(default)]
    pub version: String,
    #[serde(default)]
    pub public_key: String,
    #[serde(default)]
    pub public_key_raw_b64: String,
    #[serde(default)]
    pub algorithm: String,
    #[serde(default)]
    pub public_key_hash: String,
    #[serde(default)]
    pub status: String,
    #[serde(default)]
    pub dns_verified: bool,
    #[serde(default)]
    pub created_at: String,
}

/// Response from `GET /jacs/v1/agents/{jacs_id}/keys` — all key versions for an agent.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct AgentKeyHistory {
    #[serde(default)]
    pub jacs_id: String,
    #[serde(default)]
    pub keys: Vec<PublicKeyInfo>,
    #[serde(default)]
    pub total: usize,
}

#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub struct DocumentVerificationResult {
    #[serde(default)]
    pub valid: bool,
    #[serde(default)]
    pub verified_at: String,
    #[serde(default)]
    pub document_type: String,
    #[serde(default)]
    pub issuer_verified: bool,
    #[serde(default)]
    pub signature_verified: bool,
    #[serde(default)]
    pub signer_id: String,
    #[serde(default)]
    pub signed_at: String,
    #[serde(default)]
    pub error: Option<String>,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct SignedPayload {
    pub signed_document: String,
    pub agent_jacs_id: String,
}

#[derive(Debug, Clone, Copy, Serialize, Deserialize, PartialEq, Eq, Default)]
#[serde(rename_all = "lowercase")]
pub enum TransportType {
    #[default]
    Sse,
    Ws,
}

impl TransportType {
    pub fn as_str(self) -> &'static str {
        match self {
            Self::Sse => "sse",
            Self::Ws => "ws",
        }
    }
}

#[derive(Debug, Clone)]
pub struct ProRunOptions {
    pub transport: TransportType,
    pub poll_interval: Duration,
    pub poll_timeout: Duration,
}

impl Default for ProRunOptions {
    fn default() -> Self {
        Self {
            transport: TransportType::Sse,
            poll_interval: Duration::from_secs(2),
            poll_timeout: Duration::from_secs(300),
        }
    }
}

/// Deprecated: Use `ProRunOptions` instead.
pub type DnsCertifiedRunOptions = ProRunOptions;

#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub struct TranscriptMessage {
    #[serde(default)]
    pub role: String,
    #[serde(default)]
    pub content: String,
    #[serde(default)]
    pub timestamp: String,
    #[serde(default)]
    pub annotations: Vec<String>,
}

#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub struct FreeChaoticResult {
    #[serde(default)]
    pub success: bool,
    #[serde(default)]
    pub run_id: String,
    #[serde(default)]
    pub transcript: Vec<TranscriptMessage>,
    #[serde(default)]
    pub upsell_message: String,
    #[serde(default)]
    pub raw_response: Value,
}

#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub struct ProRunResult {
    #[serde(default)]
    pub success: bool,
    #[serde(default)]
    pub run_id: String,
    #[serde(default)]
    pub score: f64,
    #[serde(default)]
    pub transcript: Vec<TranscriptMessage>,
    #[serde(default)]
    pub payment_id: String,
    #[serde(default)]
    pub raw_response: Value,
}

/// Deprecated: Use `ProRunResult` instead.
pub type DnsCertifiedResult = ProRunResult;

// =============================================================================
// Document & Search Types (SDK boundary)
// =============================================================================

/// A signed document returned by document operations.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct SignedDocument {
    /// The document key (`id:version`).
    pub key: String,
    /// The signed document JSON.
    pub json: String,
}

/// Results from a search operation.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct DocSearchResults {
    /// The matched documents, ordered by relevance.
    pub results: Vec<DocSearchHit>,
    /// Total number of matching documents (for pagination).
    pub total_count: usize,
    /// Which search method the backend used.
    pub method: String,
}

/// A single search result with relevance metadata.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct DocSearchHit {
    /// The document key (`id:version`).
    pub key: String,
    /// The signed document JSON.
    pub json: String,
    /// Relevance score (0.0 - 1.0).
    pub score: f64,
    /// Which field(s) matched, if applicable.
    pub matched_fields: Vec<String>,
}

/// Capabilities of the configured storage backend.
#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub struct StorageCapabilities {
    /// Whether fulltext search is supported.
    pub fulltext: bool,
    /// Whether vector similarity search is supported.
    pub vector: bool,
    /// Whether field-level queries are supported.
    pub query_by_field: bool,
    /// Whether type-based queries are supported.
    pub query_by_type: bool,
    /// Whether pagination is supported.
    pub pagination: bool,
    /// Whether soft-delete (tombstone) is used instead of hard delete.
    pub tombstone: bool,
}

/// Result of a document verification operation.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct DocVerificationResult {
    /// The document key (if available).
    pub key: String,
    /// Whether the document is valid.
    pub valid: bool,
    /// Error message if verification failed.
    pub error: Option<String>,
    /// ID of the agent that signed the document.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub signer_id: Option<String>,
    /// ISO 8601 timestamp of when the document was signed.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub timestamp: Option<String>,
    /// Name of the signer (if available).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub signer_name: Option<String>,
}

#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub struct VerificationStatus {
    #[serde(default)]
    pub jacs_valid: bool,
    #[serde(default)]
    pub dns_valid: bool,
    #[serde(default)]
    pub hai_registered: bool,
    #[serde(default)]
    pub badge: String,
}

#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub struct AgentVerificationResult {
    #[serde(default)]
    pub agent_id: String,
    #[serde(default)]
    pub verification: VerificationStatus,
    #[serde(default)]
    pub hai_signatures: Vec<String>,
    #[serde(default)]
    pub verified_at: String,
    #[serde(default)]
    pub errors: Vec<String>,
}

#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub struct VerifyAgentDocumentRequest {
    pub agent_json: String,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub public_key: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub domain: Option<String>,
}

#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub struct HaiEvent {
    #[serde(default)]
    pub event_type: String,
    #[serde(default)]
    pub data: Value,
    #[serde(default)]
    pub id: Option<String>,
    #[serde(default)]
    pub raw: String,
}

// ============================================================================
// Email Template Types
// ============================================================================

/// An email template with reusable instructions for agent email workflows.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct EmailTemplate {
    pub id: String,
    pub agent_id: String,
    pub name: String,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub how_to_send: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub how_to_respond: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub goal: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub rules: Option<String>,
    pub created_at: String,
    pub updated_at: String,
}

/// Options for creating an email template.
#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub struct CreateEmailTemplateOptions {
    pub name: String,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub how_to_send: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub how_to_respond: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub goal: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub rules: Option<String>,
}

/// Serialize `Option<Option<String>>` so that `Some(None)` becomes JSON `null`
/// and `None` is skipped (via `skip_serializing_if`).
fn serialize_double_option<S>(value: &Option<Option<String>>, serializer: S) -> Result<S::Ok, S::Error>
where
    S: Serializer,
{
    match value {
        Some(inner) => inner.serialize(serializer),
        None => serializer.serialize_none(),
    }
}

/// Deserialize a double-option field: if the field is present in JSON, the outer
/// Option becomes `Some`. The inner Option distinguishes `null` from a string value.
fn deserialize_double_option<'de, D>(deserializer: D) -> Result<Option<Option<String>>, D::Error>
where
    D: Deserializer<'de>,
{
    let value: Option<String> = Option::deserialize(deserializer)?;
    Ok(Some(value))
}

/// Options for updating an email template (all fields optional).
///
/// For the four text fields (`how_to_send`, `how_to_respond`, `goal`, `rules`):
/// - `None` — field is omitted from the request (don't change the current value)
/// - `Some(None)` — field is sent as `null` (clear the value to NULL)
/// - `Some(Some(val))` — field is sent with a value (set to `val`)
#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub struct UpdateEmailTemplateOptions {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub name: Option<String>,
    #[serde(
        default,
        skip_serializing_if = "Option::is_none",
        serialize_with = "serialize_double_option",
        deserialize_with = "deserialize_double_option"
    )]
    pub how_to_send: Option<Option<String>>,
    #[serde(
        default,
        skip_serializing_if = "Option::is_none",
        serialize_with = "serialize_double_option",
        deserialize_with = "deserialize_double_option"
    )]
    pub how_to_respond: Option<Option<String>>,
    #[serde(
        default,
        skip_serializing_if = "Option::is_none",
        serialize_with = "serialize_double_option",
        deserialize_with = "deserialize_double_option"
    )]
    pub goal: Option<Option<String>>,
    #[serde(
        default,
        skip_serializing_if = "Option::is_none",
        serialize_with = "serialize_double_option",
        deserialize_with = "deserialize_double_option"
    )]
    pub rules: Option<Option<String>>,
}

/// Options for listing/searching email templates.
#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub struct ListEmailTemplatesOptions {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub limit: Option<u32>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub offset: Option<u32>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub q: Option<String>,
}

/// Result of listing email templates.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ListEmailTemplatesResult {
    pub templates: Vec<EmailTemplate>,
    pub total: i64,
    pub limit: i64,
    pub offset: i64,
}