Skip to content

Latest commit

 

History

History
555 lines (462 loc) · 19 KB

README.md

File metadata and controls

555 lines (462 loc) · 19 KB

#! https://zhuanlan.zhihu.com/p/426966480

用 TypeScript 类型运算实现一个中国象棋程序

众所周知,TypeScript 是图灵完备的,因此,只要我们愿意,那当然是可以用它来实现一个象棋程序的。于是我们就快乐地开始了,为了理解方便,我们不考虑性能优化策略,纯粹从功能实现角度去构建。另外,我们尝试用中文来编写这个程序,因为类型运算中需要用到的操作符很少,类型本质上是一种对现实世界的描述,某种程度上算是一种业务描述语言,使用中文也挺好的。

需求拆解

首先我们要先来分析一下,需要实现哪些功能。

象棋的核心功能就是走棋,走棋如果用一个函数来表达,那就是:走棋<当前棋局, 源位置, 目标位置>,因此,这里就涉及两个要素:

  • 棋局结构:棋子行、列的组织
    • 棋子结构:类型、颜色
    • 位置:横坐标,纵坐标
  • 棋子的移动
    • 这个棋子能放在这里吗?
    • 这个棋子能这么移动吗?

比较多的细节会出现在棋子的移动规则上,但是,不重要,都是细节,我们挨个来解释吧。

基础结构

棋子

棋子的话,比较简单,没有什么特别的,只要表达出颜色和种类就可以了:

type 棋子 = {
  颜色: 棋子颜色
  种类: 棋子种类
}

type 构造棋子<某颜色, 某种类> = 某颜色 extends 棋子颜色
  ? 某种类 extends 棋子种类
    ? {
        颜色: 某颜色
        种类: 某种类
      }
    : 不可能
  : 不可能

本节使用的技巧非常简单,就是一个合法性判断而已。

位置

位置就会复杂不少了,主要原因是,在类型系统里面,我们要表示整数及其运算是一件比较麻烦的事情,而偏偏在计算走棋规则的时候,这部分很重要,所以,不得不去做。

定义整数和加减法的部分,在其他文章里可以找到,不在这里叙述了,假定已经存在了一些常用的整数变量,可以得到对于位置信息的表达:

type 棋子横坐标 =  |  |  |  |  |  |  |  | 
type 棋子纵坐标 =  |  |  |  |  |  |  |  |  | 

type 最大横坐标 = 
type 最大纵坐标 = 

type 棋子坐标 = {
  : 棋子横坐标
  : 棋子纵坐标
}

type 构造棋子坐标<横坐标, 纵坐标> = 横坐标 extends 棋子横坐标
  ? 纵坐标 extends 棋子纵坐标
    ? {
        : 横坐标
        : 纵坐标
      }
    : 不可能
  : 不可能

位置比棋子要复杂的原因是,要支持一些运算,比如,判断两个位置是否相同

type 相同位置<位置1, 位置2> = 位置1 extends 棋子坐标
  ? 位置2 extends 棋子坐标
    ? 相等<位置1['横'], 位置2['横']> & 相等<位置1['纵'], 位置2['纵']>
    : 不可能
  : 不可能

然后,我们又会需要,对一个位置进行移动,用来定义寻找相对位置的快捷操作,比如,上下左右相邻位置的坐标,需要注意的是,这里要做一下合法性校验,避免越界。

type 左邻位<位置> = 位置 extends 棋子坐标
  ? 大于<位置['横'], > & 构造棋子坐标<减一<位置['横']>, 位置['纵']>
  : 不可能

type 右邻位<位置> = 位置 extends 棋子坐标
  ? 小于<位置['横'], 最大横坐标> & 构造棋子坐标<加一<位置['横']>, 位置['纵']>
  : 不可能

type 上邻位<位置> = 位置 extends 棋子坐标
  ? 大于<位置['纵'], > & 构造棋子坐标<位置['横'], 减一<位置['纵']>>
  : 不可能

type 下邻位<位置> = 位置 extends 棋子坐标
  ? 小于<位置['纵'], 最大纵坐标> & 构造棋子坐标<位置['横'], 加一<位置['纵']>>
  : 不可能

本节使用的规则稍稍复杂一些,但主要复杂度不在这里,而是在定义数字运算的地方。

棋局

在棋局这里,我们第一次面对了复杂的状况,整个这部分,需要解决这么一些问题:

  • 棋子的行列结构如何构造?
  • 如何基于某个位置的横纵坐标进行棋子的获取?
  • 如何为棋子规则提供必要的帮助?
  • 如何移动棋局中的棋子?

我们逐一来解决这些问题。

棋局结构

