How to get a reference of a type from my element attribute to create an import statement?

[atreeon]

I am generating code from a function

@mycodegen
List<Cat> getCats() {
  return cats;
}

I need now to get an import statement when I reproduce the call method when generating code in my library builder, otherwise a reference to Cat will be missing. I can do this in the generator:

callMethod.type.returnType.element.location

which produces something like

asset:example/test/Cat.dart;asset:example/test/Cat.dart;Cat

I would then have to do some string manipulation to get the bits I want to create an import statment. Is this best practice or is there an easier way?

[jakemac53]

You can get an AssetId through the assetIdForElement api which would give you a bit more directly what you need.

That has a uri getter, but it will only give an "importable" uri if it can be mapped to a package: uri. If it comes from something outside of a lib directory you would have to construct a relative path to it using the path property of the asset id.