NixOS azure-agent outdated for Azure marketplace AIRA offer

[khssnv]

While we used AIRA for in-house development we could integrate own private key for first login. Now we want to make an offer in Azure Marketplace and there is a lack of integration tools, especially for first login & configuration. We can currently make an image of AIRA with password-based root login (or integrate a common key with publishing a private pair, or make a web service that waits for a new key or something else), but it's not safe and not a common way for MS Azure users. There is a package azure-agent in nixpkgs, but it's outdated and don't work. Could someone help to update it?

[khssnv]

There is a PR with update. Author told he need help with tests.