棋局结构的建模,是与取值操作密切相关的,需要注意到的是,我们并没有使用数字字面量去代表数字,而是自定义了整数类型,这就意味着,取值操作的索引是很复杂的,它无法直接得到。

在我们使用的整数定义中,数字是用一级一级嵌套表达的,当我们表示数字,其等价形态为:

type 整数 =  | { 前一个数: 整数; 是零吗:  }

type  = 加一<加一<加一<加一<加一<>>>>>

当我们得到一个数字,并且想要知道它真正表达的是多少,需要去统计从它到零的距离,也就是说,要反向去根据是零吗前一个数这两个类型属性,一直往前找到零的位置。

有鉴于此,我们选择把棋局也构建为类似形式:

  • 棋局的内容是棋局行形成的链表,棋局持有第零行
  • 每行的内容是单元格形成的链表,棋局行持有第零格
type 构造棋局单元<内容参数, 下一个> = 内容参数 extends 单元格内容参数
  ? 下一个 extends 棋局单元 | 
    ? {
        内容: 内容参数
        下一个: 下一个
      }
    : 不可能
  : 不可能

type 构造棋局行<内容参数, 下一行> = 内容参数 extends 行内容参数
  ? 下一行 extends 棋局行 | 
    ? 内容参数 extends [
        infer 第零格,
        infer 第一格,
        infer 第二格,
        infer 第三格,
        infer 第四格,
        infer 第五格,
        infer 第六格,
        infer 第七格,
        infer 第八格
      ]
      ? {
          内容: 构造棋局单元<
            第零格,
            构造棋局单元<
              第一格,
              构造棋局单元<
                第二格,
                构造棋局单元<
                  第三格,
                  构造棋局单元<
                    第四格,
                    构造棋局单元<
                      第五格,
                      构造棋局单元<
                        第六格,
                        构造棋局单元<第七格, 构造棋局单元<第八格, >>
                      >
                    >
                  >
                >
              >
            >
          >
          下一行: 下一行
        }
      : 不可能
    : 不可能
  : 不可能

type 构造棋局<内容 extends 棋局内容参数 = 棋局内容参数> = 内容 extends [
  infer 第零行,
  infer 第一行,
  infer 第二行,
  infer 第三行,
  infer 第四行,
  infer 第五行,
  infer 第六行,
  infer 第七行,
  infer 第八行,
  infer 第九行
]
  ? {
      内容: 构造棋局行<
        第零行,
        构造棋局行<
          第一行,
          构造棋局行<
            第二行,
            构造棋局行<
              第三行,
              构造棋局行<
                第四行,
                构造棋局行<
                  第五行,
                  构造棋局行<
                    第六行,
                    构造棋局行<
                      第七行,
                      构造棋局行<第八行, 构造棋局行<第九行, >>
                    >
                  >
                >
              >
            >
          >
        >
      >
    }
  : 不可能

这个看上去非常诡异,但是我们后续对它的取值就很简单了。

本节主要使用了一个 infer 来定义临时变量,然后就是递归去构造了(这里可能是有简便写法的,不过我暂时想不到)。

棋局的取值

这一步,我们的目标是实现根据给定位置,从棋局中获取对应的棋子。

先来考虑如何根据一个行号获取指定行,其实很简单,让行号与棋局行链表同步移动就可以了,行号每次减一,行链表每次后移一行,那么,当行号减小到零的时候,行链表指向的就是所要的那行了:

type 根据链表获取棋局指定行<
  行链表 extends 棋局行 | ,
  行号 extends 棋子纵坐标
> = 行链表 extends 棋局行
  ?  extends 相等<行号, >
    ? 行链表
    : 根据链表获取棋局指定行<行链表['下一行'], 减一<行号>>
  : 不可能

同理,在一行中获取某一列,也是一样的方式:

type 根据链表获取棋局某行的指定单元格<
  单元链表 extends 棋局单元 | ,
  列号 extends 棋子横坐标
> = 单元链表 extends 棋局单元
  ?  extends 相等<列号, >
    ? 单元链表
    : 根据链表获取棋局某行的指定单元格<单元链表['下一个'], 减一<列号>>
  : 不可能

我们把这两者组合起来,即可得到根据位置查找棋子的类型函数:

type 获取棋局某位置的单元<
  某棋局 extends 棋局,
  某位置 extends 棋子坐标
> = 获取某行的指定单元格<获取棋局指定行<某棋局, 某位置['纵']>, 某位置['横']>

本节的主要技巧是类型递归,获取行、单元格的时候,都是递归调用当前的类型函数自身去取值的。

辅助规则的提供

在计算棋子位置关系的时候,我们会需要一些判断规则,这些规则的实现,都可以通过组合现有函数的方式去实现。

