Пишем плагин Parcelize для компилятора Kotlin под iOS

e8151c8cd304bee45b64e7bd8a4d677d.png

В этой статье описан мой опыт по написанию плагина для компилятора Kotlin. Моей главной целью было создание плагина под iOS (Kotlin/Native), аналогичного kotlin-parcelize под Android. Дело в том, что в iOS, как и в Android, приложения тоже могут быть убиты системой, а значит, может возникнуть необходимость сохранять стек навигации и другие данные. В результате работы над этой задачей получился kotlin-parcelize-darwin. Подробности о его создании и применении — под катом.

Parcelize в Android

Хотя в статье будет описана разработка под iOS, давайте вспомним, что собой представляют интерфейс Parcelable и плагин kotlin-parcelize под Android. Интерфейс Parcelable позволяет нам сериализовать в Parcel реализующий класс, чтобы он был представлен в виде массива байтов. Также он позволяет десериализовать класс из Parcel для восстановления всех данных. Эта возможность широко используется для записи и восстановления состояний экрана, например когда приостановленное приложение сначала убивается системой из-за нехватки памяти, а затем снова активируется. 

Реализовать интерфейс Parcelable не сложно. Нужно написать два основных метода:  

  • writeToParcel (Parcel, …) — пишет данные в Parcel,  

  • createFromParcel (Parcel) — читает из Parcel. 

Необходимо записывать информацию поле за полем, а затем считывать в том же порядке. Звучит довольно просто, но писать шаблонный код надоедает. Кроме того, можно наделать ошибок, так что в идеале нужно писать ещё и тесты для классов Parcelable.

К счастью, для компилятора Kotlin есть плагин kotlin-parcelize. Если его включить, то достаточно лишь пометить класс Parcelable с помощью аннотации @Parcelize — и плагин автоматически сгенерирует реализацию. Это избавляет нас от написания соответствующего шаблонного кода и гарантирует корректность реализации на этапе компилирования.

Применение Parcelize в iOS

Поскольку iOS-приложения тоже имеют вышеупомянутые особенности, есть аналогичные способы сохранения состояния приложения. Один из них заключается в использовании протокола NSCoding, который очень похож на интерфейс Parcelable в Android. Классы тоже должны реализовать два метода:  

  • encode (with: NSCoder)— кодирует объект в NSCoder,  

  • init?(coder: NSCoder)— декодирует объект из NSCoder.

Kotlin Native под iOS

Kotlin не ограничен Android — его можно использовать для написания Kotlin Native-фреймворков под iOS и даже мультиплатформенного кода. А поскольку iOS-приложения тоже могут быть убиты системой, то с ними возникают те же проблемы. Kotlin Native под iOS предлагает двунаправленную совместимость с Objective-C, то есть мы можем использовать NSCoding и NSCoder. 

Очень простой класс данных может выглядеть так:

6e9260272e9b4e9ca84f9eb9a0cdb0ce.png

Добавим реализацию протокола NSCoding:

8ca73f5fd44d071dcb92e6d419d4b4ce.png

Выглядит просто. Попробуем скомпилировать:

e6410aa79c0f66c87a86fdf71ce91379.png

Попробуем расширить класс NSObject с помощью класса данных User:

654bfdc419beb459aec2e38e13ca918e.png

И опять не компилируется!

2c9cb1b052fc65d15aeb8bd951f574e5.png

Интересно. Похоже, компилятор пытается переопределить и сгенерировать метод toString, но для классов, наследующихся от NSObject, нам нужно переопределить метод description. Кроме того, нам, вероятно, вообще не стоит использовать наследование, потому что это может помешать пользователям расширять их собственные классы Kotlin (из-за невозможности множественного наследования).

Parcelable для iOS

Нам нужно другое решение, без использования наследования. Определим интерфейс Parcelable:

7575a298bfdc8e9be4330877cf4bd5fa.png

Всё просто. Классы Parcelable будут иметь только метод coding, который возвращает экземпляр NSCodingProtocol. Остальное будет обработано реализацией протокола. 

Теперь давайте изменим класс User таким образом, чтобы он реализовал интерфейс Parcelable:

e54aec59fa90b9c86b3c95c383d12fcf.png

Мы создали вложенный класс CodingImpl, который реализует протокол NSCoding. Метод encodeWithCoder остался неизменным, а вот с initWithCoder ситуация чуть сложнее. Он должен возвращать экземпляр протокола NSCoding, но класс User теперь им не является. Нам нужно какое-то обходное решение, промежуточный класс:

3c3690d799648473eec02853349a3dfc.png

