#import <UIKit/UIKit.h>

@class StyleConfig;
@class ASTNodeWrapper;

NS_ASSUME_NONNULL_BEGIN

typedef void (^LinkPressHandler)(NSString *url, NSString *title);
typedef void (^MentionPressHandler)(NSString *user);
typedef void (^TaskListItemPressHandler)(NSInteger index, BOOL checked);

@interface RenderContext : NSObject

@property (nonatomic, strong) StyleConfig *styleConfig;
@property (nonatomic, strong, nullable) NSSet<NSString *> *customTags;

// Callbacks
@property (nonatomic, copy, nullable) LinkPressHandler onLinkPress;
@property (nonatomic, copy, nullable) LinkPressHandler onLinkLongPress;
@property (nonatomic, copy, nullable) MentionPressHandler onMentionPress;
@property (nonatomic, copy, nullable) TaskListItemPressHandler onTaskListItemPress;

// Rendering state
@property (nonatomic, assign) NSInteger listDepth;
@property (nonatomic, assign) NSInteger orderedListIndex;
/// Whether the innermost enclosing list is ordered. md4c only sets
/// listType on the List node itself, not on the individual list items,
/// so ListItemRenderer needs this on the context instead of reading
/// it off its own node.
@property (nonatomic, assign) BOOL currentListIsOrdered;
/// Digit count of the largest marker in the innermost enclosing
/// ordered list, used to left-pad shorter markers so the periods
/// align (e.g. " 1. … 10." instead of "1. … 10.").
@property (nonatomic, assign) NSInteger currentListMaxMarkerDigits;
@property (nonatomic, assign) BOOL isInsideBlockquote;
@property (nonatomic, assign) BOOL isInsideCodeBlock;
@property (nonatomic, assign) NSInteger taskListIndex;

// Style stack for nested inline styles
@property (nonatomic, strong) NSMutableArray<NSDictionary<NSAttributedStringKey, id> *> *attributeStack;

- (void)pushAttributes:(NSDictionary<NSAttributedStringKey, id> *)attrs;
- (void)popAttributes;
- (NSDictionary<NSAttributedStringKey, id> *)currentAttributes;

- (void)renderChildren:(ASTNodeWrapper *)node
                  into:(NSMutableAttributedString *)output;

/// Renders a top-level block node to an attributed string using a
/// fresh RenderContext, trimming trailing newlines. Thread-safe.
+ (NSAttributedString *)renderNodeToAttributedString:(ASTNodeWrapper *)node
                                         styleConfig:(StyleConfig *)styleConfig
                                          customTags:
                                              (NSArray<NSString *> *)customTags;

/// Like renderNodeToAttributedString: but starts the attribute stack
/// with the given inheritedAttrs instead of the style config's base
/// attrs. Used when rendering a child of a parent block (e.g. a
/// paragraph inside a blockquote) so the parent's text styling
/// cascades down.
+ (NSAttributedString *)renderNodeToAttributedString:(ASTNodeWrapper *)node
                                         styleConfig:(StyleConfig *)styleConfig
                                          customTags:
                                              (NSArray<NSString *> *)customTags
                                      inheritedAttrs:
                                          (nullable NSDictionary<
                                              NSAttributedStringKey, id> *)
                                              inheritedAttrs;

/// Renders a single list item, preserving the ordered index / depth
/// so bullets render correctly. isOrdered reflects the parent list's
/// type — md4c only sets listType on the List node, so the caller
/// (who has the List node in hand) must pass it in. maxMarkerDigits
/// is the digit count of the largest marker in the parent list, used
/// to left-pad shorter markers so the periods align. Thread-safe.
+ (NSAttributedString *)renderListItemContent:(ASTNodeWrapper *)item
                                    isOrdered:(BOOL)isOrdered
                                 orderedIndex:(NSInteger)orderedIndex
                              maxMarkerDigits:(NSInteger)maxMarkerDigits
                                  styleConfig:(StyleConfig *)styleConfig
                                   customTags:(NSArray<NSString *> *)customTags
                               inheritedAttrs:
                                   (nullable NSDictionary<NSAttributedStringKey,
                                                          id> *)inheritedAttrs;

/// Computes the root attribute dict for a given style config (font,
/// color from the base style). Exposed so parent block builders can
/// derive an inheritedAttrs dict and add their own text properties
/// before passing it to a child renderer.
+ (NSDictionary<NSAttributedStringKey, id> *)baseAttributesFromStyleConfig:
    (StyleConfig *)styleConfig;

@end

NS_ASSUME_NONNULL_END