一些简单规则我们就不去探讨了,比如说,马的“日”字顶角,象的“田”字顶角,还有马脚,象眼等等,都可以反复组合之前我们做好的计算邻位的四个函数,比如:

type 象可以走到右上的目标位置<当前位置> = 右邻位<
  右邻位<上邻位<上邻位<当前位置>>>
>

我们重点来看一个相对复杂的:统计一条直线上两个棋子之间的棋子数量。这里的麻烦在于,统计个数需要有一个累加计数功能,有趣的是,它仍然可以利用递归。

/**
 * 当需要寻找一条线上两点之间有几个棋子的时候,步骤是:
 * - 如果两点重合,返回零
 * - 否则沿着起点方向逐步向终点靠近,如果下一个点不是终点,并且有棋子,则累加一
 */
type 同一水平线上两个位置之间有几个棋子<
  纵坐标,
  横坐标1,
  横坐标2,
  当前棋局 extends 棋局,
  初始值 extends 整数
> = 将两个整数从小到大排序<[横坐标1, 横坐标2]> extends [infer , infer ]
  ?  extends 棋子横坐标
    ?  extends 棋子横坐标
      ?  extends 相等<, >
        ? 初始值
        : 同一水平线上两个位置之间有几个棋子<
            纵坐标,
            加一<>,
            ,
            当前棋局,
             extends 相等<加一<>, >
              ? 初始值 // 下一个就是终点了,中间没有棋子可以累加了
              :  extends 该位置有棋子<
                  当前棋局,
                  构造棋子坐标<加一<>, 纵坐标>
                >
              ? 加一<初始值>
              : 初始值
          >
      : 不可能
    : 不可能
  : 不可能

本节使用了一个小技巧,先把两个坐标排序了一下。这样后续的递归方向可以不需要判断,否则要考虑两个方向。

棋子的移动

棋子的移动,本质上很简单,就是把目标位置的棋子内容替换掉就可以了,但是,在 TypeScript 的类型系统里,“替换”是个不太可行的事情,它是 immutable 的,所谓的替换,实际上是创建了新的。

这样,我们就得实现棋局每个层级的拷贝构造操作了,这也不难:

type 从内容构造棋局单元<
  内容参数 extends 单元格内容参数,
  下一个 extends 棋局单元 | 
> = {
  内容: 内容参数
  下一个: 下一个
}

type 从内容构造棋局行<单元链表 extends 棋局单元, 下一行 extends 棋局行 | > = {
  内容: 单元链表
  下一行: 下一行
}

type 从内容构造棋局<行链表 extends 棋局行> = {
  内容: 行链表
}

然后,真正复杂的是替换过程中的重新构造,以替换棋局中的某一行为例,这实际上是一个一边遍历,一边根据需要构造的过程,本质上是一个 copy on write 结构,理解了这点之后,写起来就不难了,仍然是递归:

type 将棋局的某行替换为指定行<
  某棋局 extends 棋局,
  行号 extends 棋子纵坐标,
  新内容 extends 棋局单元,
  当前迭代行号 extends 整数
> = 当前迭代行号 extends 棋子纵坐标
  ? 根据链表获取棋局指定行<
      某棋局['内容'],
      当前迭代行号
    > extends infer 原先的这行
    ? 原先的这行 extends 棋局行
      ?  extends 相等<行号, 当前迭代行号>
        ? 从内容构造棋局行<新内容, 原先的这行['下一行']> // 如果是要替换的这行,重新构造整个行,把原来的后续单元挂接过来就行了
        : 从内容构造棋局行<
            原先的这行['内容'],
            将棋局的某行替换为指定行<某棋局, 行号, 新内容, 加一<当前迭代行号>>
          >
      : 不可能
    : 不可能
  : 

同理,替换单元格也可以写出来:

type 将棋局行的某单元替换为指定单元<
  某行 extends 棋局行,
  列号 extends 棋子横坐标,
  新内容 extends 单元格内容参数,
  当前迭代列号 extends 整数
> = 当前迭代列号 extends 棋子横坐标
  ? 根据链表获取棋局某行的指定单元格<
      某行['内容'],
      当前迭代列号
    > extends infer 原先的这个单元
    ? 原先的这个单元 extends 棋局单元
      ?  extends 相等<列号, 当前迭代列号>
        ? 从内容构造棋局单元<新内容, 原先的这个单元['下一个']> // 如果是要替换的单元格,重新构造这个单元格,把原来的后续单元挂接过来就行了
        : 从内容构造棋局单元<
            原先的这个单元['内容'],
            将棋局行的某单元替换为指定单元<
              某行,
              列号,
              新内容,
              加一<当前迭代列号>
            >
          >
      : 不可能
    : 不可能
  : 

然后组合两者,就可以得到在棋局上指定位置替换棋子的函数:

