forked from coop-cloud-mirrors/godotenv
		
	Compare commits
	
		
			1 Commits
		
	
	
		
			v1.5.0
			...
			release-ac
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
| 41e88df24e | 
							
								
								
									
										10
									
								
								.github/dependabot.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										10
									
								
								.github/dependabot.yml
									
									
									
									
										vendored
									
									
								
							| @ -1,10 +0,0 @@ | |||||||
| version: 2 |  | ||||||
| updates: |  | ||||||
|   - package-ecosystem: "gomod" |  | ||||||
|     directory: / |  | ||||||
|     schedule: |  | ||||||
|       interval: "daily" |  | ||||||
|   - package-ecosystem: "github-actions" |  | ||||||
|     directory: "/" |  | ||||||
|     schedule: |  | ||||||
|       interval: "daily" |  | ||||||
							
								
								
									
										48
									
								
								.github/workflows/ci.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										48
									
								
								.github/workflows/ci.yml
									
									
									
									
										vendored
									
									
								
							| @ -8,13 +8,55 @@ jobs: | |||||||
|     strategy: |     strategy: | ||||||
|       fail-fast: false |       fail-fast: false | ||||||
|       matrix: |       matrix: | ||||||
|         go: [ '1.20', '1.19', '1.18', '1.17', '1.16' ] |         go: [ '1.15', '1.14' ] | ||||||
|         os: [ ubuntu-latest, macOS-latest, windows-latest ] |         os: [ ubuntu-latest, macOS-latest, windows-latest ] | ||||||
|     name: ${{ matrix.os }} Go ${{ matrix.go }} Tests |     name: ${{ matrix.os }} Go ${{ matrix.go }} Tests | ||||||
|     steps: |     steps: | ||||||
|       - uses: actions/checkout@v3 |       - uses: actions/checkout@v2 | ||||||
|       - name: Setup go |       - name: Setup go | ||||||
|         uses: actions/setup-go@v3 |         uses: actions/setup-go@v2 | ||||||
|         with: |         with: | ||||||
|           go-version: ${{ matrix.go }} |           go-version: ${{ matrix.go }} | ||||||
|       - run: go test |       - run: go test | ||||||
|  |  | ||||||
|  |   test-non-amd64: | ||||||
|  |     strategy: | ||||||
|  |       fail-fast: false | ||||||
|  |       matrix: | ||||||
|  |         arch: | ||||||
|  |           # For some reasons this is segfaulting on go env | ||||||
|  |           # - name: IBM Z and LinuxONE | ||||||
|  |           #   architecture: "s390x" | ||||||
|  |           - name: POWER8 | ||||||
|  |             architecture: "ppc64le" | ||||||
|  |      | ||||||
|  |     runs-on: ubuntu-latest | ||||||
|  |     name: Test on ${{ matrix.arch.name }}  | ||||||
|  |     steps: | ||||||
|  |       - uses: actions/checkout@v2 | ||||||
|  |       - uses: uraimo/run-on-arch-action@master | ||||||
|  |         with: | ||||||
|  |           arch: ${{ matrix.arch.architecture }} | ||||||
|  |           distro: ubuntu20.04 | ||||||
|  |           env: | # YAML pipe | ||||||
|  |             GOARCH: ${{ matrix.arch.architecture }} | ||||||
|  |             CGO_ENABLED: 0 | ||||||
|  |           run: | | ||||||
|  |             apt-get update | ||||||
|  |             apt-get install -q -y curl wget git | ||||||
|  |             latestGo=$(curl "https://golang.org/VERSION?m=text") | ||||||
|  |             wget "https://dl.google.com/go/${latestGo}.linux-${GOARCH}.tar.gz" | ||||||
|  |             rm -f $(which go) | ||||||
|  |             rm -rf /usr/local/go | ||||||
|  |             tar -C /usr/local -xzf "${latestGo}.linux-${GOARCH}.tar.gz" | ||||||
|  |             export PATH=/usr/local/go/bin:$PATH | ||||||
|  |             printf "Using go at: $(which go)\n" | ||||||
|  |             printf "Go version: $(go version)\n" | ||||||
|  |             printf "\n\nGo environment:\n\n" | ||||||
|  |             go env | ||||||
|  |             printf "\n\nSystem environment:\n\n" | ||||||
|  |             env | ||||||
|  |             go get -insecure -v -t -d ./... | ||||||
|  |             go test ./... | ||||||
|  |             cd ./cmd/godotenv | ||||||
|  |             go build -trimpath -ldflags="-w -s" -v | ||||||
							
								
								
									
										72
									
								
								.github/workflows/codeql-analysis.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										72
									
								
								.github/workflows/codeql-analysis.yml
									
									
									
									
										vendored
									
									
								
							| @ -1,72 +0,0 @@ | |||||||
