How to add Kafka offset, partition details in the message’s metadata section using NiFi

Actual message read from Kafka:

JOLT transformation to generate array

I’m working on a JOLT transformation but haven’t been very successful so far. I have this input: