如何实现数据结构前N层格式化打印，其余层紧凑输出？

现有工具情况 #

目前Hackage上没有直接支持按指定深度格式化结构化记录的现成工具，但可以基于pretty、wl-pprint-text这类常用pretty-printer库扩展实现。

手动实现思路 #

核心是给数据类型实现自定义的Pretty实例，打印时跟踪当前深度，达到指定阈值后切换为紧凑格式输出。

1. 定义深度跟踪辅助函数 #

先写一个工具函数，判断当前深度是否超过阈值，决定输出格式：

import Text.PrettyPrint

printWithDepth :: Int -> Int -> Doc -> Doc
printWithDepth current maxDepth doc
  | current >= maxDepth = compact doc  -- 达到阈值用紧凑格式
  | otherwise = doc                    -- 未达到则正常格式化

-- 紧凑格式：直接用show转成字符串输出
compact :: Show a => a -> Doc
compact = text . show

2. 为嵌套数据类型实现Pretty实例 #

假设你有如下嵌套记录类型：

data User = User
  { userName :: String
  , userProfile :: Profile
  } deriving (Show)

data Profile = Profile
  { profileBio :: String
  , profilePosts :: [Post]
  } deriving (Show)

data Post = Post
  { postTitle :: String
  , postContent :: String
  } deriving (Show)

为其实现带深度控制的Pretty实例，比如指定仅格式化第一层：

instance Pretty User where
  pretty user = 
    let maxDepth = 1
    in printWithDepth 0 maxDepth $
       text "User" <> braces
         ( text "userName = " <> text (userName user)
         <> comma
         <> text "userProfile = " <> printWithDepth 1 maxDepth (pretty (userProfile user))
         )

instance Pretty Profile where
  pretty profile = 
    text "Profile" <> braces
      ( text "profileBio = " <> text (profileBio profile)
      <> comma
      <> text "profilePosts = " <> pretty (profilePosts profile)
      )

instance Pretty Post where
  pretty post = 
    text "Post" <> braces
      ( text "postTitle = " <> text (postTitle post)
      <> comma
      <> text "postContent = " <> text (postContent post)
      )

3. 灵活调用打印逻辑 #

如果需要动态指定最大深度，可以封装成专用函数：

prettyWithMaxDepth :: Int -> User -> Doc
prettyWithMaxDepth maxDepth user =
  printWithDepth 0 maxDepth $
    text "User" <> braces
      ( text "userName = " <> text (userName user)
      <> comma
      <> text "userProfile = " <> printWithDepth 1 maxDepth (pretty (userProfile user))
      )

这样当maxDepth=1时，User会被格式化展开，而Profile及内部的Post会以紧凑的字符串形式输出，避免排版过于稀疏冗长。

通用扩展方案 #

如果要支持任意数据类型，可以借助Generic自动推导带深度控制的Pretty实例，减少重复代码。核心逻辑是：

• 遍历Generic类型的结构，同时跟踪当前深度
• 深度超过阈值时，直接使用show输出紧凑格式

这种方式需要熟悉Generic的结构遍历机制，实现复杂度更高，但复用性更强。

内容的提问来源于stack exchange，提问作者user855443