引数の型によって特定の型の返り値（戻り値）を返す関数（入力された型で出力される型が決まる関数）の作り方

Question

型ArgAか型ArgBを受け取って、受け取った型に応じてRetAかRetBの型を返す関数を例とします。

type ArgA = [string, number, boolean]
type ArgB = [number, number]
type RetA = {abc: number}
type RetB = {vw: string, x: {y: number, z: number}}

function sample<T extends ArgA | ArgB>(arg: T): RetA | RetB{
  if (typeof arg[0] == 'string') { //ArgA
    return {abc: 1} //RetA
  }
  return {vw: 'foo', x: {y: 222, z: 333}} //RetB
}

let foo = sample(['a', 2, false]) //RetA に違いないが RetA | RetB
let bar = sample([5, 10]) //RetB に違いないが RetA | RetB

戻り値の型は RetA | RetB ですが、明らかに入力時点で出力される型の予想はつくので、TypeScriptに出力まで推測してもらおうと思い、関数宣言部分に戻り値の型 Result<T> を指定しました。

type Result<T extends ArgA | ArgB> = T extends ArgA ? RetA : RetB
function sample<T extends ArgA | ArgB>(arg: T): Result<T> {
  if (typeof arg[0] == 'string') {
    return {abc: 1} //甲
  }
  return {vw: 'foo', x: {y: 222, z: 333}} //乙
}

そうするとreturn の部分（甲、乙）で「型'‥' を 'Result<T>' に割り当てることができません。(ts2322) 」とvscodeのLSPがエラーを表示しました。

以下のようにas型アノテーションをつけることでエラーを抑えました。

function sample<T extends ArgA | ArgB>(arg: T): Result<T> {
  if (typeof arg[0] == 'string') {
    return {abc: 1} as Result<T>
  }
  return {vw: 'foo', x: {y: 222, z: 333}} as Result<T>
}

型アノテーションを使うのは釈然としませんが、現状このような解決方法しかないのでしょうか？
もっといい方法があればご教授ください。

---

（質問2）
また、関数の型をオーバーロードを使って定義するように書き換えようとしたところ、うまくいきませんでした。
＜＊追記：オーバーロードは使わないことにしました。＞

type ArgA = [string, number, boolean]
type ArgB = [number, number]
type RetA = {abc: number}
type RetB = {vw: string, x: {y: number, z: number}}
type Result<T extends ArgA | ArgB> = T extends ArgA ? RetA : RetB
type Sample = {
  (arg: ArgA): RetA
  (arg: ArgB): RetB
}
const sample2: Sample = <T extends ArgA | ArgB>(arg: T)=>{ //<-Error
  if (typeof arg[0] == 'string') {
    return {abc: 1} as Result<T>
  }
  return {vw: 'foo', x: {y: 222, z: 333}} as Result<T>
}

vscodeのLSPエラー内容

型 '<T extends ArgA | ArgB>(arg: T) => Result<T>' を型 'Sample' に割り当てることはできません。
  型 'RetA | RetB' を型 'RetA' に割り当てることはできません。
    プロパティ 'abc' は型 'RetB' にありませんが、型 'RetA' では必須です。ts(2322)

これはどうしてでしょうか。オーバーロードを使って、「型に応じて返す型が決まる関数」の定義方法があるのなら教えていただけませんか？ オーバーロードは仕様を表現するだけで型を仕分けするのはコーダーの仕事のようです。

Answer 1

オーバーロードを使って

自分の印象論かもしれないですが、TypeScriptはコンパイルすると型が消えるという特徴もあるので、オーバーロードはJavaScriptの関数の挙動を表現するために使うもの、という感覚です。

（オーバーロードしても内部では条件分岐して泥臭くなってしまうので、自分でTypeScriptを使って関数を書くなら、無理に1つの関数にまとめずに、引数のパターンごとに関数を分けたほうがやりやすいのではないかと考えます）

Comments

[noc]

ありがとうございます。オーバーロードは仕様を表現するだけで、型の仕分けはコーダーが手動で行わないといけないということですね。ユニオンされた型が作られるだけであんまり役立つ気がしないので、結局オーバーロードは使わないことにしました。
今のところジェネリックで捕捉した引数を使って、戻り値に型アノテーションをつける解決法に落ち着きそうです。