Класс DecodedValue реализует протокол NSCoding и хранит некоторый объект value. Все методы могут быть пустыми, потому что этот класс не будет ни кодироваться, ни декодироваться. Теперь мы можем использовать его в методе initWithCoder класса User:

9bc19928016800e4ec9b1b56f2fe372d.png

Тестирование

Давайте напишем тест, чтобы проверить, всё ли работает правильно. 

  1. Создаём экземпляр класса User с какими-нибудь данными.

  2. Кодируем его с помощью NSKeyedArchiver, в качестве результата получаем NSData.

  3. Декодируем NSData с помощью NSKeyedUnarchiver.

  4. Убеждаемся, что декодированный объект аналогичен исходному.

Пишем плагин для компилятора

Мы определили интерфейс Parcelable под iOS, попробовали его в работе с помощью класса User и протестировали код. Теперь можно автоматизировать реализацию Parcelable, чтобы код генерировался автоматически, как при использовании kotlin-parcelize под Android.

Мы не можем использовать Kotlin Symbol Processing (KSP), потому что это не позволит нам менять существующие классы, а только генерировать новые. Так что единственным решением будет написать плагин для компилятора Kotlin. Но это не так просто, в основном потому, что документации до сих пор нет, API работает нестабильно и т. д. Если вы всё же соберётесь писать плагин для компилятора, то рекомендую обратиться к этим источникам:

Плагин работает так же, как kotlin-parcelize. Классы должны реализовывать интерфейс Parcelable и быть помечены с помощью аннотации @Parcelize. При компилировании плагин генерирует реализации Parcelable. Классы Parcelable выглядят так:

27f2e7f178665cbfa5ae67837c718b5e.png

Название плагина

Плагин называется kotlin-parcelize-darwin. Часть »-darwin» означает, что плагин должен работать со всеми платформами Darwin (Apple), но сейчас нас интересует только iOS.

Gradle-модули

  1. Kotlin-parcelize-darwin — первый модуль, который нам понадобится. Он содержит плагин для Gradle, который регистрирует плагин для компилятора, и ссылается на два артефакта: один — для плагина компилятора Kotlin/Native, второй — для плагина компилятора под все другие платформы.

  2. kotlin-parcelize-darwin-compiler — модуль плагина для компилятора Kotlin/Native.

  3. kotlin-parcelize-darwin-compiler-j — модуль плагина для ненативного компилятора. Он необходим, потому что является обязательным и на него ссылается Gradle-плагин. Хотя на самом деле этот модуль пустой, поскольку нам ничего не нужно из ненативного варианта.

  4. kotlin-parcelize-darwin-runtime — содержит зависимости времени выполнения (runtime) для компиляторного плагина. Например, здесь находятся интерфейс Parcelable и аннотация @Parcelize.

  5. tests — содержит тесты для компиляторного плагина, добавляет в плагин модули в виде included builds.

Процесс установки плагина 

В корневом файле build.gradle:

fca7c9db563b2750a0cb43c2b7130640.png

В файле build.gradle проекта:

7035a9f88ad9fbd8de580e9bf9c4a4cc.png

Реализация

Генерирование кода в Parcelable состоит из двух основных этапов. Нам нужно:

  1. Сделать код компилируемым с помощью добавления синтетических заглушек для отсутствующих методов fun coding (): NSCodingProtocol из интерфейса Parcelable.

  2. Сгенерировать реализации для заглушек, добавленных на предыдущем этапе.

Добавление заглушек

Это делается с помощью класса ParcelizeResolveExtension, который реализует интерфейс SyntheticResolveExtension. Всё очень просто: класс реализует методы getSyntheticFunctionNames и generateSyntheticMethods, которые вызываются при компилировании для каждого класса.

1fbe3e92fa6d353137b4b08c165562eb.png

Как видите, сначала нужно проверить, можно ли применять логику Parcelize к текущему классу. Для этого используется функция isValidForParcelize:

d776ca52161146cee26a62a24af8ec7f.png

Мы обрабатываем только те классы, у которых есть аннотация @Parcelize и которые реализуют интерфейс Parcelable.

Генерирование реализаций заглушек

Как вы могли догадаться, этот этап создания плагина значительно сложнее. За него отвечает класс ParcelizeGenerationExtension, который реализует интерфейс IrGenerationExtension. Он содержит всего один метод:

ee4cd3cfa1b8dcb088c6acdd1f37e9b8.png

Нам необходимо пройтись по всем классам, содержащимся в предоставленном нам  IrModuleFragment. Для этого используется класс ParcelizeClassLoweringPass, который наследует ClassLoweringPass. Класс ParcelizeClassLoweringPass переопределяет только один метод:

