They also set up a system so that non-EU retail sellers can collect VAT directly on payment

That’s what Australia does too. Since the sellers already had their systems set up to handle it for Australia, it was probably easy for them to extend it to be used for the EU too.