【Swift】Extensionの定義の仕方について


アプリのような比較的大きなプロジェクトにおけるExtensionの定義の仕方について考えてみました。
まだ実践したわけではないのですが、一旦まとまったので共有してみます。

## 定義の仕方

* **一つのファイルで拡張する既存型は一つにする。**
* **一つのファイル内でExtensionを複数定義してもよい。**
* **追加される型・メソッドで関連するものは、共通のExtensionのスコープ内でグルーピングする。**
* **ファイル名は <拡張する既存型名>+<分類名>.swift にする。**
* **プロジェクトに<拡張する既存型名>グループを作成してファイルをまとめる。**
	* **プロジェクト外で定義された既存型に関するグループはわかりやすい場所にまとめる。**
	* **プロジェクト内で定義された既存型に関するグループは、その型が定義されたファイルもその下に置く。**


## 例

プロジェクトの構成例。

![extension_files.png](https://qiita-image-store.s3.amazonaws.com/0/38775/4f369c9c-587b-1e4e-e8c9-17777193738f.png "extension_files.png")



Extensionファイル例。特に依存関係にあるメソッドは同じスコープに入れると良さそうです。

```swift:String+ConsecutiveCharacters.swift
extension String {
    
    // repeatCountの数だけ文字列中の"!"を連続させた文字列を返すメソッド
    // ex) "Hello!".addConsecutiveExclamation(5) -> "Hello!!!!!"
    func addConsecutiveExclamation(repeatCount: UInt) -> String {
        return addConsecutiveCharacters("!" as Character, repeatCount: repeatCount)
    }
    
    // 引数cの文字をrepeatCountだけ連続させた文字列を返すメソッド
    func addConsecutiveCharacters(c: Character, repeatCount: UInt) -> String {
        return self.characters.reduce("") { str, char in
            var newStr = str
            let rc = (c == char) ? repeatCount : 1
            for _ in 0..<rc {
                newStr.append(char)
            }
            return newStr
        }
    }
    
}

extension String {
    
    // 連続するスペースを1つにまとめた文字列を返すメソッド
    // ex) "Hello     World".removeConsecutiveSpace() -> "Hello World"
    func removeConsecutiveSpace() -> String {
        return removeConsecutiveCharacters(" " as Character)
    }
    
    // 連続する引数の文字を1つにまとめた文字列を返すメソッド
    func removeConsecutiveCharacters(c: Character) -> String {
        return self.characters.reduce("") { str, char in
            if let last = str.characters.last where char == c && last == char {
                return str
            } else {
                var newStr = str
                newStr.append(char)
                return newStr
            }
        }
    }
    
}
```

## そもそもExtensionについて考え出した背景

以前Alamofireのソースコードを読んでいたときに、Extensionを定義する方法が気になりました。

例えば[Alamofire/Alamofire](https://github.com/Alamofire/Alamofire#response-json-handler)に`responseJSON`を呼ぶ例が記載されています。

```
Alamofire.request(.GET, "https://httpbin.org/get")
         .responseJSON { response in
             debugPrint(response)
         }
```

「この`responseJSON`の実装はどうなってるのかな？」と思い、`request`メソッドの戻り値の型が`Request`なので Request.swift を見るのですが、`responseJSON`は見当たりません。「あ、Extensionなのか。多分 Response.swift あたりに定義されてそうだな。」と思って覗いてみましたが、ここにもありません。

正解は ResponseSerialization.swift でした。

このとき気づいたのですが、よく見ると`Request`型のExtensionはAlamofireの至るところに存在しており、実際ビルドされたときに`Request`がどうなっているのかよくわかりません。このようなExtensionの定義方法は、Alamofireのような小さいライブラリなら良いものの（むしろ巧みにextensionを使っていて勉強になります）、大きなアプリを作るときは困りそうだと感じました。

## Alamofire的なExtension定義の特徴

* 追加される型・メソッドで関連するものは、一つのExtensionのスコープ内で定義される。
* Extensionは分類ごとにファイル分けされている。
* ファイル名は <分類名>.swift となっており、このファイルによってどの既存型が拡張されるかは中身を見ないと分からない。
* 一つのファイルに複数の既存型に対するExtensionが定義されることもある。

#### 良いと思った点

* 追加される型・メソッドがExtensionのスコープによってグルーピングされているため、関連がわかりやすい。

#### 困りそうな点

* 目的の定義がどこにあるか分かりにくく、ソースコードが追いにくい。特にGitHub/GitLabなどにおいてWebブラウザでコードレビューしたりするときに困る。
* チーム開発の際は何かルールを決めないと、いろんなファイルにExtensionができてカオスになりそう。

## Extensionを管理しやすくするために

上記内容を踏まえつつ、管理しやすいExtensionにするために次のようなことを考えました。

* どの既存型に対するExtensionかはファイル名で判別できるようにしたい。
* 同じ既存型に対するファイルは同一箇所にまとめるようにしたい。

これらを踏まえたのが冒頭で述べた定義の仕方になります。

## まとめ

アプリプロジェクトのような比較的大きなプロジェクトにおけるExtensionの定義の仕方について書きました。
結果的にObjective-Cのときと同じような管理になりましたが、これが良さそうに思います。


もし他に良い案があればコメントいただけるとうれしいです。


## 参考

[Swift で Extension につけるファイル名のベストプラクティス | FIVETEESIXONE](http://fiveteesixone.lackland.io/2015/01/08/best-practice-for-naming-swift-extension-file-name/)