9e3e9562c970c7a3da966a78c5d93076.png

Проходить по классам не сложно:

3935b8b46eea4904024cd4c3bf032b16.png

Помимо этих основных, в процессе генерации кода есть ещё несколько этапов. Я не буду описывать все подробности реализации, потому что тогда придётся привести слишком много кода. Вместо этого я в общих чертах расскажу об основных вызовах и покажу, как выглядел бы сгенерированный код, если бы мы писали его вручную. Думаю, в контексте статьи это более полезная информация. Но если вам хочется узнать больше, то подробности реализации вы найдёте здесь.

Итак, сначала нам снова нужно проверить, можно ли применять логику Parcelize к текущему классу (irClass):

5473db2a8a72f4d127b5da0cf6b7ad8e.png

Затем добавим в irClass вложенный класс CodingImpl, определим его супертипы (NSObject и NSCoding) и пометим аннотацией @ExportObjCClass (чтобы класс был доступен при поиске во время выполнения):

4cca4d846b458abc146503b67014b764.png

Теперь добавим в класс CodingImpl первичный конструктор. У него должен быть только один аргумент — data: TheClass, поэтому нам также надо сгенерировать поле (field) data, свойство (property) и метод считывания (getter).

eb25f306968cf4eb399fb4501501b647.png

Вот что у нас получается:

b511e837e84145efcd538bd330084410.png

Добавим реализацию протокола NSCoding:

5df905538763364e0019b9e547afafd3.png

Теперь сгенерированный класс выглядит так:

6df58d61d5d68053b063a36019076f61.png

Наконец нам нужно сгенерировать тело метода coding (), просто создав экземпляр класса CodingImpl:

b3fdf6b9d70ec8e827d5037fcf057b32.png

Сгенерированный код:

aaffb2869a1e0b3396d99a6952c65ca2.png

Использование плагина

Плагин задействуется, когда мы пишем на Kotlin классы Parcelable. Обычно его используют для сохранения состояний экрана. Плагин позволяет восстанавливать исходное состояние приложения после того, как оно было убито iOS. Другой сценарий использования — сохранение стека навигации в тех случаях, когда она реализована на Kotlin.

Вот обобщённый пример использования Parcelable в Kotlin, который демонстрирует, как можно сохранить и восстановить данные:

6589458248a51023d93b58a5b73a477c.png

А вот пример того, как мы можем кодировать и декодировать классы Parcelable в iOS-приложении:

4c59d60045520126cc7a09e47e48d58c.png

Parcelize в Kotlin Multiplatform

Теперь у нас есть два плагина: kotlin-parcelize для Android и kotlin-parcelize-darwin — для iOS. И мы можем применить их оба и использовать @Parcelize в общем коде!

Файл build.gradle нашего общего модуля будет выглядеть так:

3b58770445b7371c25dbf42c22d7c620.png

Теперь у нас в наборах androidMain и iosMain есть доступ к интерфейсам Parcelable и аннотациям @Parcelize. Чтобы использовать их в commonMain, нам нужно вручную определить их с помощью expect/actual.

Напишем в commonMain:

5f4ee65cd5b5b1c13aa842f4ab3d1999.png

В iosMain:

21c89bf7a8c9db95c556aef4180f8894.png

В androidMain:

e8a809017b3eef3210f190cf45880bcf.png

Во всех остальных наборах:

c8731be2788f6e916eae0fad891bac1f.png

Теперь мы можем использовать Parcelize как обычно в commonMain. При сборке под Android код будет обработан плагином kotlin-parcelize, а при сборке под iOS — плагином kotlin-parcelize-darwin. В случае с другими платформами ничего не будет сделано, потому что интерфейс Parcelable будет пуст, а аннотация будет отсутствовать.

Заключение

Мы рассмотрели компиляторный плагин kotlin-parcelize-darwin. Исследовали его структуру и принцип работы, узнали, как его можно применять в Kotlin Native, как подружить его с Android-плагином kotlin-parcelize в Kotlin Multiplatform, а также как использовать Parcelable на стороне iOS.

Исходный код лежит на GitHub. Плагин ещё не опубликован, но вы уже можете с ним экспериментировать, опубликовав в локальном Maven-репозитории или с помощью Gradle Composite builds.

В репозитории лежит очень простой пример проекта, в котором есть общий модуль, а также Android- и iOS-приложения. Спасибо, что дочитали, и не забудьте подписаться на меня в Twitter!

© Habrahabr.ru