type 将棋局某位置替换为指定棋子<
  某棋局 extends 棋局,
  某位置 extends 棋子坐标,
  某棋子或者空 extends 单元格内容参数
> = 获取棋局某位置的单元<某棋局, 某位置> extends infer 原位置的单元
  ? 原位置的单元 extends 棋局单元
    ? 获取棋局指定行<某棋局, 某位置['纵']> extends infer 原有的行
      ? 原有的行 extends 棋局行
        ? 从内容构造棋局<
            将棋局的某行替换为指定行<
              某棋局,
              某位置['纵'],
              将棋局行的某单元替换为指定单元<
                原有的行,
                某位置['横'],
                某棋子或者空,
                
              >,
              
            >
          >
        : 不可能
      : 不可能
    : 不可能
  : 不可能

两次使用这个替换,即可得到走棋的函数:

type 走棋<
  当前棋局 extends 棋局,
  起始位置 extends 棋子坐标,
  目标位置 extends 棋子坐标
> = 获取棋局某位置的棋子<当前棋局, 起始位置> extends infer 待移动的棋子
  ? 待移动的棋子 extends 棋子
    ?  extends 棋子可以从这里走到那里吗<起始位置, 目标位置, 当前棋局>
      ? 将棋局某位置替换为指定棋子<
          将棋局某位置替换为指定棋子<当前棋局, 起始位置, >,
          目标位置,
          待移动的棋子
        >
      : 不可能
    : 不可能
  : 不可能

本节没有出现新的技巧,只是因为之前的链表结构实现,导致复制起来相对复杂一些。

棋局的渲染

当我们建立了棋局的走棋规则之后,实际上象棋已经可用了,但为了显示的友好,还需要多做一步。

最基础的渲染单元,是单个的单元格,这个非常简单,直接根据颜色和种类进行两级映射就可以了。

type 渲染棋子<某个棋子> = 某个棋子 extends 棋子
  ? {
      : {
        : '帅'
        : '仕'
        : '相'
        : '傌'
        : '俥'
        : '炮'
        : '兵'
      }
      : {
        : '将'
        : '士'
        : '象'
        : '马'
        : '车'
        : '炮'
        : '卒'
      }
    }[某个棋子['颜色']][某个棋子['种类']]
  : 不可能

type 渲染单元格<某单元格> = 某单元格 extends 棋子 ? 渲染棋子<某单元格> : '➕'

稍微复杂一些的是渲染行,考虑到我们之前把棋局行实现成了单元格链表,这个就需要一个递归了。

type 渲染单元链表<
  某单元 extends 棋局单元,
  初始渲染结果 extends string
> = 某单元['下一个'] extends 棋局单元
  ? 渲染单元链表<
      某单元['下一个'],
      `${初始渲染结果} ${渲染单元格<某单元['内容']>}`
    >
  : `${初始渲染结果} ${渲染单元格<某单元['内容']>}`

接下来,我们需要把每行的渲染结果合并输出,由于在 TypeScript 中,不管是字符串还是数组,都无法控制在 IDE 提示中的换行,因此,我们需要把结果渲染成对象结构。又因为通过字符串形态的整数,无法关联到真实的整数类型,所以需要建立一个辅助索引。

type 数字键值对 = {
  : 
  : 
  : 
  : 
  : 
  : 
  : 
  : 
  : 
  : 
}

type 渲染棋局<某个棋局 extends 棋局> = {
  [key in
    | '零'
    | '一'
    | '二'
    | '三'
    | '四'
    | '河'
    | '五'
    | '六'
    | '七'
    | '八'
    | '九']: key extends '河'
    ? '┠ ~ 楚 河 ~ ~ ~ 汉 界 ~ ┨'
    : 渲染指定行<
        某个棋局,
        key extends keyof 数字键值对 ? 数字键值对[key] : 不可能
      >
}

这样,整个棋局就可以被渲染出来了。

本节使用的技巧主要是字符串的拼接,以及对象结构的输出。

在线预览地址:https://tsplay.dev/Nd4n0N 把鼠标放在最后几行的走棋结果上看提示即可。

小结

本文的意图主要是在不考虑优化的情况下,如何借助类型系统的运算来解构这么一个象棋逻辑。

主体功能都还是比较容易实现的,从目前的实现来看,调用栈太深了,现在走棋已经很难跑出来了,需要优化,比如把数字的计算改成打表,可以优化不少,暂时先不去做它了。(最新消息:在 TypeScript 4.5 中,优化了尾递归深度,可以完美支持了:https://devblogs.microsoft.com/typescript/announcing-typescript-4-5-beta/#tailrec-conditional)

特别鸣谢在此文编写过程中提供帮助的引证老师。