| # For most projects, this workflow file will not need changing; you simply need |  | ||||||
| # to commit it to your repository. |  | ||||||
| # |  | ||||||
| # You may wish to alter this file to override the set of languages analyzed, |  | ||||||
| # or to provide custom queries or build logic. |  | ||||||
| # |  | ||||||
| # ******** NOTE ******** |  | ||||||
| # We have attempted to detect the languages in your repository. Please check |  | ||||||
| # the `language` matrix defined below to confirm you have the correct set of |  | ||||||
| # supported CodeQL languages. |  | ||||||
| # |  | ||||||
| name: "CodeQL" |  | ||||||
|  |  | ||||||
| on: |  | ||||||
|   push: |  | ||||||
|     branches: [ "main" ] |  | ||||||
|   pull_request: |  | ||||||
|     # The branches below must be a subset of the branches above |  | ||||||
|     branches: [ "main" ] |  | ||||||
|   schedule: |  | ||||||
|     - cron: '31 4 * * 2' |  | ||||||
|  |  | ||||||
| jobs: |  | ||||||
|   analyze: |  | ||||||
|     name: Analyze |  | ||||||
|     runs-on: ubuntu-latest |  | ||||||
|     permissions: |  | ||||||
|       actions: read |  | ||||||
|       contents: read |  | ||||||
|       security-events: write |  | ||||||
|  |  | ||||||
|     strategy: |  | ||||||
|       fail-fast: false |  | ||||||
|       matrix: |  | ||||||
|         language: [ 'go' ] |  | ||||||
|         # CodeQL supports [ 'cpp', 'csharp', 'go', 'java', 'javascript', 'python', 'ruby' ] |  | ||||||
|         # Learn more about CodeQL language support at https://aka.ms/codeql-docs/language-support |  | ||||||
|  |  | ||||||
|     steps: |  | ||||||
|     - name: Checkout repository |  | ||||||
|       uses: actions/checkout@v3 |  | ||||||
|  |  | ||||||
|     # Initializes the CodeQL tools for scanning. |  | ||||||
|     - name: Initialize CodeQL |  | ||||||
|       uses: github/codeql-action/init@v2 |  | ||||||
|       with: |  | ||||||
|         languages: ${{ matrix.language }} |  | ||||||
|         # If you wish to specify custom queries, you can do so here or in a config file. |  | ||||||
|         # By default, queries listed here will override any specified in a config file. |  | ||||||
|         # Prefix the list here with "+" to use these queries and those in the config file. |  | ||||||
|          |  | ||||||
|         # Details on CodeQL's query packs refer to : https://docs.github.com/en/code-security/code-scanning/automatically-scanning-your-code-for-vulnerabilities-and-errors/configuring-code-scanning#using-queries-in-ql-packs |  | ||||||
|         # queries: security-extended,security-and-quality |  | ||||||
|  |  | ||||||
|          |  | ||||||
|     # Autobuild attempts to build any compiled languages  (C/C++, C#, or Java). |  | ||||||
|     # If this step fails, then you should remove it and run the build manually (see below) |  | ||||||
|     - name: Autobuild |  | ||||||
|       uses: github/codeql-action/autobuild@v2 |  | ||||||
|  |  | ||||||
|     # ℹ️ Command-line programs to run using the OS shell. |  | ||||||
|     # 📚 See https://docs.github.com/en/actions/using-workflows/workflow-syntax-for-github-actions#jobsjob_idstepsrun |  | ||||||
|  |  | ||||||
|     #   If the Autobuild fails above, remove it and uncomment the following three lines.  |  | ||||||
|     #   modify them (or add more) to build your code if your project, please refer to the EXAMPLE below for guidance. |  | ||||||
|  |  | ||||||
|     # - run: | |  | ||||||
|     #   echo "Run, Build Application using script" |  | ||||||
|     #   ./location_of_script_within_repo/buildscript.sh |  | ||||||
|  |  | ||||||
|     - name: Perform CodeQL Analysis |  | ||||||
|       uses: github/codeql-action/analyze@v2 |  | ||||||
							
								
								
									
										27
									
								
								.github/workflows/release.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										27
									
								
								.github/workflows/release.yml
									
									
									
									
										vendored
									
									
								
							| @ -11,21 +11,32 @@ jobs: | |||||||
