Cannot set identityField to custom attribute using @auth

Describe the bug When using @auth directive and “owner” type rule with identityField set to custom:fieldName the generator creates invalid template code

To Reproduce Create a model

type PigRun @model @auth(rules: [{
  allow: owner,
  ownerField: "pigRunClientId",
  identityField: "custom:company"
}]) {
  id: ID!
  client: Company! @connection(name: "CompanyPigRuns")
}

type Company @model {
  id: ID!
  name: String!
}

The generated resolver templates will use invalid $ctx.identity.custom:company syntax in Ownership check sections

Expected behavior I would expect the generators to use $ctx.identity.claims.get("custom:company_id") syntax that have worked for me with ID token before.

Desktop (please complete the following information):

• amplify -v 0.1.28

[mkrn]

Another gotcha for newcomers is to make sure to use idToken:

In your front-end code:

Amplify.configure({
  ...aws_exports,
  graphql_headers: async () => {
    try {
      const token = (await Auth.currentSession()).idToken.jwtToken;
      return { Authorization: token }
    }
    catch (e) {
        console.error(e);
        return {};
        // return { Authorization: await asyncLocalStorage.getItem("jwt") }
    }
  }
});

[github-actions[bot]]

This issue has been automatically locked since there hasn’t been any recent activity after it was closed. Please open a new issue for related bugs.

Looking for a help forum? We recommend joining the Amplify Community Discord server *-help channels for those types of questions.