cosmos-sdk/depinject/internal/codegen/ident.go
Aaron Craelius 9bc59e6af8
feat(depinject): codegen part 1 (imports & idents) (#12557)
## Description

Ref #12556 

To make things easier I've starting splitting up #12469 into smaller PRs.

This first PR for codegen adds basic infrastructure for:
* adding package imports
* creating new var names (`*ast.Ident`) without conflicts

Other PRs will add the actual codegen functionality.



---

### Author Checklist

*All items are required. Please add a note to the item if the item is not applicable and
please add links to any relevant follow up issues.*

I have...

- [ ] included the correct [type prefix](https://github.com/commitizen/conventional-commit-types/blob/v3.0.0/index.json) in the PR title
- [ ] added `!` to the type prefix if API or client breaking change
- [ ] targeted the correct branch (see [PR Targeting](https://github.com/cosmos/cosmos-sdk/blob/main/CONTRIBUTING.md#pr-targeting))
- [ ] provided a link to the relevant issue or specification
- [ ] followed the guidelines for [building modules](https://github.com/cosmos/cosmos-sdk/blob/main/docs/building-modules)
- [ ] included the necessary unit and integration [tests](https://github.com/cosmos/cosmos-sdk/blob/main/CONTRIBUTING.md#testing)
- [ ] added a changelog entry to `CHANGELOG.md`
- [ ] included comments for [documenting Go code](https://blog.golang.org/godoc)
- [ ] updated the relevant documentation or specification
- [ ] reviewed "Files changed" and left comments if necessary
- [ ] confirmed all CI checks have passed

### Reviewers Checklist

*All items are required. Please add a note if the item is not applicable and please add
your handle next to the items reviewed if you only reviewed selected items.*

I have...

- [ ] confirmed the correct [type prefix](https://github.com/commitizen/conventional-commit-types/blob/v3.0.0/index.json) in the PR title
- [ ] confirmed `!` in the type prefix if API or client breaking change
- [ ] confirmed all author checklist items have been addressed 
- [ ] reviewed state machine logic
- [ ] reviewed API design and naming
- [ ] reviewed documentation is accurate
- [ ] reviewed tests and test coverage
- [ ] manually tested (if applicable)
2022-07-15 11:16:35 +00:00

30 lines
628 B
Go

package codegen
import (
"fmt"
"go/ast"
)
// CreateIdent creates a new ident that doesn't conflict with reserved symbols,
// top-level declarations and other defined idents. Idents are unique across
// the whole file as it is assumed that codegen usually happens on one function
// per file.
func (g *FileGen) CreateIdent(namePrefix string) *ast.Ident {
return ast.NewIdent(g.doCreateIdent(namePrefix))
}
func (g *FileGen) doCreateIdent(namePrefix string) string {
v := namePrefix
i := 2
for {
_, ok := g.idents[v]
if !ok {
g.idents[v] = true
return v
}
v = fmt.Sprintf("%s%d", namePrefix, i)
i++
}
}