|     runs-on: ubuntu-latest |     runs-on: ubuntu-latest | ||||||
|     steps: |     steps: | ||||||
|       - name: Checkout code |       - name: Checkout code | ||||||
|         uses: actions/checkout@v3 |         uses: actions/checkout@v2 | ||||||
|       - name: Generate build files |       - name: Generate build files | ||||||
|         uses: thatisuday/go-cross-build@v1.0.2 |         uses: thatisuday/go-cross-build@v1 | ||||||
|         with: |         with: | ||||||
|             platforms: 'linux/amd64, linux/ppc64le, darwin/amd64, darwin/arm64, windows/amd64' |             platforms: 'linux/amd64, linux/ppc64le, darwin/amd64, windows/amd64' | ||||||
|             package: 'cmd/godotenv' |             package: 'cmd/godotenv' | ||||||
|             name: 'godotenv' |             name: 'godotenv' | ||||||
|             compress: 'true' |             compress: 'true' | ||||||
|             dest: 'dist' |             dest: 'dist' | ||||||
|       - name: Publish Binaries |       - name: Create Release | ||||||
|         uses: svenstaro/upload-release-action@v2 |         id: create_release | ||||||
|  |         uses: actions/create-release@v1 | ||||||
|  |         env: | ||||||
|  |           GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} | ||||||
|         with: |         with: | ||||||
|           repo_token: ${{ secrets.GITHUB_TOKEN }} |           tag_name: ${{ github.ref }} | ||||||
|           release_name: Release ${{ github.ref }} |           release_name: Release ${{ github.ref }} | ||||||
|           tag: ${{ github.ref }} |           draft: false | ||||||
|  |           prerelease: ${{ contains(github.ref, 'pre') }} | ||||||
|  |       - name: Upload Release Assets | ||||||
|  |         id: upload-release-asset  | ||||||
|  |         uses: actions/upload-release-asset@v1 | ||||||
|  |         env: | ||||||
|  |           GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} | ||||||
|  |         with: | ||||||
|  |           upload_url: ${{ steps.create_release.outputs.upload_url }} | ||||||
|           file: dist/* |           file: dist/* | ||||||
|           file_glob: true |  | ||||||
|           overwrite: true |           overwrite: true | ||||||
|  |           file_glob: true | ||||||
							
								
								
									
										38
									
								
								README.md
									
									
									
									
									
								
							
							
						
						
									
										38
									
								
								README.md
									
									
									
									
									
								
							| @ -1,6 +1,6 @@ | |||||||
| # GoDotEnv  [](https://goreportcard.com/report/github.com/joho/godotenv) | # GoDotEnv  [](https://goreportcard.com/report/github.com/joho/godotenv) | ||||||
|  |  | ||||||
| A Go (golang) port of the Ruby [dotenv](https://github.com/bkeepers/dotenv) project (which loads env vars from a .env file). | A Go (golang) port of the Ruby dotenv project (which loads env vars from a .env file) | ||||||
|  |  | ||||||
| From the original Library: | From the original Library: | ||||||
|  |  | ||||||
| @ -8,9 +8,9 @@ From the original Library: | |||||||
| > | > | ||||||
| > But it is not always practical to set environment variables on development machines or continuous integration servers where multiple projects are run. Dotenv load variables from a .env file into ENV when the environment is bootstrapped. | > But it is not always practical to set environment variables on development machines or continuous integration servers where multiple projects are run. Dotenv load variables from a .env file into ENV when the environment is bootstrapped. | ||||||
|  |  | ||||||
| It can be used as a library (for loading in env for your own daemons etc.) or as a bin command. | It can be used as a library (for loading in env for your own daemons etc) or as a bin command. | ||||||
|  |  | ||||||
| There is test coverage and CI for both linuxish and Windows environments, but I make no guarantees about the bin version working on Windows. | There is test coverage and CI for both linuxish and windows environments, but I make no guarantees about the bin version working on windows. | ||||||
|  |  | ||||||
| ## Installation | ## Installation | ||||||
|  |  | ||||||
| @ -21,13 +21,6 @@ go get github.com/joho/godotenv | |||||||
| ``` | ``` | ||||||
|  |  | ||||||
| or if you want to use it as a bin command | or if you want to use it as a bin command | ||||||
|  |  | ||||||
| go >= 1.17 |  | ||||||
| ```shell |  | ||||||
| go install github.com/joho/godotenv/cmd/godotenv@latest |  | ||||||
| ``` |  | ||||||
|  |  | ||||||
| go < 1.17 |  | ||||||
| ```shell | ```shell | ||||||
| go get github.com/joho/godotenv/cmd/godotenv | go get github.com/joho/godotenv/cmd/godotenv | ||||||
| ``` | ``` | ||||||
| @ -47,10 +40,9 @@ Then in your Go app you can do something like | |||||||
| package main | package main | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|  |     "github.com/joho/godotenv" | ||||||
|     "log" |     "log" | ||||||
|     "os" |     "os" | ||||||
|  |  | ||||||
|     "github.com/joho/godotenv" |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| func main() { | func main() { | ||||||
| @ -75,8 +67,8 @@ import _ "github.com/joho/godotenv/autoload" | |||||||
| While `.env` in the project root is the default, you don't have to be constrained, both examples below are 100% legit | While `.env` in the project root is the default, you don't have to be constrained, both examples below are 100% legit | ||||||
|  |  | ||||||
| ```go | ```go | ||||||
| godotenv.Load("somerandomfile") | _ = godotenv.Load("somerandomfile") | ||||||
| godotenv.Load("filenumberone.env", "filenumbertwo.env") | _ = godotenv.Load("filenumberone.env", "filenumbertwo.env") | ||||||
| ``` | ``` | ||||||
|  |  | ||||||
| If you want to be really fancy with your env file you can do comments and exports (below is a valid env file) | If you want to be really fancy with your env file you can do comments and exports (below is a valid env file) | ||||||
| @ -153,8 +145,6 @@ godotenv -f /some/path/to/.env some_command with some args | |||||||
|  |  | ||||||
| If you don't specify `-f` it will fall back on the default of loading `.env` in `PWD` | If you don't specify `-f` it will fall back on the default of loading `.env` in `PWD` | ||||||
|  |  | ||||||
| By default, it won't override existing environment variables; you can do that with the `-o` flag. |  | ||||||
|  |  | ||||||
| ### Writing Env Files | ### Writing Env Files | ||||||
|  |  | ||||||
| Godotenv can also write a map representing the environment to a correctly-formatted and escaped file | Godotenv can also write a map representing the environment to a correctly-formatted and escaped file | ||||||
| @ -173,17 +163,9 @@ content, err := godotenv.Marshal(env) | |||||||
|  |  | ||||||
| ## Contributing | ## Contributing | ||||||
|  |  | ||||||
| Contributions are welcome, but with some caveats. | Contributions are most welcome! The parser itself is pretty stupidly naive and I wouldn't be surprised if it breaks with edge cases. | ||||||
|  |  | ||||||
| This library has been declared feature complete (see [#182](https://github.com/joho/godotenv/issues/182) for background) and will not be accepting issues or pull requests adding new functionality or breaking the library API. | *code changes without tests will not be accepted* | ||||||
|  |  | ||||||
| Contributions would be gladly accepted that: |  | ||||||
|  |  | ||||||
| * bring this library's parsing into closer compatibility with the mainline dotenv implementations, in particular [Ruby's dotenv](https://github.com/bkeepers/dotenv) and [Node.js' dotenv](https://github.com/motdotla/dotenv) |  | ||||||
| * keep the library up to date with the go ecosystem (ie CI bumps, documentation changes, changes in the core libraries) |  | ||||||
| * bug fixes for use cases that pertain to the library's purpose of easing development of codebases deployed into twelve factor environments |  | ||||||
|  |  | ||||||
| *code changes without tests and references to peer dotenv implementations will not be accepted* |  | ||||||
|  |  | ||||||
| 1. Fork it | 1. Fork it | ||||||
| 2. Create your feature branch (`git checkout -b my-new-feature`) | 2. Create your feature branch (`git checkout -b my-new-feature`) | ||||||
| @ -197,6 +179,10 @@ Releases should follow [Semver](http://semver.org/) though the first couple of r | |||||||
|  |  | ||||||
| Use [annotated tags for all releases](https://github.com/joho/godotenv/issues/30). Example `git tag -a v1.2.1` | Use [annotated tags for all releases](https://github.com/joho/godotenv/issues/30). Example `git tag -a v1.2.1` | ||||||
|  |  | ||||||
|  | ## CI | ||||||
|  |  | ||||||
|  | Linux: [](https://travis-ci.org/joho/godotenv) Windows: [](https://ci.appveyor.com/project/joho/godotenv) | ||||||
|  |  | ||||||
| ## Who? | ## Who? | ||||||
|  |  | ||||||
| The original library [dotenv](https://github.com/bkeepers/dotenv) was written by [Brandon Keepers](http://opensoul.org/), and this port was done by [John Barton](https://johnbarton.co/) based off the tests/fixtures in the original library. | The original library [dotenv](https://github.com/bkeepers/dotenv) was written by [Brandon Keepers](http://opensoul.org/), and this port was done by [John Barton](https://johnbarton.co/) based off the tests/fixtures in the original library. | ||||||
|  | |||||||
| @ -15,15 +15,13 @@ func main() { | |||||||
| 	flag.BoolVar(&showHelp, "h", false, "show help") | 	flag.BoolVar(&showHelp, "h", false, "show help") | ||||||
| 	var rawEnvFilenames string | 	var rawEnvFilenames string | ||||||
| 	flag.StringVar(&rawEnvFilenames, "f", "", "comma separated paths to .env files") | 	flag.StringVar(&rawEnvFilenames, "f", "", "comma separated paths to .env files") | ||||||
| 	var overload bool |  | ||||||
| 	flag.BoolVar(&overload, "o", false, "override existing .env variables") |  | ||||||
|  |  | ||||||
| 	flag.Parse() | 	flag.Parse() | ||||||
|  |  | ||||||
| 	usage := ` | 	usage := ` | ||||||
| Run a process with an env setup from a .env file | Run a process with an env setup from a .env file | ||||||
|  |  | ||||||
| godotenv [-o] [-f ENV_FILE_PATHS] COMMAND_ARGS | godotenv [-f ENV_FILE_PATHS] COMMAND_ARGS | ||||||
|  |  | ||||||
| ENV_FILE_PATHS: comma separated paths to .env files | ENV_FILE_PATHS: comma separated paths to .env files | ||||||
| COMMAND_ARGS: command and args you want to run | COMMAND_ARGS: command and args you want to run | ||||||
| @ -49,7 +47,7 @@ example | |||||||
| 	cmd := args[0] | 	cmd := args[0] | ||||||
| 	cmdArgs := args[1:] | 	cmdArgs := args[1:] | ||||||
|  |  | ||||||
| 	err := godotenv.Exec(envFilenames, cmd, cmdArgs, overload) | 	err := godotenv.Exec(envFilenames, cmd, cmdArgs) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		log.Fatal(err) | 		log.Fatal(err) | ||||||
| 	} | 	} | ||||||
|  | |||||||
| @ -1,4 +0,0 @@ | |||||||
| # Full line comment |  | ||||||
| foo=bar # baz |  | ||||||
| bar=foo#baz |  | ||||||
| baz="foo"#bar |  | ||||||
| @ -1 +1,2 @@ | |||||||
| export OPTION_A='postgres://localhost:5432/database?sslmode=disable' | export OPTION_A='postgres://localhost:5432/database?sslmode=disable' | ||||||
|  |  | ||||||
|  | |||||||
| @ -7,13 +7,3 @@ OPTION_F="2" | |||||||
| OPTION_G="" | OPTION_G="" | ||||||
| OPTION_H="\n" | OPTION_H="\n" | ||||||
| OPTION_I = "echo 'asd'" | OPTION_I = "echo 'asd'" | ||||||
| OPTION_J='line 1 |  | ||||||
| line 2' |  | ||||||
| OPTION_K='line one |  | ||||||
| this is \'quoted\' |  | ||||||
| one more line' |  | ||||||
| OPTION_L="line 1 |  | ||||||
| line 2" |  | ||||||
| OPTION_M="line one |  | ||||||
| this is \"quoted\" |  | ||||||
| one more line" |  | ||||||
|  | |||||||
							
								
								
									
										113
									
								
								godotenv.go
									
									
									
									
									
								
							
							
						
						
									
										113
									
								
								godotenv.go
									
									
									
									
									
								
							| @ -1,20 +1,20 @@ | |||||||
| // Package godotenv is a go port of the ruby dotenv library (https://github.com/bkeepers/dotenv) | // Package godotenv is a go port of the ruby dotenv library (https://github.com/bkeepers/dotenv) | ||||||
| // | // | ||||||
| // Examples/readme can be found on the GitHub page at https://github.com/joho/godotenv | // Examples/readme can be found on the github page at https://github.com/joho/godotenv | ||||||
| // | // | ||||||
| // The TL;DR is that you make a .env file that looks something like | // The TL;DR is that you make a .env file that looks something like | ||||||
| // | // | ||||||
| //	SOME_ENV_VAR=somevalue | // 		SOME_ENV_VAR=somevalue | ||||||
| // | // | ||||||
| // and then in your go code you can call | // and then in your go code you can call | ||||||
| // | // | ||||||
| //	godotenv.Load() | // 		godotenv.Load() | ||||||
| // | // | ||||||
| // and all the env vars declared in .env will be available through os.Getenv("SOME_ENV_VAR") | // and all the env vars declared in .env will be available through os.Getenv("SOME_ENV_VAR") | ||||||
| package godotenv | package godotenv | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"bytes" | 	"bufio" | ||||||
| 	"errors" | 	"errors" | ||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"io" | 	"io" | ||||||
| @ -28,28 +28,17 @@ import ( | |||||||
|  |  | ||||||
| const doubleQuoteSpecialChars = "\\\n\r\"!$`" | const doubleQuoteSpecialChars = "\\\n\r\"!$`" | ||||||
|  |  | ||||||
| // Parse reads an env file from io.Reader, returning a map of keys and values. |  | ||||||
| func Parse(r io.Reader) (map[string]string, error) { |  | ||||||
| 	var buf bytes.Buffer |  | ||||||
| 	_, err := io.Copy(&buf, r) |  | ||||||
| 	if err != nil { |  | ||||||
| 		return nil, err |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	return UnmarshalBytes(buf.Bytes()) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Load will read your env file(s) and load them into ENV for this process. | // Load will read your env file(s) and load them into ENV for this process. | ||||||
| // | // | ||||||
| // Call this function as close as possible to the start of your program (ideally in main). | // Call this function as close as possible to the start of your program (ideally in main) | ||||||
| // | // | ||||||
| // If you call Load without any args it will default to loading .env in the current path. | // If you call Load without any args it will default to loading .env in the current path | ||||||
| // | // | ||||||
| // You can otherwise tell it which files to load (there can be more than one) like: | // You can otherwise tell it which files to load (there can be more than one) like | ||||||
| // | // | ||||||
| //	godotenv.Load("fileone", "filetwo") | //		godotenv.Load("fileone", "filetwo") | ||||||
| // | // | ||||||
| // It's important to note that it WILL NOT OVERRIDE an env variable that already exists - consider the .env file to set dev vars or sensible defaults. | // It's important to note that it WILL NOT OVERRIDE an env variable that already exists - consider the .env file to set dev vars or sensible defaults | ||||||
| func Load(filenames ...string) (err error) { | func Load(filenames ...string) (err error) { | ||||||
| 	filenames = filenamesOrDefault(filenames) | 	filenames = filenamesOrDefault(filenames) | ||||||
|  |  | ||||||
| @ -64,15 +53,15 @@ func Load(filenames ...string) (err error) { | |||||||
|  |  | ||||||
| // Overload will read your env file(s) and load them into ENV for this process. | // Overload will read your env file(s) and load them into ENV for this process. | ||||||
| // | // | ||||||
| // Call this function as close as possible to the start of your program (ideally in main). | // Call this function as close as possible to the start of your program (ideally in main) | ||||||
| // | // | ||||||
| // If you call Overload without any args it will default to loading .env in the current path. | // If you call Overload without any args it will default to loading .env in the current path | ||||||
| // | // | ||||||
| // You can otherwise tell it which files to load (there can be more than one) like: | // You can otherwise tell it which files to load (there can be more than one) like | ||||||
| // | // | ||||||
| //	godotenv.Overload("fileone", "filetwo") | //		godotenv.Overload("fileone", "filetwo") | ||||||
| // | // | ||||||
| // It's important to note this WILL OVERRIDE an env variable that already exists - consider the .env file to forcefully set all vars. | // It's important to note this WILL OVERRIDE an env variable that already exists - consider the .env file to forcefilly set all vars. | ||||||
| func Overload(filenames ...string) (err error) { | func Overload(filenames ...string) (err error) { | ||||||
| 	filenames = filenamesOrDefault(filenames) | 	filenames = filenamesOrDefault(filenames) | ||||||
|  |  | ||||||
| @ -107,34 +96,48 @@ func Read(filenames ...string) (envMap map[string]string, err error) { | |||||||
| 	return | 	return | ||||||
| } | } | ||||||
|  |  | ||||||
| // Unmarshal reads an env file from a string, returning a map of keys and values. | // Parse reads an env file from io.Reader, returning a map of keys and values. | ||||||
| func Unmarshal(str string) (envMap map[string]string, err error) { | func Parse(r io.Reader) (envMap map[string]string, err error) { | ||||||
| 	return UnmarshalBytes([]byte(str)) | 	envMap = make(map[string]string) | ||||||
|  |  | ||||||
|  | 	var lines []string | ||||||
|  | 	scanner := bufio.NewScanner(r) | ||||||
|  | 	for scanner.Scan() { | ||||||
|  | 		lines = append(lines, scanner.Text()) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if err = scanner.Err(); err != nil { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	for _, fullLine := range lines { | ||||||
|  | 		if !isIgnoredLine(fullLine) { | ||||||
|  | 			var key, value string | ||||||
|  | 			key, value, err = parseLine(fullLine, envMap) | ||||||
|  |  | ||||||
|  | 			if err != nil { | ||||||
|  | 				return | ||||||
|  | 			} | ||||||
|  | 			envMap[key] = value | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return | ||||||
| } | } | ||||||
|  |  | ||||||
| // UnmarshalBytes parses env file from byte slice of chars, returning a map of keys and values. | //Unmarshal reads an env file from a string, returning a map of keys and values. | ||||||
| func UnmarshalBytes(src []byte) (map[string]string, error) { | func Unmarshal(str string) (envMap map[string]string, err error) { | ||||||
| 	out := make(map[string]string) | 	return Parse(strings.NewReader(str)) | ||||||
| 	err := parseBytes(src, out) |  | ||||||
|  |  | ||||||
| 	return out, err |  | ||||||
| } | } | ||||||
|  |  | ||||||
| // Exec loads env vars from the specified filenames (empty map falls back to default) | // Exec loads env vars from the specified filenames (empty map falls back to default) | ||||||
| // then executes the cmd specified. | // then executes the cmd specified. | ||||||
| // | // | ||||||
| // Simply hooks up os.Stdin/err/out to the command and calls Run(). | // Simply hooks up os.Stdin/err/out to the command and calls Run() | ||||||
| // | // | ||||||
| // If you want more fine grained control over your command it's recommended | // If you want more fine grained control over your command it's recommended | ||||||
| // that you use `Load()`, `Overload()` or `Read()` and the `os/exec` package yourself. | // that you use `Load()` or `Read()` and the `os/exec` package yourself. | ||||||
| func Exec(filenames []string, cmd string, cmdArgs []string, overload bool) error { | func Exec(filenames []string, cmd string, cmdArgs []string) error { | ||||||
| 	op := Load | 	Load(filenames...) | ||||||
| 	if overload { |  | ||||||
| 		op = Overload |  | ||||||
| 	} |  | ||||||
| 	if err := op(filenames...); err != nil { |  | ||||||
| 		return err |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	command := exec.Command(cmd, cmdArgs...) | 	command := exec.Command(cmd, cmdArgs...) | ||||||
| 	command.Stdin = os.Stdin | 	command.Stdin = os.Stdin | ||||||
| @ -143,7 +146,7 @@ func Exec(filenames []string, cmd string, cmdArgs []string, overload bool) error | |||||||
| 	return command.Run() | 	return command.Run() | ||||||
| } | } | ||||||
|  |  | ||||||
| // Write serializes the given environment and writes it to a file. | // Write serializes the given environment and writes it to a file | ||||||
| func Write(envMap map[string]string, filename string) error { | func Write(envMap map[string]string, filename string) error { | ||||||
| 	content, err := Marshal(envMap) | 	content, err := Marshal(envMap) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| @ -154,11 +157,12 @@ func Write(envMap map[string]string, filename string) error { | |||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
| 	defer file.Close() | 	defer file.Close() | ||||||
| 	_, err = file.WriteString(content + "\n") | 	_, err = file.WriteString(content) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
| 	return file.Sync() | 	file.Sync() | ||||||
|  | 	return err | ||||||
| } | } | ||||||
|  |  | ||||||
| // Marshal outputs the given environment as a dotenv-formatted environment file. | // Marshal outputs the given environment as a dotenv-formatted environment file. | ||||||
| @ -198,7 +202,7 @@ func loadFile(filename string, overload bool) error { | |||||||
|  |  | ||||||
| 	for key, value := range envMap { | 	for key, value := range envMap { | ||||||
| 		if !currentEnv[key] || overload { | 		if !currentEnv[key] || overload { | ||||||
| 			_ = os.Setenv(key, value) | 			os.Setenv(key, value) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| @ -255,15 +259,15 @@ func parseLine(line string, envMap map[string]string) (key string, value string, | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if len(splitString) != 2 { | 	if len(splitString) != 2 { | ||||||
| 		err = errors.New("can't separate key from value") | 		err = errors.New("Can't separate key from value") | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// Parse the key | 	// Parse the key | ||||||
| 	key = splitString[0] | 	key = splitString[0] | ||||||
|  | 	if strings.HasPrefix(key, "export") { | ||||||
| 	key = strings.TrimPrefix(key, "export") | 		key = strings.TrimPrefix(key, "export") | ||||||
|  | 	} | ||||||
| 	key = strings.TrimSpace(key) | 	key = strings.TrimSpace(key) | ||||||
|  |  | ||||||
| 	key = exportRegex.ReplaceAllString(splitString[0], "$1") | 	key = exportRegex.ReplaceAllString(splitString[0], "$1") | ||||||
| @ -339,6 +343,11 @@ func expandVariables(v string, m map[string]string) string { | |||||||
| 	}) | 	}) | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func isIgnoredLine(line string) bool { | ||||||
|  | 	trimmedLine := strings.TrimSpace(line) | ||||||
|  | 	return len(trimmedLine) == 0 || strings.HasPrefix(trimmedLine, "#") | ||||||
|  | } | ||||||
|  |  | ||||||
| func doubleQuoteEscape(line string) string { | func doubleQuoteEscape(line string) string { | ||||||
| 	for _, c := range doubleQuoteSpecialChars { | 	for _, c := range doubleQuoteSpecialChars { | ||||||
| 		toReplace := "\\" + string(c) | 		toReplace := "\\" + string(c) | ||||||
|  | |||||||
							
								
								
									
										106
									
								
								godotenv_test.go
									
									
									
									
									
								
							
							
						
						
									
										106
									
								
								godotenv_test.go
									
									
									
									
									
								
							| @ -35,7 +35,7 @@ func loadEnvAndCompareValues(t *testing.T, loader func(files ...string) error, e | |||||||
| 		envValue := os.Getenv(k) | 		envValue := os.Getenv(k) | ||||||
| 		v := expectedValues[k] | 		v := expectedValues[k] | ||||||
| 		if envValue != v { | 		if envValue != v { | ||||||
| 			t.Errorf("Mismatch for key '%v': expected '%#v' got '%#v'", k, v, envValue) | 			t.Errorf("Mismatch for key '%v': expected '%v' got '%v'", k, v, envValue) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
| @ -131,7 +131,7 @@ func TestLoadDoesNotOverride(t *testing.T) { | |||||||
| 	loadEnvAndCompareValues(t, Load, envFileName, expectedValues, presets) | 	loadEnvAndCompareValues(t, Load, envFileName, expectedValues, presets) | ||||||
| } | } | ||||||
|  |  | ||||||
| func TestOverloadDoesOverride(t *testing.T) { | func TestOveroadDoesOverride(t *testing.T) { | ||||||
| 	envFileName := "fixtures/plain.env" | 	envFileName := "fixtures/plain.env" | ||||||
|  |  | ||||||
| 	// ensure NO overload | 	// ensure NO overload | ||||||
| @ -189,10 +189,6 @@ func TestLoadQuotedEnv(t *testing.T) { | |||||||
| 		"OPTION_G": "", | 		"OPTION_G": "", | ||||||
| 		"OPTION_H": "\n", | 		"OPTION_H": "\n", | ||||||
| 		"OPTION_I": "echo 'asd'", | 		"OPTION_I": "echo 'asd'", | ||||||
| 		"OPTION_J": "line 1\nline 2", |  | ||||||
| 		"OPTION_K": "line one\nthis is \\'quoted\\'\none more line", |  | ||||||
| 		"OPTION_L": "line 1\nline 2", |  | ||||||
| 		"OPTION_M": "line one\nthis is \"quoted\"\none more line", |  | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	loadEnvAndCompareValues(t, Load, envFileName, expectedValues, noopPresets) | 	loadEnvAndCompareValues(t, Load, envFileName, expectedValues, noopPresets) | ||||||
| @ -275,34 +271,6 @@ func TestExpanding(t *testing.T) { | |||||||
|  |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func TestVariableStringValueSeparator(t *testing.T) { |  | ||||||
| 	input := "TEST_URLS=\"stratum+tcp://stratum.antpool.com:3333\nstratum+tcp://stratum.antpool.com:443\"" |  | ||||||
| 	want := map[string]string{ |  | ||||||
| 		"TEST_URLS": "stratum+tcp://stratum.antpool.com:3333\nstratum+tcp://stratum.antpool.com:443", |  | ||||||
| 	} |  | ||||||
| 	got, err := Parse(strings.NewReader(input)) |  | ||||||
| 	if err != nil { |  | ||||||
| 		t.Error(err) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if len(got) != len(want) { |  | ||||||
| 		t.Fatalf( |  | ||||||
| 			"unexpected value:\nwant:\n\t%#v\n\ngot:\n\t%#v", want, got) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	for k, wantVal := range want { |  | ||||||
| 		gotVal, ok := got[k] |  | ||||||
| 		if !ok { |  | ||||||
| 			t.Fatalf("key %q doesn't present in result", k) |  | ||||||
| 		} |  | ||||||
| 		if wantVal != gotVal { |  | ||||||
| 			t.Fatalf( |  | ||||||
| 				"mismatch in %q value:\nwant:\n\t%s\n\ngot:\n\t%s", k, |  | ||||||
| 				wantVal, gotVal) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestActualEnvVarsAreLeftAlone(t *testing.T) { | func TestActualEnvVarsAreLeftAlone(t *testing.T) { | ||||||
| 	os.Clearenv() | 	os.Clearenv() | ||||||
| 	os.Setenv("OPTION_A", "actualenv") | 	os.Setenv("OPTION_A", "actualenv") | ||||||
| @ -357,11 +325,11 @@ func TestParsing(t *testing.T) { | |||||||
| 	// expect(env('FOO="bar\nbaz"')).to eql('FOO' => "bar\nbaz") | 	// expect(env('FOO="bar\nbaz"')).to eql('FOO' => "bar\nbaz") | ||||||
| 	parseAndCompare(t, `FOO="bar\nbaz"`, "FOO", "bar\nbaz") | 	parseAndCompare(t, `FOO="bar\nbaz"`, "FOO", "bar\nbaz") | ||||||
|  |  | ||||||
| 	// it 'parses variables with "." in the name' do | 	// it 'parses varibales with "." in the name' do | ||||||
| 	// expect(env('FOO.BAR=foobar')).to eql('FOO.BAR' => 'foobar') | 	// expect(env('FOO.BAR=foobar')).to eql('FOO.BAR' => 'foobar') | ||||||
| 	parseAndCompare(t, "FOO.BAR=foobar", "FOO.BAR", "foobar") | 	parseAndCompare(t, "FOO.BAR=foobar", "FOO.BAR", "foobar") | ||||||
|  |  | ||||||
| 	// it 'parses variables with several "=" in the value' do | 	// it 'parses varibales with several "=" in the value' do | ||||||
| 	// expect(env('FOO=foobar=')).to eql('FOO' => 'foobar=') | 	// expect(env('FOO=foobar=')).to eql('FOO' => 'foobar=') | ||||||
| 	parseAndCompare(t, "FOO=foobar=", "FOO", "foobar=") | 	parseAndCompare(t, "FOO=foobar=", "FOO", "foobar=") | ||||||
|  |  | ||||||
| @ -409,38 +377,33 @@ func TestParsing(t *testing.T) { | |||||||
| } | } | ||||||
|  |  | ||||||
| func TestLinesToIgnore(t *testing.T) { | func TestLinesToIgnore(t *testing.T) { | ||||||
| 	cases := map[string]struct { | 	// it 'ignores empty lines' do | ||||||
| 		input string | 	// expect(env("\n \t  \nfoo=bar\n \nfizz=buzz")).to eql('foo' => 'bar', 'fizz' => 'buzz') | ||||||
| 		want  string | 	if !isIgnoredLine("\n") { | ||||||
| 	}{ | 		t.Error("Line with nothing but line break wasn't ignored") | ||||||
| 		"Line with nothing but line break": { |  | ||||||
| 			input: "\n", |  | ||||||
| 		}, |  | ||||||
| 		"Line with nothing but windows-style line break": { |  | ||||||
| 			input: "\r\n", |  | ||||||
| 		}, |  | ||||||
| 		"Line full of whitespace": { |  | ||||||
| 			input: "\t\t ", |  | ||||||
| 		}, |  | ||||||
| 		"Comment": { |  | ||||||
| 			input: "# Comment", |  | ||||||
| 		}, |  | ||||||
| 		"Indented comment": { |  | ||||||
| 			input: "\t # comment", |  | ||||||
| 		}, |  | ||||||
| 		"non-ignored value": { |  | ||||||
| 			input: `export OPTION_B='\n'`, |  | ||||||
| 			want:  `export OPTION_B='\n'`, |  | ||||||
| 		}, |  | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	for n, c := range cases { | 	if !isIgnoredLine("\r\n") { | ||||||
| 		t.Run(n, func(t *testing.T) { | 		t.Error("Line with nothing but windows-style line break wasn't ignored") | ||||||
| 			got := string(getStatementStart([]byte(c.input))) | 	} | ||||||
| 			if got != c.want { |  | ||||||
| 				t.Errorf("Expected:\t %q\nGot:\t %q", c.want, got) | 	if !isIgnoredLine("\t\t ") { | ||||||
| 			} | 		t.Error("Line full of whitespace wasn't ignored") | ||||||
| 		}) | 	} | ||||||
|  |  | ||||||
|  | 	// it 'ignores comment lines' do | ||||||
|  | 	// expect(env("\n\n\n # HERE GOES FOO \nfoo=bar")).to eql('foo' => 'bar') | ||||||
|  | 	if !isIgnoredLine("# comment") { | ||||||
|  | 		t.Error("Comment wasn't ignored") | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if !isIgnoredLine("\t#comment") { | ||||||
|  | 		t.Error("Indented comment wasn't ignored") | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// make sure we're not getting false positives | ||||||
|  | 	if isIgnoredLine(`export OPTION_B='\n'`) { | ||||||
|  | 		t.Error("ignoring a perfectly valid line to parse") | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| @ -461,17 +424,6 @@ func TestErrorParsing(t *testing.T) { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func TestComments(t *testing.T) { |  | ||||||
| 	envFileName := "fixtures/comments.env" |  | ||||||
| 	expectedValues := map[string]string{ |  | ||||||
| 		"foo": "bar", |  | ||||||
| 		"bar": "foo#baz", |  | ||||||
| 		"baz": "foo", |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	loadEnvAndCompareValues(t, Load, envFileName, expectedValues, noopPresets) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestWrite(t *testing.T) { | func TestWrite(t *testing.T) { | ||||||
| 	writeAndCompare := func(env string, expected string) { | 	writeAndCompare := func(env string, expected string) { | ||||||
| 		envMap, _ := Unmarshal(env) | 		envMap, _ := Unmarshal(env) | ||||||
|  | |||||||
							
								
								
									
										207
									
								
								parser.go
									
									
									
									
									
								
							
							
						
						
									
										207
									
								
								parser.go
									
									
									
									
									
								
							| @ -1,207 +0,0 @@ | |||||||
| package godotenv |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"bytes" |  | ||||||
| 	"errors" |  | ||||||
| 	"fmt" |  | ||||||
| 	"strings" |  | ||||||
| 	"unicode" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| const ( |  | ||||||
| 	charComment       = '#' |  | ||||||
| 	prefixSingleQuote = '\'' |  | ||||||
| 	prefixDoubleQuote = '"' |  | ||||||
|  |  | ||||||
| 	exportPrefix = "export" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| func parseBytes(src []byte, out map[string]string) error { |  | ||||||
| 	src = bytes.Replace(src, []byte("\r\n"), []byte("\n"), -1) |  | ||||||
| 	cutset := src |  | ||||||
| 	for { |  | ||||||
| 		cutset = getStatementStart(cutset) |  | ||||||
| 		if cutset == nil { |  | ||||||
| 			// reached end of file |  | ||||||
| 			break |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		key, left, err := locateKeyName(cutset) |  | ||||||
| 		if err != nil { |  | ||||||
| 			return err |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		value, left, err := extractVarValue(left, out) |  | ||||||
| 		if err != nil { |  | ||||||
| 			return err |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		out[key] = value |  | ||||||
| 		cutset = left |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // getStatementPosition returns position of statement begin. |  | ||||||
| // |  | ||||||
| // It skips any comment line or non-whitespace character. |  | ||||||
| func getStatementStart(src []byte) []byte { |  | ||||||
| 	pos := indexOfNonSpaceChar(src) |  | ||||||
| 	if pos == -1 { |  | ||||||
| 		return nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	src = src[pos:] |  | ||||||
| 	if src[0] != charComment { |  | ||||||
| 		return src |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// skip comment section |  | ||||||
| 	pos = bytes.IndexFunc(src, isCharFunc('\n')) |  | ||||||
| 	if pos == -1 { |  | ||||||
| 		return nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	return getStatementStart(src[pos:]) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // locateKeyName locates and parses key name and returns rest of slice |  | ||||||
| func locateKeyName(src []byte) (key string, cutset []byte, err error) { |  | ||||||
| 	// trim "export" and space at beginning |  | ||||||
| 	src = bytes.TrimLeftFunc(bytes.TrimPrefix(src, []byte(exportPrefix)), isSpace) |  | ||||||
|  |  | ||||||
| 	// locate key name end and validate it in single loop |  | ||||||
| 	offset := 0 |  | ||||||
| loop: |  | ||||||
| 	for i, char := range src { |  | ||||||
| 		rchar := rune(char) |  | ||||||
| 		if isSpace(rchar) { |  | ||||||
| 			continue |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		switch char { |  | ||||||
| 		case '=', ':': |  | ||||||
| 			// library also supports yaml-style value declaration |  | ||||||
| 			key = string(src[0:i]) |  | ||||||
| 			offset = i + 1 |  | ||||||
| 			break loop |  | ||||||
| 		case '_': |  | ||||||
| 		default: |  | ||||||
| 			// variable name should match [A-Za-z0-9_] |  | ||||||
| 			if unicode.IsLetter(rchar) || unicode.IsNumber(rchar) { |  | ||||||
| 				continue |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			return "", nil, fmt.Errorf( |  | ||||||
| 				`unexpected character %q in variable name near %q`, |  | ||||||
| 				string(char), string(src)) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if len(src) == 0 { |  | ||||||
| 		return "", nil, errors.New("zero length string") |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// trim whitespace |  | ||||||
| 	key = strings.TrimRightFunc(key, unicode.IsSpace) |  | ||||||
| 	cutset = bytes.TrimLeftFunc(src[offset:], isSpace) |  | ||||||
| 	return key, cutset, nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // extractVarValue extracts variable value and returns rest of slice |  | ||||||
| func extractVarValue(src []byte, vars map[string]string) (value string, rest []byte, err error) { |  | ||||||
| 	quote, hasPrefix := hasQuotePrefix(src) |  | ||||||
| 	if !hasPrefix { |  | ||||||
| 		// unquoted value - read until whitespace |  | ||||||
| 		end := bytes.IndexFunc(src, unicode.IsSpace) |  | ||||||
| 		if end == -1 { |  | ||||||
| 			return expandVariables(string(src), vars), nil, nil |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		return expandVariables(string(src[0:end]), vars), src[end:], nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// lookup quoted string terminator |  | ||||||
| 	for i := 1; i < len(src); i++ { |  | ||||||
| 		if char := src[i]; char != quote { |  | ||||||
| 			continue |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		// skip escaped quote symbol (\" or \', depends on quote) |  | ||||||
| 		if prevChar := src[i-1]; prevChar == '\\' { |  | ||||||
| 			continue |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		// trim quotes |  | ||||||
| 		trimFunc := isCharFunc(rune(quote)) |  | ||||||
| 		value = string(bytes.TrimLeftFunc(bytes.TrimRightFunc(src[0:i], trimFunc), trimFunc)) |  | ||||||
| 		if quote == prefixDoubleQuote { |  | ||||||
| 			// unescape newlines for double quote (this is compat feature) |  | ||||||
| 			// and expand environment variables |  | ||||||
| 			value = expandVariables(expandEscapes(value), vars) |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		return value, src[i+1:], nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// return formatted error if quoted string is not terminated |  | ||||||
| 	valEndIndex := bytes.IndexFunc(src, isCharFunc('\n')) |  | ||||||
| 	if valEndIndex == -1 { |  | ||||||
| 		valEndIndex = len(src) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	return "", nil, fmt.Errorf("unterminated quoted value %s", src[:valEndIndex]) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func expandEscapes(str string) string { |  | ||||||
| 	out := escapeRegex.ReplaceAllStringFunc(str, func(match string) string { |  | ||||||
| 		c := strings.TrimPrefix(match, `\`) |  | ||||||
| 		switch c { |  | ||||||
| 		case "n": |  | ||||||
| 			return "\n" |  | ||||||
| 		case "r": |  | ||||||
| 			return "\r" |  | ||||||
| 		default: |  | ||||||
| 			return match |  | ||||||
| 		} |  | ||||||
| 	}) |  | ||||||
| 	return unescapeCharsRegex.ReplaceAllString(out, "$1") |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func indexOfNonSpaceChar(src []byte) int { |  | ||||||
| 	return bytes.IndexFunc(src, func(r rune) bool { |  | ||||||
| 		return !unicode.IsSpace(r) |  | ||||||
| 	}) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // hasQuotePrefix reports whether charset starts with single or double quote and returns quote character |  | ||||||
| func hasQuotePrefix(src []byte) (prefix byte, isQuored bool) { |  | ||||||
| 	if len(src) == 0 { |  | ||||||
| 		return 0, false |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	switch prefix := src[0]; prefix { |  | ||||||
| 	case prefixDoubleQuote, prefixSingleQuote: |  | ||||||
| 		return prefix, true |  | ||||||
| 	default: |  | ||||||
| 		return 0, false |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func isCharFunc(char rune) func(rune) bool { |  | ||||||
| 	return func(v rune) bool { |  | ||||||
| 		return v == char |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // isSpace reports whether the rune is a space character but not line break character |  | ||||||
| // |  | ||||||
| // this differs from unicode.IsSpace, which also applies line break as space |  | ||||||
| func isSpace(r rune) bool { |  | ||||||
| 	switch r { |  | ||||||
| 	case '\t', '\v', '\f', '\r', ' ', 0x85, 0xA0: |  | ||||||
| 		return true |  | ||||||
| 	} |  | ||||||
| 	return false |  | ||||||
| } |  | ||||||
		Reference in New Issue
	
	Block a user
	