fix custom option parser

Author: switchupcb

README.md

@@ -34,9 +34,9 @@ Each example has a **README**.
 | [tmpl](examples/tmpl/)           | Uses `.tmpl` templates.                  |
 | [program](examples/program/)     | Uses Copygen programmatically.           |
 
-This [example](examples/main/) uses three type-structs to generate the `ModelsToDomain()` function using a Command Line Interface.
+_*[`multi`](examples/_tests/multi/setup/setup.go) tests every type._
 
-_[`multi`](examples/_tests/multi/setup/setup.go) tests every type._
+This [example](examples/main/) uses three type-structs to generate the `ModelsToDomain()` function using a Command Line Interface.
 
 ### Types
 
@@ -131,7 +131,7 @@ You can specify options for your Copygen functions using comments: Do **NOT** pu
 | `depth field level` | Use a specific field depth.                              | Copygen uses full-field [depth](#depth) by default. <br /> Override this using `depth` with _regex_ and a [depth-level](#depth) integer.                                           | `depth .* 2` <br /> `depth models.Account.* 1`                               |
 | `deepcopy field`    | Deepcopy from-fields.                                    | Copygen shallow copies fields by default. <br /> Override this using `deepcopy` with _regex_. <br /> For more info, view [Shallow Copy vs. Deep Copy](#shallow-copy-vs-deep-copy). | `deepcopy package.Type.Field` <br /> `deepcopy .*` _(all fields)_            |
 | `automatch field`   | Use the automatcher selectively or with `map` and `tag`. | Using `map` or `tag` disables the default automatcher. <br /> Enable it using `automatch` with _regex_. <br />                                                                     | `automatch package.Type.Field` <br /> `automatch models.User.*`              |
-| `custom option`     | Specify custom options.                                  | Use custom options with [templates](#templates). <br /> `custom` options are **function** options. <br /> Returns `map[string][]string` _(trim-spaced)_.                           | `ignore true` <br /> `swap false`                                            |
+| `custom option`     | Specify custom function options.                         | Use custom options with [templates](#templates). <br /> Returns `map[string][]string` _(trim-spaced)_.                                                                             | `ignore true` <br /> `swap false`                                            |
 
 _[View a reference on Regex.](https://cheatography.com/davechild/cheat-sheets/regular-expressions/)_
 
@@ -162,7 +162,7 @@ go install github.com/switchupcb/copygen@main
 
 Install a specific version by specifying a tag.
 ```
-go install github.com/switchupcb/copygen@v0.3.6
+go install github.com/switchupcb/copygen@v0.3.7
 ```
 
 Run the executable with given options.

cli/parser/function.go

@@ -56,7 +56,7 @@ func (p *Parser) parseFunctions(astcopygen *ast.InterfaceType) ([]models.Functio
 		// map the function custom options.
 		var customoptionmap map[string][]string
 		for _, option := range fieldoptions {
-			err := options.MapCustomOption(customoptionmap, option)
+			customoptionmap, err = options.MapCustomOption(customoptionmap, option)
 			if err != nil {
 				fmt.Printf("WARNING: %v\n", err)
 			}

cli/parser/keep.go

@@ -60,6 +60,8 @@ func (p *Parser) assignFieldOption(comments []*ast.Comment) error {
 
 	for _, comment := range comments {
 		text := comment.Text
+
+		// do NOT parse comments that have already been parsed.
 		if p.Options.CommentOptionMap[text] != nil {
 			continue
 		}

cli/parser/options/custom.go

@@ -2,32 +2,15 @@ package options
 
 import (
 	"fmt"
-	"strings"
 )
 
 const (
 	CategoryCustom = "custom"
 	FormatCustom   = "<option><whitespaces><value>"
 )
 
-// ParseCustom parses a custom option.
-func ParseCustom(option string) (*Option, error) {
-	splitoption := strings.Fields(option)
-
-	if len(splitoption) == 0 {
-		return nil, fmt.Errorf("there is an unspecified %s option at an unknown line", CategoryCustom)
-	} else if len(splitoption) == 1 {
-		return nil, fmt.Errorf("there is a misconfigured %s option: %q.\nIs it in format %s?", CategoryCustom, option, FormatCustom)
-	}
-
-	return &Option{
-		Category: splitoption[0],
-		Value:    splitoption[:1],
-	}, nil
-}
-
 // MapCustomOption maps a custom option in an optionmap[category][]values.
-func MapCustomOption(optionmap map[string][]string, option *Option) error {
+func MapCustomOption(optionmap map[string][]string, option *Option) (map[string][]string, error) {
 	if optionmap == nil {
 		optionmap = make(map[string][]string)
 	}
@@ -38,18 +21,22 @@ func MapCustomOption(optionmap map[string][]string, option *Option) error {
 				optionmap[k] = append(optionmap[k], v)
 			}
 		} else {
-			return fmt.Errorf("failed to map custom option: %v", option.Value)
+			return optionmap, fmt.Errorf("failed to map custom option: %v", option.Value)
 		}
 	}
 
-	return nil
+	return optionmap, nil
 }
 
 // MapCustomOptions maps options with custom categories in a list of options to a customoptionmap[category][]value.
 func MapCustomOptions(options []*Option) (map[string][]string, error) {
-	var optionmap map[string][]string
+	var (
+		optionmap map[string][]string
+		err       error
+	)
+
 	for _, option := range options {
-		err := MapCustomOption(optionmap, option)
+		optionmap, err = MapCustomOption(optionmap, option)
 		if err != nil {
 			return nil, err
 		}

cli/parser/options/options.go

@@ -78,6 +78,8 @@ func SetFieldOptions(field *models.Field, fieldoptions []*Option) {
 		case CategoryDeepcopy:
 			SetDeepcopy(field, *option)
 
+		case CategoryCustom:
+			SetConvert(field, *option)
 		}
 	}
 }