@@ -441,31 +441,58 @@ extension PrettyPrinter {
441
441
break
442
442
443
443
case . intersection( let first, let second) :
444
- printAsPattern ( first)
444
+ if wrap, first. isSimplePrint {
445
+ indent ( )
446
+ output ( " One( " )
447
+ }
448
+
449
+ printAsPattern ( first, wrap: false )
445
450
printIndented { printer in
446
451
printer. indent ( )
447
452
printer. output ( " .intersection( " )
448
- printer. printAsPattern ( second, terminateLine: false )
453
+ printer. printAsPattern ( second, wrap : false , terminateLine: false )
449
454
printer. output ( " ) " )
450
455
}
451
456
457
+ if wrap, first. isSimplePrint {
458
+ output ( " ) " )
459
+ }
460
+
452
461
case . subtraction( let first, let second) :
453
- printAsPattern ( first)
462
+ if wrap, first. isSimplePrint {
463
+ indent ( )
464
+ output ( " One( " )
465
+ }
466
+
467
+ printAsPattern ( first, wrap: false )
454
468
printIndented { printer in
455
469
printer. indent ( )
456
470
printer. output ( " .subtracting( " )
457
- printer. printAsPattern ( second, terminateLine: false )
471
+ printer. printAsPattern ( second, wrap : false , terminateLine: false )
458
472
printer. output ( " ) " )
459
473
}
460
474
475
+ if wrap, first. isSimplePrint {
476
+ output ( " ) " )
477
+ }
478
+
461
479
case . symmetricDifference( let first, let second) :
462
- printAsPattern ( first)
480
+ if wrap, first. isSimplePrint {
481
+ indent ( )
482
+ output ( " One( " )
483
+ }
484
+
485
+ printAsPattern ( first, wrap: false )
463
486
printIndented { printer in
464
487
printer. indent ( )
465
488
printer. output ( " .symmetricDifference( " )
466
- printer. printAsPattern ( second, terminateLine: false )
489
+ printer. printAsPattern ( second, wrap : false , terminateLine: false )
467
490
printer. output ( " ) " )
468
491
}
492
+
493
+ if wrap, first. isSimplePrint {
494
+ output ( " ) " )
495
+ }
469
496
}
470
497
}
471
498
0 commit comments