parent
c0904f1942
commit
de8b73dd92
@ -0,0 +1,23 @@ |
|||||||
|
- |
||||||
|
id: 1 |
||||||
|
uid: 1 |
||||||
|
name: Token A |
||||||
|
sha1: hash1 |
||||||
|
created_unix: 946687980 |
||||||
|
updated_unix: 946687980 |
||||||
|
|
||||||
|
- |
||||||
|
id: 2 |
||||||
|
uid: 1 |
||||||
|
name: Token B |
||||||
|
sha1: hash2 |
||||||
|
created_unix: 946687980 |
||||||
|
updated_unix: 946687980 |
||||||
|
|
||||||
|
- |
||||||
|
id: 3 |
||||||
|
uid: 2 |
||||||
|
name: Token A |
||||||
|
sha1: hash3 |
||||||
|
created_unix: 946687980 |
||||||
|
updated_unix: 946687980 |
@ -0,0 +1,20 @@ |
|||||||
|
- |
||||||
|
id: 1 |
||||||
|
uuid: uuid1 |
||||||
|
ref_name: refName1 |
||||||
|
old_commit_id: oldCommitId1 |
||||||
|
new_commit_id: newCommitId1 |
||||||
|
|
||||||
|
- |
||||||
|
id: 2 |
||||||
|
uuid: uuid2 |
||||||
|
ref_name: refName2 |
||||||
|
old_commit_id: oldCommitId2 |
||||||
|
new_commit_id: newCommitId2 |
||||||
|
|
||||||
|
- |
||||||
|
id: 3 |
||||||
|
uuid: uuid3 |
||||||
|
ref_name: refName3 |
||||||
|
old_commit_id: oldCommitId3 |
||||||
|
new_commit_id: newCommitId3 |
@ -0,0 +1,47 @@ |
|||||||
|
// Copyright 2016 The Gitea Authors. All rights reserved.
|
||||||
|
// Use of this source code is governed by a MIT-style
|
||||||
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
|
package models |
||||||
|
|
||||||
|
import ( |
||||||
|
"fmt" |
||||||
|
"os" |
||||||
|
"testing" |
||||||
|
|
||||||
|
"github.com/go-xorm/core" |
||||||
|
"github.com/go-xorm/xorm" |
||||||
|
_ "github.com/mattn/go-sqlite3" // for the test engine
|
||||||
|
"gopkg.in/testfixtures.v2" |
||||||
|
) |
||||||
|
|
||||||
|
func TestMain(m *testing.M) { |
||||||
|
if err := CreateTestEngine(); err != nil { |
||||||
|
fmt.Printf("Error creating test engine: %v\n", err) |
||||||
|
os.Exit(1) |
||||||
|
} |
||||||
|
os.Exit(m.Run()) |
||||||
|
} |
||||||
|
|
||||||
|
var fixtures *testfixtures.Context |
||||||
|
|
||||||
|
// CreateTestEngine create an xorm engine for testing
|
||||||
|
func CreateTestEngine() error { |
||||||
|
testfixtures.SkipDatabaseNameCheck(true) |
||||||
|
var err error |
||||||
|
x, err = xorm.NewEngine("sqlite3", "file::memory:?cache=shared") |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
x.SetMapper(core.GonicMapper{}) |
||||||
|
if err = x.StoreEngine("InnoDB").Sync2(tables...); err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
fixtures, err = testfixtures.NewFolder(x.DB().DB, &testfixtures.SQLite{}, "fixtures/") |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
// PrepareTestDatabase load test fixtures into test database
|
||||||
|
func PrepareTestDatabase() error { |
||||||
|
return fixtures.Load() |
||||||
|
} |
@ -0,0 +1,107 @@ |
|||||||
|
// Copyright 2016 The Gitea Authors. All rights reserved.
|
||||||
|
// Use of this source code is governed by a MIT-style
|
||||||
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
|
package models |
||||||
|
|
||||||
|
import ( |
||||||
|
"testing" |
||||||
|
|
||||||
|
"github.com/stretchr/testify/assert" |
||||||
|
) |
||||||
|
|
||||||
|
func TestNewAccessToken(t *testing.T) { |
||||||
|
assert.NoError(t, PrepareTestDatabase()) |
||||||
|
token := &AccessToken{ |
||||||
|
UID: 3, |
||||||
|
Name: "Token C", |
||||||
|
} |
||||||
|
assert.NoError(t, NewAccessToken(token)) |
||||||
|
sess := x.NewSession() |
||||||
|
defer sess.Close() |
||||||
|
has, err := sess.Get(*token) |
||||||
|
assert.NoError(t, err) |
||||||
|
assert.True(t, has) |
||||||
|
assert.Equal(t, int64(3), token.UID) |
||||||
|
assert.Equal(t, "Token C", token.Name) |
||||||
|
|
||||||
|
invalidToken := &AccessToken{ |
||||||
|
ID: token.ID, // duplicate
|
||||||
|
UID: 2, |
||||||
|
Name: "Token F", |
||||||
|
} |
||||||
|
assert.Error(t, NewAccessToken(invalidToken)) |
||||||
|
} |
||||||
|
|
||||||
|
func TestGetAccessTokenBySHA(t *testing.T) { |
||||||
|
assert.NoError(t, PrepareTestDatabase()) |
||||||
|
token, err := GetAccessTokenBySHA("hash1") |
||||||
|
assert.NoError(t, err) |
||||||
|
assert.Equal(t, int64(1), token.UID) |
||||||
|
assert.Equal(t, "Token A", token.Name) |
||||||
|
assert.Equal(t, "hash1", token.Sha1) |
||||||
|
|
||||||
|
token, err = GetAccessTokenBySHA("notahash") |
||||||
|
assert.Error(t, err) |
||||||
|
assert.True(t, IsErrAccessTokenNotExist(err)) |
||||||
|
|
||||||
|
token, err = GetAccessTokenBySHA("") |
||||||
|
assert.Error(t, err) |
||||||
|
assert.True(t, IsErrAccessTokenEmpty(err)) |
||||||
|
} |
||||||
|
|
||||||
|
func TestListAccessTokens(t *testing.T) { |
||||||
|
assert.NoError(t, PrepareTestDatabase()) |
||||||
|
tokens, err := ListAccessTokens(1) |
||||||
|
assert.NoError(t, err) |
||||||
|
assert.Len(t, tokens, 2) |
||||||
|
assert.Equal(t, int64(1), tokens[0].UID) |
||||||
|
assert.Equal(t, int64(1), tokens[1].UID) |
||||||
|
assert.Contains(t, []string{tokens[0].Name, tokens[1].Name}, "Token A") |
||||||
|
assert.Contains(t, []string{tokens[0].Name, tokens[1].Name}, "Token B") |
||||||
|
|
||||||
|
tokens, err = ListAccessTokens(2) |
||||||
|
assert.NoError(t, err) |
||||||
|
assert.Len(t, tokens, 1) |
||||||
|
assert.Equal(t, int64(2), tokens[0].UID) |
||||||
|
assert.Equal(t, "Token A", tokens[0].Name) |
||||||
|
|
||||||
|
tokens, err = ListAccessTokens(100) |
||||||
|
assert.NoError(t, err) |
||||||
|
assert.Empty(t, tokens) |
||||||
|
} |
||||||
|
|
||||||
|
func TestUpdateAccessToken(t *testing.T) { |
||||||
|
assert.NoError(t, PrepareTestDatabase()) |
||||||
|
token, err := GetAccessTokenBySHA("hash2") |
||||||
|
assert.NoError(t, err) |
||||||
|
token.Name = "Token Z" |
||||||
|
|
||||||
|
assert.NoError(t, UpdateAccessToken(token)) |
||||||
|
|
||||||
|
sess := x.NewSession() |
||||||
|
defer sess.Close() |
||||||
|
has, err := sess.Get(token) |
||||||
|
assert.NoError(t, err) |
||||||
|
assert.True(t, has) |
||||||
|
assert.Equal(t, token.Name, "Token Z") |
||||||
|
} |
||||||
|
|
||||||
|
func TestDeleteAccessTokenByID(t *testing.T) { |
||||||
|
assert.NoError(t, PrepareTestDatabase()) |
||||||
|
|
||||||
|
token, err := GetAccessTokenBySHA("hash2") |
||||||
|
assert.NoError(t, err) |
||||||
|
assert.Equal(t, int64(1), token.UID) |
||||||
|
|
||||||
|
assert.NoError(t, DeleteAccessTokenByID(token.ID, 1)) |
||||||
|
sess := x.NewSession() |
||||||
|
defer sess.Close() |
||||||
|
has, err := sess.Get(token) |
||||||
|
assert.NoError(t, err) |
||||||
|
assert.False(t, has) |
||||||
|
|
||||||
|
err = DeleteAccessTokenByID(100, 100) |
||||||
|
assert.Error(t, err) |
||||||
|
assert.True(t, IsErrAccessTokenNotExist(err)) |
||||||
|
} |
@ -0,0 +1,124 @@ |
|||||||
|
// Copyright 2016 The Gitea Authors. All rights reserved.
|
||||||
|
// Use of this source code is governed by a MIT-style
|
||||||
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
|
package models |
||||||
|
|
||||||
|
import ( |
||||||
|
"container/list" |
||||||
|
"testing" |
||||||
|
"time" |
||||||
|
|
||||||
|
"code.gitea.io/git" |
||||||
|
|
||||||
|
"github.com/stretchr/testify/assert" |
||||||
|
) |
||||||
|
|
||||||
|
func TestAddUpdateTask(t *testing.T) { |
||||||
|
assert.NoError(t, PrepareTestDatabase()) |
||||||
|
task := &UpdateTask{ |
||||||
|
UUID: "uuid4", |
||||||
|
RefName: "refName4", |
||||||
|
OldCommitID: "oldCommitId4", |
||||||
|
NewCommitID: "newCommitId4", |
||||||
|
} |
||||||
|
assert.NoError(t, AddUpdateTask(task)) |
||||||
|
|
||||||
|
sess := x.NewSession() |
||||||
|
defer sess.Close() |
||||||
|
has, err := sess.Get(task) |
||||||
|
assert.NoError(t, err) |
||||||
|
assert.True(t, has) |
||||||
|
assert.Equal(t, "uuid4", task.UUID) |
||||||
|
} |
||||||
|
|
||||||
|
func TestGetUpdateTaskByUUID(t *testing.T) { |
||||||
|
assert.NoError(t, PrepareTestDatabase()) |
||||||
|
task, err := GetUpdateTaskByUUID("uuid1") |
||||||
|
assert.NoError(t, err) |
||||||
|
assert.Equal(t, "uuid1", task.UUID) |
||||||
|
assert.Equal(t, "refName1", task.RefName) |
||||||
|
assert.Equal(t, "oldCommitId1", task.OldCommitID) |
||||||
|
assert.Equal(t, "newCommitId1", task.NewCommitID) |
||||||
|
} |
||||||
|
|
||||||
|
func TestDeleteUpdateTaskByUUID(t *testing.T) { |
||||||
|
assert.NoError(t, PrepareTestDatabase()) |
||||||
|
assert.NoError(t, DeleteUpdateTaskByUUID("uuid1")) |
||||||
|
sess := x.NewSession() |
||||||
|
defer sess.Close() |
||||||
|
has, err := sess.Get(&UpdateTask{UUID: "uuid1"}) |
||||||
|
assert.NoError(t, err) |
||||||
|
assert.False(t, has) |
||||||
|
|
||||||
|
assert.NoError(t, DeleteUpdateTaskByUUID("invalid")) |
||||||
|
} |
||||||
|
|
||||||
|
func TestCommitToPushCommit(t *testing.T) { |
||||||
|
now := time.Now() |
||||||
|
sig := &git.Signature{ |
||||||
|
Email: "example@example.com", |
||||||
|
Name: "John Doe", |
||||||
|
When: now, |
||||||
|
} |
||||||
|
const hexString = "0123456789abcdef0123456789abcdef01234567" |
||||||
|
sha1, err := git.NewIDFromString(hexString) |
||||||
|
assert.NoError(t, err) |
||||||
|
pushCommit := CommitToPushCommit(&git.Commit{ |
||||||
|
ID: sha1, |
||||||
|
Author: sig, |
||||||
|
Committer: sig, |
||||||
|
CommitMessage: "Commit Message", |
||||||
|
}) |
||||||
|
assert.Equal(t, hexString, pushCommit.Sha1) |
||||||
|
assert.Equal(t, "Commit Message", pushCommit.Message) |
||||||
|
assert.Equal(t, "example@example.com", pushCommit.AuthorEmail) |
||||||
|
assert.Equal(t, "John Doe", pushCommit.AuthorName) |
||||||
|
assert.Equal(t, "example@example.com", pushCommit.CommitterEmail) |
||||||
|
assert.Equal(t, "John Doe", pushCommit.CommitterName) |
||||||
|
assert.Equal(t, now, pushCommit.Timestamp) |
||||||
|
} |
||||||
|
|
||||||
|
func TestListToPushCommits(t *testing.T) { |
||||||
|
now := time.Now() |
||||||
|
sig := &git.Signature{ |
||||||
|
Email: "example@example.com", |
||||||
|
Name: "John Doe", |
||||||
|
When: now, |
||||||
|
} |
||||||
|
|
||||||
|
const hexString1 = "0123456789abcdef0123456789abcdef01234567" |
||||||
|
hash1, err := git.NewIDFromString(hexString1) |
||||||
|
assert.NoError(t, err) |
||||||
|
const hexString2 = "fedcba9876543210fedcba9876543210fedcba98" |
||||||
|
hash2, err := git.NewIDFromString(hexString2) |
||||||
|
assert.NoError(t, err) |
||||||
|
|
||||||
|
l := list.New() |
||||||
|
l.PushBack(&git.Commit{ |
||||||
|
ID: hash1, |
||||||
|
Author: sig, |
||||||
|
Committer: sig, |
||||||
|
CommitMessage: "Message1", |
||||||
|
}) |
||||||
|
l.PushBack(&git.Commit{ |
||||||
|
ID: hash2, |
||||||
|
Author: sig, |
||||||
|
Committer: sig, |
||||||
|
CommitMessage: "Message2", |
||||||
|
}) |
||||||
|
|
||||||
|
pushCommits := ListToPushCommits(l) |
||||||
|
assert.Equal(t, 2, pushCommits.Len) |
||||||
|
assert.Equal(t, 2, len(pushCommits.Commits)) |
||||||
|
|
||||||
|
assert.Equal(t, "Message1", pushCommits.Commits[0].Message) |
||||||
|
assert.Equal(t, hexString1, pushCommits.Commits[0].Sha1) |
||||||
|
assert.Equal(t, "example@example.com", pushCommits.Commits[0].AuthorEmail) |
||||||
|
assert.Equal(t, now, pushCommits.Commits[0].Timestamp) |
||||||
|
|
||||||
|
assert.Equal(t, "Message2", pushCommits.Commits[1].Message) |
||||||
|
assert.Equal(t, hexString2, pushCommits.Commits[1].Sha1) |
||||||
|
assert.Equal(t, "example@example.com", pushCommits.Commits[1].AuthorEmail) |
||||||
|
assert.Equal(t, now, pushCommits.Commits[1].Timestamp) |
||||||
|
} |
@ -0,0 +1,28 @@ |
|||||||
|
// Package require implements the same assertions as the `assert` package but
|
||||||
|
// stops test execution when a test fails.
|
||||||
|
//
|
||||||
|
// Example Usage
|
||||||
|
//
|
||||||
|
// The following is a complete example using require in a standard test function:
|
||||||
|
// import (
|
||||||
|
// "testing"
|
||||||
|
// "github.com/stretchr/testify/require"
|
||||||
|
// )
|
||||||
|
//
|
||||||
|
// func TestSomething(t *testing.T) {
|
||||||
|
//
|
||||||
|
// var a string = "Hello"
|
||||||
|
// var b string = "Hello"
|
||||||
|
//
|
||||||
|
// require.Equal(t, a, b, "The two words should be the same.")
|
||||||
|
//
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// Assertions
|
||||||
|
//
|
||||||
|
// The `require` package have same global functions as in the `assert` package,
|
||||||
|
// but instead of returning a boolean result they call `t.FailNow()`.
|
||||||
|
//
|
||||||
|
// Every assertion function also takes an optional string message as the final argument,
|
||||||
|
// allowing custom error messages to be appended to the message the assertion method outputs.
|
||||||
|
package require |
@ -0,0 +1,16 @@ |
|||||||
|
package require |
||||||
|
|
||||||
|
// Assertions provides assertion methods around the
|
||||||
|
// TestingT interface.
|
||||||
|
type Assertions struct { |
||||||
|
t TestingT |
||||||
|
} |
||||||
|
|
||||||
|
// New makes a new Assertions object for the specified TestingT.
|
||||||
|
func New(t TestingT) *Assertions { |
||||||
|
return &Assertions{ |
||||||
|
t: t, |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
//go:generate go run ../_codegen/main.go -output-package=require -template=require_forward.go.tmpl
|
@ -0,0 +1,429 @@ |
|||||||
|
/* |
||||||
|
* CODE GENERATED AUTOMATICALLY WITH github.com/stretchr/testify/_codegen |
||||||
|
* THIS FILE MUST NOT BE EDITED BY HAND |
||||||
|
*/ |
||||||
|
|
||||||
|
package require |
||||||
|
|
||||||
|
import ( |
||||||
|
assert "github.com/stretchr/testify/assert" |
||||||
|
http "net/http" |
||||||
|
url "net/url" |
||||||
|
time "time" |
||||||
|
) |
||||||
|
|
||||||
|
// Condition uses a Comparison to assert a complex condition.
|
||||||
|
func Condition(t TestingT, comp assert.Comparison, msgAndArgs ...interface{}) { |
||||||
|
if !assert.Condition(t, comp, msgAndArgs...) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// Contains asserts that the specified string, list(array, slice...) or map contains the
|
||||||
|
// specified substring or element.
|
||||||
|
//
|
||||||
|
// assert.Contains(t, "Hello World", "World", "But 'Hello World' does contain 'World'")
|
||||||
|
// assert.Contains(t, ["Hello", "World"], "World", "But ["Hello", "World"] does contain 'World'")
|
||||||
|
// assert.Contains(t, {"Hello": "World"}, "Hello", "But {'Hello': 'World'} does contain 'Hello'")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func Contains(t TestingT, s interface{}, contains interface{}, msgAndArgs ...interface{}) { |
||||||
|
if !assert.Contains(t, s, contains, msgAndArgs...) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// Empty asserts that the specified object is empty. I.e. nil, "", false, 0 or either
|
||||||
|
// a slice or a channel with len == 0.
|
||||||
|
//
|
||||||
|
// assert.Empty(t, obj)
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func Empty(t TestingT, object interface{}, msgAndArgs ...interface{}) { |
||||||
|
if !assert.Empty(t, object, msgAndArgs...) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// Equal asserts that two objects are equal.
|
||||||
|
//
|
||||||
|
// assert.Equal(t, 123, 123, "123 and 123 should be equal")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
//
|
||||||
|
// Pointer variable equality is determined based on the equality of the
|
||||||
|
// referenced values (as opposed to the memory addresses).
|
||||||
|
func Equal(t TestingT, expected interface{}, actual interface{}, msgAndArgs ...interface{}) { |
||||||
|
if !assert.Equal(t, expected, actual, msgAndArgs...) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// EqualError asserts that a function returned an error (i.e. not `nil`)
|
||||||
|
// and that it is equal to the provided error.
|
||||||
|
//
|
||||||
|
// actualObj, err := SomeFunction()
|
||||||
|
// assert.EqualError(t, err, expectedErrorString, "An error was expected")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func EqualError(t TestingT, theError error, errString string, msgAndArgs ...interface{}) { |
||||||
|
if !assert.EqualError(t, theError, errString, msgAndArgs...) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// EqualValues asserts that two objects are equal or convertable to the same types
|
||||||
|
// and equal.
|
||||||
|
//
|
||||||
|
// assert.EqualValues(t, uint32(123), int32(123), "123 and 123 should be equal")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func EqualValues(t TestingT, expected interface{}, actual interface{}, msgAndArgs ...interface{}) { |
||||||
|
if !assert.EqualValues(t, expected, actual, msgAndArgs...) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// Error asserts that a function returned an error (i.e. not `nil`).
|
||||||
|
//
|
||||||
|
// actualObj, err := SomeFunction()
|
||||||
|
// if assert.Error(t, err, "An error was expected") {
|
||||||
|
// assert.Equal(t, err, expectedError)
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func Error(t TestingT, err error, msgAndArgs ...interface{}) { |
||||||
|
if !assert.Error(t, err, msgAndArgs...) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// Exactly asserts that two objects are equal is value and type.
|
||||||
|
//
|
||||||
|
// assert.Exactly(t, int32(123), int64(123), "123 and 123 should NOT be equal")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func Exactly(t TestingT, expected interface{}, actual interface{}, msgAndArgs ...interface{}) { |
||||||
|
if !assert.Exactly(t, expected, actual, msgAndArgs...) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// Fail reports a failure through
|
||||||
|
func Fail(t TestingT, failureMessage string, msgAndArgs ...interface{}) { |
||||||
|
if !assert.Fail(t, failureMessage, msgAndArgs...) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// FailNow fails test
|
||||||
|
func FailNow(t TestingT, failureMessage string, msgAndArgs ...interface{}) { |
||||||
|
if !assert.FailNow(t, failureMessage, msgAndArgs...) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// False asserts that the specified value is false.
|
||||||
|
//
|
||||||
|
// assert.False(t, myBool, "myBool should be false")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func False(t TestingT, value bool, msgAndArgs ...interface{}) { |
||||||
|
if !assert.False(t, value, msgAndArgs...) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// HTTPBodyContains asserts that a specified handler returns a
|
||||||
|
// body that contains a string.
|
||||||
|
//
|
||||||
|
// assert.HTTPBodyContains(t, myHandler, "www.google.com", nil, "I'm Feeling Lucky")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func HTTPBodyContains(t TestingT, handler http.HandlerFunc, method string, url string, values url.Values, str interface{}) { |
||||||
|
if !assert.HTTPBodyContains(t, handler, method, url, values, str) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// HTTPBodyNotContains asserts that a specified handler returns a
|
||||||
|
// body that does not contain a string.
|
||||||
|
//
|
||||||
|
// assert.HTTPBodyNotContains(t, myHandler, "www.google.com", nil, "I'm Feeling Lucky")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func HTTPBodyNotContains(t TestingT, handler http.HandlerFunc, method string, url string, values url.Values, str interface{}) { |
||||||
|
if !assert.HTTPBodyNotContains(t, handler, method, url, values, str) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// HTTPError asserts that a specified handler returns an error status code.
|
||||||
|
//
|
||||||
|
// assert.HTTPError(t, myHandler, "POST", "/a/b/c", url.Values{"a": []string{"b", "c"}}
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func HTTPError(t TestingT, handler http.HandlerFunc, method string, url string, values url.Values) { |
||||||
|
if !assert.HTTPError(t, handler, method, url, values) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// HTTPRedirect asserts that a specified handler returns a redirect status code.
|
||||||
|
//
|
||||||
|
// assert.HTTPRedirect(t, myHandler, "GET", "/a/b/c", url.Values{"a": []string{"b", "c"}}
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func HTTPRedirect(t TestingT, handler http.HandlerFunc, method string, url string, values url.Values) { |
||||||
|
if !assert.HTTPRedirect(t, handler, method, url, values) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// HTTPSuccess asserts that a specified handler returns a success status code.
|
||||||
|
//
|
||||||
|
// assert.HTTPSuccess(t, myHandler, "POST", "http://www.google.com", nil)
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func HTTPSuccess(t TestingT, handler http.HandlerFunc, method string, url string, values url.Values) { |
||||||
|
if !assert.HTTPSuccess(t, handler, method, url, values) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// Implements asserts that an object is implemented by the specified interface.
|
||||||
|
//
|
||||||
|
// assert.Implements(t, (*MyInterface)(nil), new(MyObject), "MyObject")
|
||||||
|
func Implements(t TestingT, interfaceObject interface{}, object interface{}, msgAndArgs ...interface{}) { |
||||||
|
if !assert.Implements(t, interfaceObject, object, msgAndArgs...) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// InDelta asserts that the two numerals are within delta of each other.
|
||||||
|
//
|
||||||
|
// assert.InDelta(t, math.Pi, (22 / 7.0), 0.01)
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func InDelta(t TestingT, expected interface{}, actual interface{}, delta float64, msgAndArgs ...interface{}) { |
||||||
|
if !assert.InDelta(t, expected, actual, delta, msgAndArgs...) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// InDeltaSlice is the same as InDelta, except it compares two slices.
|
||||||
|
func InDeltaSlice(t TestingT, expected interface{}, actual interface{}, delta float64, msgAndArgs ...interface{}) { |
||||||
|
if !assert.InDeltaSlice(t, expected, actual, delta, msgAndArgs...) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// InEpsilon asserts that expected and actual have a relative error less than epsilon
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func InEpsilon(t TestingT, expected interface{}, actual interface{}, epsilon float64, msgAndArgs ...interface{}) { |
||||||
|
if !assert.InEpsilon(t, expected, actual, epsilon, msgAndArgs...) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// InEpsilonSlice is the same as InEpsilon, except it compares each value from two slices.
|
||||||
|
func InEpsilonSlice(t TestingT, expected interface{}, actual interface{}, epsilon float64, msgAndArgs ...interface{}) { |
||||||
|
if !assert.InEpsilonSlice(t, expected, actual, epsilon, msgAndArgs...) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// IsType asserts that the specified objects are of the same type.
|
||||||
|
func IsType(t TestingT, expectedType interface{}, object interface{}, msgAndArgs ...interface{}) { |
||||||
|
if !assert.IsType(t, expectedType, object, msgAndArgs...) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// JSONEq asserts that two JSON strings are equivalent.
|
||||||
|
//
|
||||||
|
// assert.JSONEq(t, `{"hello": "world", "foo": "bar"}`, `{"foo": "bar", "hello": "world"}`)
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func JSONEq(t TestingT, expected string, actual string, msgAndArgs ...interface{}) { |
||||||
|
if !assert.JSONEq(t, expected, actual, msgAndArgs...) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// Len asserts that the specified object has specific length.
|
||||||
|
// Len also fails if the object has a type that len() not accept.
|
||||||
|
//
|
||||||
|
// assert.Len(t, mySlice, 3, "The size of slice is not 3")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func Len(t TestingT, object interface{}, length int, msgAndArgs ...interface{}) { |
||||||
|
if !assert.Len(t, object, length, msgAndArgs...) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// Nil asserts that the specified object is nil.
|
||||||
|
//
|
||||||
|
// assert.Nil(t, err, "err should be nothing")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func Nil(t TestingT, object interface{}, msgAndArgs ...interface{}) { |
||||||
|
if !assert.Nil(t, object, msgAndArgs...) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// NoError asserts that a function returned no error (i.e. `nil`).
|
||||||
|
//
|
||||||
|
// actualObj, err := SomeFunction()
|
||||||
|
// if assert.NoError(t, err) {
|
||||||
|
// assert.Equal(t, actualObj, expectedObj)
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func NoError(t TestingT, err error, msgAndArgs ...interface{}) { |
||||||
|
if !assert.NoError(t, err, msgAndArgs...) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// NotContains asserts that the specified string, list(array, slice...) or map does NOT contain the
|
||||||
|
// specified substring or element.
|
||||||
|
//
|
||||||
|
// assert.NotContains(t, "Hello World", "Earth", "But 'Hello World' does NOT contain 'Earth'")
|
||||||
|
// assert.NotContains(t, ["Hello", "World"], "Earth", "But ['Hello', 'World'] does NOT contain 'Earth'")
|
||||||
|
// assert.NotContains(t, {"Hello": "World"}, "Earth", "But {'Hello': 'World'} does NOT contain 'Earth'")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func NotContains(t TestingT, s interface{}, contains interface{}, msgAndArgs ...interface{}) { |
||||||
|
if !assert.NotContains(t, s, contains, msgAndArgs...) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// NotEmpty asserts that the specified object is NOT empty. I.e. not nil, "", false, 0 or either
|
||||||
|
// a slice or a channel with len == 0.
|
||||||
|
//
|
||||||
|
// if assert.NotEmpty(t, obj) {
|
||||||
|
// assert.Equal(t, "two", obj[1])
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func NotEmpty(t TestingT, object interface{}, msgAndArgs ...interface{}) { |
||||||
|
if !assert.NotEmpty(t, object, msgAndArgs...) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// NotEqual asserts that the specified values are NOT equal.
|
||||||
|
//
|
||||||
|
// assert.NotEqual(t, obj1, obj2, "two objects shouldn't be equal")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
//
|
||||||
|
// Pointer variable equality is determined based on the equality of the
|
||||||
|
// referenced values (as opposed to the memory addresses).
|
||||||
|
func NotEqual(t TestingT, expected interface{}, actual interface{}, msgAndArgs ...interface{}) { |
||||||
|
if !assert.NotEqual(t, expected, actual, msgAndArgs...) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// NotNil asserts that the specified object is not nil.
|
||||||
|
//
|
||||||
|
// assert.NotNil(t, err, "err should be something")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func NotNil(t TestingT, object interface{}, msgAndArgs ...interface{}) { |
||||||
|
if !assert.NotNil(t, object, msgAndArgs...) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// NotPanics asserts that the code inside the specified PanicTestFunc does NOT panic.
|
||||||
|
//
|
||||||
|
// assert.NotPanics(t, func(){
|
||||||
|
// RemainCalm()
|
||||||
|
// }, "Calling RemainCalm() should NOT panic")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func NotPanics(t TestingT, f assert.PanicTestFunc, msgAndArgs ...interface{}) { |
||||||
|
if !assert.NotPanics(t, f, msgAndArgs...) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// NotRegexp asserts that a specified regexp does not match a string.
|
||||||
|
//
|
||||||
|
// assert.NotRegexp(t, regexp.MustCompile("starts"), "it's starting")
|
||||||
|
// assert.NotRegexp(t, "^start", "it's not starting")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func NotRegexp(t TestingT, rx interface{}, str interface{}, msgAndArgs ...interface{}) { |
||||||
|
if !assert.NotRegexp(t, rx, str, msgAndArgs...) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// NotZero asserts that i is not the zero value for its type and returns the truth.
|
||||||
|
func NotZero(t TestingT, i interface{}, msgAndArgs ...interface{}) { |
||||||
|
if !assert.NotZero(t, i, msgAndArgs...) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// Panics asserts that the code inside the specified PanicTestFunc panics.
|
||||||
|
//
|
||||||
|
// assert.Panics(t, func(){
|
||||||
|
// GoCrazy()
|
||||||
|
// }, "Calling GoCrazy() should panic")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func Panics(t TestingT, f assert.PanicTestFunc, msgAndArgs ...interface{}) { |
||||||
|
if !assert.Panics(t, f, msgAndArgs...) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// Regexp asserts that a specified regexp matches a string.
|
||||||
|
//
|
||||||
|
// assert.Regexp(t, regexp.MustCompile("start"), "it's starting")
|
||||||
|
// assert.Regexp(t, "start...$", "it's not starting")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func Regexp(t TestingT, rx interface{}, str interface{}, msgAndArgs ...interface{}) { |
||||||
|
if !assert.Regexp(t, rx, str, msgAndArgs...) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// True asserts that the specified value is true.
|
||||||
|
//
|
||||||
|
// assert.True(t, myBool, "myBool should be true")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func True(t TestingT, value bool, msgAndArgs ...interface{}) { |
||||||
|
if !assert.True(t, value, msgAndArgs...) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// WithinDuration asserts that the two times are within duration delta of each other.
|
||||||
|
//
|
||||||
|
// assert.WithinDuration(t, time.Now(), time.Now(), 10*time.Second, "The difference should not be more than 10s")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func WithinDuration(t TestingT, expected time.Time, actual time.Time, delta time.Duration, msgAndArgs ...interface{}) { |
||||||
|
if !assert.WithinDuration(t, expected, actual, delta, msgAndArgs...) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// Zero asserts that i is the zero value for its type and returns the truth.
|
||||||
|
func Zero(t TestingT, i interface{}, msgAndArgs ...interface{}) { |
||||||
|
if !assert.Zero(t, i, msgAndArgs...) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
@ -0,0 +1,6 @@ |
|||||||
|
{{.Comment}} |
||||||
|
func {{.DocInfo.Name}}(t TestingT, {{.Params}}) { |
||||||
|
if !assert.{{.DocInfo.Name}}(t, {{.ForwardedParams}}) { |
||||||
|
t.FailNow() |
||||||
|
} |
||||||
|
} |
@ -0,0 +1,353 @@ |
|||||||
|
/* |
||||||
|
* CODE GENERATED AUTOMATICALLY WITH github.com/stretchr/testify/_codegen |
||||||
|
* THIS FILE MUST NOT BE EDITED BY HAND |
||||||
|
*/ |
||||||
|
|
||||||
|
package require |
||||||
|
|
||||||
|
import ( |
||||||
|
assert "github.com/stretchr/testify/assert" |
||||||
|
http "net/http" |
||||||
|
url "net/url" |
||||||
|
time "time" |
||||||
|
) |
||||||
|
|
||||||
|
// Condition uses a Comparison to assert a complex condition.
|
||||||
|
func (a *Assertions) Condition(comp assert.Comparison, msgAndArgs ...interface{}) { |
||||||
|
Condition(a.t, comp, msgAndArgs...) |
||||||
|
} |
||||||
|
|
||||||
|
// Contains asserts that the specified string, list(array, slice...) or map contains the
|
||||||
|
// specified substring or element.
|
||||||
|
//
|
||||||
|
// a.Contains("Hello World", "World", "But 'Hello World' does contain 'World'")
|
||||||
|
// a.Contains(["Hello", "World"], "World", "But ["Hello", "World"] does contain 'World'")
|
||||||
|
// a.Contains({"Hello": "World"}, "Hello", "But {'Hello': 'World'} does contain 'Hello'")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func (a *Assertions) Contains(s interface{}, contains interface{}, msgAndArgs ...interface{}) { |
||||||
|
Contains(a.t, s, contains, msgAndArgs...) |
||||||
|
} |
||||||
|
|
||||||
|
// Empty asserts that the specified object is empty. I.e. nil, "", false, 0 or either
|
||||||
|
// a slice or a channel with len == 0.
|
||||||
|
//
|
||||||
|
// a.Empty(obj)
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func (a *Assertions) Empty(object interface{}, msgAndArgs ...interface{}) { |
||||||
|
Empty(a.t, object, msgAndArgs...) |
||||||
|
} |
||||||
|
|
||||||
|
// Equal asserts that two objects are equal.
|
||||||
|
//
|
||||||
|
// a.Equal(123, 123, "123 and 123 should be equal")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
//
|
||||||
|
// Pointer variable equality is determined based on the equality of the
|
||||||
|
// referenced values (as opposed to the memory addresses).
|
||||||
|
func (a *Assertions) Equal(expected interface{}, actual interface{}, msgAndArgs ...interface{}) { |
||||||
|
Equal(a.t, expected, actual, msgAndArgs...) |
||||||
|
} |
||||||
|
|
||||||
|
// EqualError asserts that a function returned an error (i.e. not `nil`)
|
||||||
|
// and that it is equal to the provided error.
|
||||||
|
//
|
||||||
|
// actualObj, err := SomeFunction()
|
||||||
|
// a.EqualError(err, expectedErrorString, "An error was expected")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func (a *Assertions) EqualError(theError error, errString string, msgAndArgs ...interface{}) { |
||||||
|
EqualError(a.t, theError, errString, msgAndArgs...) |
||||||
|
} |
||||||
|
|
||||||
|
// EqualValues asserts that two objects are equal or convertable to the same types
|
||||||
|
// and equal.
|
||||||
|
//
|
||||||
|
// a.EqualValues(uint32(123), int32(123), "123 and 123 should be equal")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func (a *Assertions) EqualValues(expected interface{}, actual interface{}, msgAndArgs ...interface{}) { |
||||||
|
EqualValues(a.t, expected, actual, msgAndArgs...) |
||||||
|
} |
||||||
|
|
||||||
|
// Error asserts that a function returned an error (i.e. not `nil`).
|
||||||
|
//
|
||||||
|
// actualObj, err := SomeFunction()
|
||||||
|
// if a.Error(err, "An error was expected") {
|
||||||
|
// assert.Equal(t, err, expectedError)
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func (a *Assertions) Error(err error, msgAndArgs ...interface{}) { |
||||||
|
Error(a.t, err, msgAndArgs...) |
||||||
|
} |
||||||
|
|
||||||
|
// Exactly asserts that two objects are equal is value and type.
|
||||||
|
//
|
||||||
|
// a.Exactly(int32(123), int64(123), "123 and 123 should NOT be equal")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func (a *Assertions) Exactly(expected interface{}, actual interface{}, msgAndArgs ...interface{}) { |
||||||
|
Exactly(a.t, expected, actual, msgAndArgs...) |
||||||
|
} |
||||||
|
|
||||||
|
// Fail reports a failure through
|
||||||
|
func (a *Assertions) Fail(failureMessage string, msgAndArgs ...interface{}) { |
||||||
|
Fail(a.t, failureMessage, msgAndArgs...) |
||||||
|
} |
||||||
|
|
||||||
|
// FailNow fails test
|
||||||
|
func (a *Assertions) FailNow(failureMessage string, msgAndArgs ...interface{}) { |
||||||
|
FailNow(a.t, failureMessage, msgAndArgs...) |
||||||
|
} |
||||||
|
|
||||||
|
// False asserts that the specified value is false.
|
||||||
|
//
|
||||||
|
// a.False(myBool, "myBool should be false")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func (a *Assertions) False(value bool, msgAndArgs ...interface{}) { |
||||||
|
False(a.t, value, msgAndArgs...) |
||||||
|
} |
||||||
|
|
||||||
|
// HTTPBodyContains asserts that a specified handler returns a
|
||||||
|
// body that contains a string.
|
||||||
|
//
|
||||||
|
// a.HTTPBodyContains(myHandler, "www.google.com", nil, "I'm Feeling Lucky")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func (a *Assertions) HTTPBodyContains(handler http.HandlerFunc, method string, url string, values url.Values, str interface{}) { |
||||||
|
HTTPBodyContains(a.t, handler, method, url, values, str) |
||||||
|
} |
||||||
|
|
||||||
|
// HTTPBodyNotContains asserts that a specified handler returns a
|
||||||
|
// body that does not contain a string.
|
||||||
|
//
|
||||||
|
// a.HTTPBodyNotContains(myHandler, "www.google.com", nil, "I'm Feeling Lucky")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func (a *Assertions) HTTPBodyNotContains(handler http.HandlerFunc, method string, url string, values url.Values, str interface{}) { |
||||||
|
HTTPBodyNotContains(a.t, handler, method, url, values, str) |
||||||
|
} |
||||||
|
|
||||||
|
// HTTPError asserts that a specified handler returns an error status code.
|
||||||
|
//
|
||||||
|
// a.HTTPError(myHandler, "POST", "/a/b/c", url.Values{"a": []string{"b", "c"}}
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func (a *Assertions) HTTPError(handler http.HandlerFunc, method string, url string, values url.Values) { |
||||||
|
HTTPError(a.t, handler, method, url, values) |
||||||
|
} |
||||||
|
|
||||||
|
// HTTPRedirect asserts that a specified handler returns a redirect status code.
|
||||||
|
//
|
||||||
|
// a.HTTPRedirect(myHandler, "GET", "/a/b/c", url.Values{"a": []string{"b", "c"}}
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func (a *Assertions) HTTPRedirect(handler http.HandlerFunc, method string, url string, values url.Values) { |
||||||
|
HTTPRedirect(a.t, handler, method, url, values) |
||||||
|
} |
||||||
|
|
||||||
|
// HTTPSuccess asserts that a specified handler returns a success status code.
|
||||||
|
//
|
||||||
|
// a.HTTPSuccess(myHandler, "POST", "http://www.google.com", nil)
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func (a *Assertions) HTTPSuccess(handler http.HandlerFunc, method string, url string, values url.Values) { |
||||||
|
HTTPSuccess(a.t, handler, method, url, values) |
||||||
|
} |
||||||
|
|
||||||
|
// Implements asserts that an object is implemented by the specified interface.
|
||||||
|
//
|
||||||
|
// a.Implements((*MyInterface)(nil), new(MyObject), "MyObject")
|
||||||
|
func (a *Assertions) Implements(interfaceObject interface{}, object interface{}, msgAndArgs ...interface{}) { |
||||||
|
Implements(a.t, interfaceObject, object, msgAndArgs...) |
||||||
|
} |
||||||
|
|
||||||
|
// InDelta asserts that the two numerals are within delta of each other.
|
||||||
|
//
|
||||||
|
// a.InDelta(math.Pi, (22 / 7.0), 0.01)
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func (a *Assertions) InDelta(expected interface{}, actual interface{}, delta float64, msgAndArgs ...interface{}) { |
||||||
|
InDelta(a.t, expected, actual, delta, msgAndArgs...) |
||||||
|
} |
||||||
|
|
||||||
|
// InDeltaSlice is the same as InDelta, except it compares two slices.
|
||||||
|
func (a *Assertions) InDeltaSlice(expected interface{}, actual interface{}, delta float64, msgAndArgs ...interface{}) { |
||||||
|
InDeltaSlice(a.t, expected, actual, delta, msgAndArgs...) |
||||||
|
} |
||||||
|
|
||||||
|
// InEpsilon asserts that expected and actual have a relative error less than epsilon
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func (a *Assertions) InEpsilon(expected interface{}, actual interface{}, epsilon float64, msgAndArgs ...interface{}) { |
||||||
|
InEpsilon(a.t, expected, actual, epsilon, msgAndArgs...) |
||||||
|
} |
||||||
|
|
||||||
|
// InEpsilonSlice is the same as InEpsilon, except it compares each value from two slices.
|
||||||
|
func (a *Assertions) InEpsilonSlice(expected interface{}, actual interface{}, epsilon float64, msgAndArgs ...interface{}) { |
||||||
|
InEpsilonSlice(a.t, expected, actual, epsilon, msgAndArgs...) |
||||||
|
} |
||||||
|
|
||||||
|
// IsType asserts that the specified objects are of the same type.
|
||||||
|
func (a *Assertions) IsType(expectedType interface{}, object interface{}, msgAndArgs ...interface{}) { |
||||||
|
IsType(a.t, expectedType, object, msgAndArgs...) |
||||||
|
} |
||||||
|
|
||||||
|
// JSONEq asserts that two JSON strings are equivalent.
|
||||||
|
//
|
||||||
|
// a.JSONEq(`{"hello": "world", "foo": "bar"}`, `{"foo": "bar", "hello": "world"}`)
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func (a *Assertions) JSONEq(expected string, actual string, msgAndArgs ...interface{}) { |
||||||
|
JSONEq(a.t, expected, actual, msgAndArgs...) |
||||||
|
} |
||||||
|
|
||||||
|
// Len asserts that the specified object has specific length.
|
||||||
|
// Len also fails if the object has a type that len() not accept.
|
||||||
|
//
|
||||||
|
// a.Len(mySlice, 3, "The size of slice is not 3")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func (a *Assertions) Len(object interface{}, length int, msgAndArgs ...interface{}) { |
||||||
|
Len(a.t, object, length, msgAndArgs...) |
||||||
|
} |
||||||
|
|
||||||
|
// Nil asserts that the specified object is nil.
|
||||||
|
//
|
||||||
|
// a.Nil(err, "err should be nothing")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func (a *Assertions) Nil(object interface{}, msgAndArgs ...interface{}) { |
||||||
|
Nil(a.t, object, msgAndArgs...) |
||||||
|
} |
||||||
|
|
||||||
|
// NoError asserts that a function returned no error (i.e. `nil`).
|
||||||
|
//
|
||||||
|
// actualObj, err := SomeFunction()
|
||||||
|
// if a.NoError(err) {
|
||||||
|
// assert.Equal(t, actualObj, expectedObj)
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func (a *Assertions) NoError(err error, msgAndArgs ...interface{}) { |
||||||
|
NoError(a.t, err, msgAndArgs...) |
||||||
|
} |
||||||
|
|
||||||
|
// NotContains asserts that the specified string, list(array, slice...) or map does NOT contain the
|
||||||
|
// specified substring or element.
|
||||||
|
//
|
||||||
|
// a.NotContains("Hello World", "Earth", "But 'Hello World' does NOT contain 'Earth'")
|
||||||
|
// a.NotContains(["Hello", "World"], "Earth", "But ['Hello', 'World'] does NOT contain 'Earth'")
|
||||||
|
// a.NotContains({"Hello": "World"}, "Earth", "But {'Hello': 'World'} does NOT contain 'Earth'")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func (a *Assertions) NotContains(s interface{}, contains interface{}, msgAndArgs ...interface{}) { |
||||||
|
NotContains(a.t, s, contains, msgAndArgs...) |
||||||
|
} |
||||||
|
|
||||||
|
// NotEmpty asserts that the specified object is NOT empty. I.e. not nil, "", false, 0 or either
|
||||||
|
// a slice or a channel with len == 0.
|
||||||
|
//
|
||||||
|
// if a.NotEmpty(obj) {
|
||||||
|
// assert.Equal(t, "two", obj[1])
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func (a *Assertions) NotEmpty(object interface{}, msgAndArgs ...interface{}) { |
||||||
|
NotEmpty(a.t, object, msgAndArgs...) |
||||||
|
} |
||||||
|
|
||||||
|
// NotEqual asserts that the specified values are NOT equal.
|
||||||
|
//
|
||||||
|
// a.NotEqual(obj1, obj2, "two objects shouldn't be equal")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
//
|
||||||
|
// Pointer variable equality is determined based on the equality of the
|
||||||
|
// referenced values (as opposed to the memory addresses).
|
||||||
|
func (a *Assertions) NotEqual(expected interface{}, actual interface{}, msgAndArgs ...interface{}) { |
||||||
|
NotEqual(a.t, expected, actual, msgAndArgs...) |
||||||
|
} |
||||||
|
|
||||||
|
// NotNil asserts that the specified object is not nil.
|
||||||
|
//
|
||||||
|
// a.NotNil(err, "err should be something")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func (a *Assertions) NotNil(object interface{}, msgAndArgs ...interface{}) { |
||||||
|
NotNil(a.t, object, msgAndArgs...) |
||||||
|
} |
||||||
|
|
||||||
|
// NotPanics asserts that the code inside the specified PanicTestFunc does NOT panic.
|
||||||
|
//
|
||||||
|
// a.NotPanics(func(){
|
||||||
|
// RemainCalm()
|
||||||
|
// }, "Calling RemainCalm() should NOT panic")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func (a *Assertions) NotPanics(f assert.PanicTestFunc, msgAndArgs ...interface{}) { |
||||||
|
NotPanics(a.t, f, msgAndArgs...) |
||||||
|
} |
||||||
|
|
||||||
|
// NotRegexp asserts that a specified regexp does not match a string.
|
||||||
|
//
|
||||||
|
// a.NotRegexp(regexp.MustCompile("starts"), "it's starting")
|
||||||
|
// a.NotRegexp("^start", "it's not starting")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func (a *Assertions) NotRegexp(rx interface{}, str interface{}, msgAndArgs ...interface{}) { |
||||||
|
NotRegexp(a.t, rx, str, msgAndArgs...) |
||||||
|
} |
||||||
|
|
||||||
|
// NotZero asserts that i is not the zero value for its type and returns the truth.
|
||||||
|
func (a *Assertions) NotZero(i interface{}, msgAndArgs ...interface{}) { |
||||||
|
NotZero(a.t, i, msgAndArgs...) |
||||||
|
} |
||||||
|
|
||||||
|
// Panics asserts that the code inside the specified PanicTestFunc panics.
|
||||||
|
//
|
||||||
|
// a.Panics(func(){
|
||||||
|
// GoCrazy()
|
||||||
|
// }, "Calling GoCrazy() should panic")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func (a *Assertions) Panics(f assert.PanicTestFunc, msgAndArgs ...interface{}) { |
||||||
|
Panics(a.t, f, msgAndArgs...) |
||||||
|
} |
||||||
|
|
||||||
|
// Regexp asserts that a specified regexp matches a string.
|
||||||
|
//
|
||||||
|
// a.Regexp(regexp.MustCompile("start"), "it's starting")
|
||||||
|
// a.Regexp("start...$", "it's not starting")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func (a *Assertions) Regexp(rx interface{}, str interface{}, msgAndArgs ...interface{}) { |
||||||
|
Regexp(a.t, rx, str, msgAndArgs...) |
||||||
|
} |
||||||
|
|
||||||
|
// True asserts that the specified value is true.
|
||||||
|
//
|
||||||
|
// a.True(myBool, "myBool should be true")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func (a *Assertions) True(value bool, msgAndArgs ...interface{}) { |
||||||
|
True(a.t, value, msgAndArgs...) |
||||||
|
} |
||||||
|
|
||||||
|
// WithinDuration asserts that the two times are within duration delta of each other.
|
||||||
|
//
|
||||||
|
// a.WithinDuration(time.Now(), time.Now(), 10*time.Second, "The difference should not be more than 10s")
|
||||||
|
//
|
||||||
|
// Returns whether the assertion was successful (true) or not (false).
|
||||||
|
func (a *Assertions) WithinDuration(expected time.Time, actual time.Time, delta time.Duration, msgAndArgs ...interface{}) { |
||||||
|
WithinDuration(a.t, expected, actual, delta, msgAndArgs...) |
||||||
|
} |
||||||
|
|
||||||
|
// Zero asserts that i is the zero value for its type and returns the truth.
|
||||||
|
func (a *Assertions) Zero(i interface{}, msgAndArgs ...interface{}) { |
||||||
|
Zero(a.t, i, msgAndArgs...) |
||||||
|
} |
@ -0,0 +1,4 @@ |
|||||||
|
{{.CommentWithoutT "a"}} |
||||||
|
func (a *Assertions) {{.DocInfo.Name}}({{.Params}}) { |
||||||
|
{{.DocInfo.Name}}(a.t, {{.ForwardedParams}}) |
||||||
|
} |
@ -0,0 +1,9 @@ |
|||||||
|
package require |
||||||
|
|
||||||
|
// TestingT is an interface wrapper around *testing.T
|
||||||
|
type TestingT interface { |
||||||
|
Errorf(format string, args ...interface{}) |
||||||
|
FailNow() |
||||||
|
} |
||||||
|
|
||||||
|
//go:generate go run ../_codegen/main.go -output-package=require -template=require.go.tmpl
|
@ -0,0 +1,21 @@ |
|||||||
|
The MIT License (MIT) |
||||||
|
|
||||||
|
Copyright (c) 2016 Andrey Nering |
||||||
|
|
||||||
|
Permission is hereby granted, free of charge, to any person obtaining a copy |
||||||
|
of this software and associated documentation files (the "Software"), to deal |
||||||
|
in the Software without restriction, including without limitation the rights |
||||||
|
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell |
||||||
|
copies of the Software, and to permit persons to whom the Software is |
||||||
|
furnished to do so, subject to the following conditions: |
||||||
|
|
||||||
|
The above copyright notice and this permission notice shall be included in all |
||||||
|
copies or substantial portions of the Software. |
||||||
|
|
||||||
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
||||||
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
||||||
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE |
||||||
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
||||||
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
||||||
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE |
||||||
|
SOFTWARE. |
@ -0,0 +1,323 @@ |
|||||||
|
# Go Test Fixtures |
||||||
|
|
||||||
|
[![license](https://img.shields.io/github/license/mashape/apistatus.svg?maxAge=2592000)](https://github.com/go-testfixtures/testfixtures/blob/master/LICENSE) |
||||||
|
[![Join the chat at https://gitter.im/go-testfixtures/testfixtures](https://badges.gitter.im/go-testfixtures/testfixtures.svg)](https://gitter.im/go-testfixtures/testfixtures?utm_source=badge&utm_medium=badge&utm_campaign=pr-badge&utm_content=badge) |
||||||
|
[![GoDoc](https://godoc.org/gopkg.in/testfixtures.v1?status.svg)](https://godoc.org/gopkg.in/testfixtures.v1) |
||||||
|
[![Build Status](https://travis-ci.org/go-testfixtures/testfixtures.svg?branch=master)](https://travis-ci.org/go-testfixtures/testfixtures) |
||||||
|
[![Go Report Card](https://goreportcard.com/badge/github.com/go-testfixtures/testfixtures)](https://goreportcard.com/report/github.com/go-testfixtures/testfixtures) |
||||||
|
|
||||||
|
> ***Warning***: this package will wipe the database data before loading the |
||||||
|
fixtures! It is supposed to be used on a test database. Please, double check |
||||||
|
if you are running it against the correct database. |
||||||
|
|
||||||
|
Writing tests is hard, even more when you have to deal with an SQL database. |
||||||
|
This package aims to make writing functional tests for web apps written in |
||||||
|
Go easier. |
||||||
|
|
||||||
|
Basically this package mimics the ["Rails' way"][railstests] of writing tests |
||||||
|
for database applications, where sample data is kept in fixtures files. Before |
||||||
|
the execution of every test, the test database is cleaned and the fixture data |
||||||
|
is loaded into the database. |
||||||
|
|
||||||
|
The idea is running tests against a real database, instead of relying in mocks, |
||||||
|
which is boring to setup and may lead to production bugs not to being catch in |
||||||
|
the tests. |
||||||
|
|
||||||
|
## Installation |
||||||
|
|
||||||
|
First, get it: |
||||||
|
|
||||||
|
```bash |
||||||
|
go get -u gopkg.in/testfixtures.v2 |
||||||
|
``` |
||||||
|
|
||||||
|
## Usage |
||||||
|
|
||||||
|
Create a folder for the fixture files. Each file should contain data for a |
||||||
|
single table and have the name `<table-name>.yml`: |
||||||
|
|
||||||
|
```yml |
||||||
|
myapp |
||||||
|
- myapp.go |
||||||
|
- myapp_test.go |
||||||
|
- ... |
||||||
|
- fixtures: |
||||||
|
- posts.yml |
||||||
|
- comments.yml |
||||||
|
- tags.yml |
||||||
|
- posts_tags.yml |
||||||
|
- ... |
||||||
|
``` |
||||||
|
|
||||||
|
The file would look like this (it can have as many record you want): |
||||||
|
|
||||||
|
```yml |
||||||
|
# comments.yml |
||||||
|
- |
||||||
|
id: 1 |
||||||
|
post_id: 1 |
||||||
|
content: This post is awesome! |
||||||
|
author_name: John Doe |
||||||
|
author_email: john@doe.com |
||||||
|
created_at: 2016-01-01 12:30:12 |
||||||
|
updated_at: 2016-01-01 12:30:12 |
||||||
|
|
||||||
|
- |
||||||
|
id: 2 |
||||||
|
post_id: 2 |
||||||
|
content: Are you kidding me? |
||||||
|
author_name: John Doe |
||||||
|
author_email: john@doe.com |
||||||
|
created_at: 2016-01-01 12:30:12 |
||||||
|
updated_at: 2016-01-01 12:30:12 |
||||||
|
|
||||||
|
# ... |
||||||
|
``` |
||||||
|
|
||||||
|
Your tests would look like this: |
||||||
|
|
||||||
|
```go |
||||||
|
package myapp |
||||||
|
|
||||||
|
import ( |
||||||
|
"database/sql" |
||||||
|
"log" |
||||||
|
|
||||||
|
_ "github.com/lib/pq" |
||||||
|
"gopkg.in/testfixtures.v2" |
||||||
|
) |
||||||
|
|
||||||
|
var ( |
||||||
|
db *sql.DB |
||||||
|
fixtures *testfixtures.Context |
||||||
|
) |
||||||
|
|
||||||
|
func TestMain(m *testing.M) { |
||||||
|
var err error |
||||||
|
|
||||||
|
// Open connection with the test database. |
||||||
|
// Do NOT import fixtures in a production database! |
||||||
|
// Existing data would be deleted |
||||||
|
db, err = sql.Open("postgres", "dbname=myapp_test") |
||||||
|
if err != nil { |
||||||
|
log.Fatal(err) |
||||||
|
} |
||||||
|
|
||||||
|
// creating the context that hold the fixtures |
||||||
|
// see about all compatible databases in this page below |
||||||
|
c, err = testfixtures.NewFolder(db, &testfixtures.PostgreSQL{}, "testdata/fixtures") |
||||||
|
if err != nil { |
||||||
|
log.Fatal(err) |
||||||
|
} |
||||||
|
|
||||||
|
os.Exit(m.Run()) |
||||||
|
} |
||||||
|
|
||||||
|
func prepareTestDatabase() { |
||||||
|
if err := fixtures.Load(); err != nil { |
||||||
|
log.Fatal(err) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
func TestX(t *testing.T) { |
||||||
|
prepareTestDatabase() |
||||||
|
// your test here ... |
||||||
|
} |
||||||
|
|
||||||
|
func TestY(t *testing.T) { |
||||||
|
prepareTestDatabase() |
||||||
|
// your test here ... |
||||||
|
} |
||||||
|
|
||||||
|
func TestZ(t *testing.T) { |
||||||
|
prepareTestDatabase() |
||||||
|
// your test here ... |
||||||
|
} |
||||||
|
``` |
||||||
|
|
||||||
|
Alternatively, you can use the `NewFiles` function, to specify which |
||||||
|
files you want to load into the database: |
||||||
|
|
||||||
|
```go |
||||||
|
fixtures, err := testfixtures.NewFiles(db, &testfixtures.PostgreSQL{}, |
||||||
|
"fixtures/orders.yml", |
||||||
|
"fixtures/customers.yml", |
||||||
|
// add as many files you want |
||||||
|
) |
||||||
|
if err != nil { |
||||||
|
log.Fatal(err) |
||||||
|
} |
||||||
|
``` |
||||||
|
|
||||||
|
## Security check |
||||||
|
|
||||||
|
In order to prevent you from accidentally wiping the wrong database, this |
||||||
|
package will refuse to load fixtures if the database name (or database |
||||||
|
filename for SQLite) doesn't contains "test". If you want to disable this |
||||||
|
check, use: |
||||||
|
|
||||||
|
```go |
||||||
|
testfixtures.SkipDatabaseNameCheck(true) |
||||||
|
``` |
||||||
|
|
||||||
|
## Sequences |
||||||
|
|
||||||
|
For PostgreSQL or Oracle, this package also resets all sequences to a high |
||||||
|
number to prevent duplicated primary keys while running the tests. |
||||||
|
The default is 10000, but you can change that with: |
||||||
|
|
||||||
|
```go |
||||||
|
testfixtures.ResetSequencesTo(10000) |
||||||
|
``` |
||||||
|
|
||||||
|
## Compatible databases |
||||||
|
|
||||||
|
### PostgreSQL |
||||||
|
|
||||||
|
This package has two approaches to disable foreign keys while importing fixtures |
||||||
|
in PostgreSQL databases: |
||||||
|
|
||||||
|
#### With `DISABLE TRIGGER` |
||||||
|
|
||||||
|
This is the default approach. For that use: |
||||||
|
|
||||||
|
```go |
||||||
|
&testfixtures.PostgreSQL{} |
||||||
|
``` |
||||||
|
|
||||||
|
With the above snippet this package will use `DISABLE TRIGGER` to temporarily |
||||||
|
disabling foreign key constraints while loading fixtures. This work with any |
||||||
|
version of PostgreSQL, but it is **required** to be connected in the database |
||||||
|
as a SUPERUSER. You can make a PostgreSQL user a SUPERUSER with: |
||||||
|
|
||||||
|
```sql |
||||||
|
ALTER USER your_user SUPERUSER; |
||||||
|
``` |
||||||
|
|
||||||
|
#### With `ALTER CONSTRAINT` |
||||||
|
|
||||||
|
This approach don't require to be connected as a SUPERUSER, but only work with |
||||||
|
PostgreSQL versions >= 9.4. Try this if you are getting foreign key violation |
||||||
|
errors with the previous approach. It is as simple as using: |
||||||
|
|
||||||
|
```go |
||||||
|
&testfixtures.PostgreSQL{UseAlterConstraint: true} |
||||||
|
``` |
||||||
|
|
||||||
|
### MySQL |
||||||
|
|
||||||
|
Just make sure the connection string have |
||||||
|
[the multistatement parameter](https://github.com/go-sql-driver/mysql#multistatements) |
||||||
|
set to true, and use: |
||||||
|
|
||||||
|
```go |
||||||
|
&testfixtures.MySQL{} |
||||||
|
``` |
||||||
|
|
||||||
|
### SQLite |
||||||
|
|
||||||
|
SQLite is also supported. It is recommended to create foreign keys as |
||||||
|
`DEFERRABLE` (the default) to prevent problems. See more |
||||||
|
[on the SQLite documentation](https://www.sqlite.org/foreignkeys.html#fk_deferred). |
||||||
|
(Foreign key constraints are no-op by default on SQLite, but enabling it is |
||||||
|
recommended). |
||||||
|
|
||||||
|
```go |
||||||
|
&testfixtures.SQLite{} |
||||||
|
``` |
||||||
|
|
||||||
|
### Microsoft SQL Server |
||||||
|
|
||||||
|
SQL Server support requires SQL Server >= 2008. Inserting on `IDENTITY` columns |
||||||
|
are handled as well. Just make sure you are logged in with a user with |
||||||
|
`ALTER TABLE` permission. |
||||||
|
|
||||||
|
```go |
||||||
|
&testfixtures.SQLServer{} |
||||||
|
``` |
||||||
|
|
||||||
|
### Oracle |
||||||
|
|
||||||
|
Oracle is supported as well. Use: |
||||||
|
|
||||||
|
```go |
||||||
|
&testfixtures.Oracle{} |
||||||
|
``` |
||||||
|
|
||||||
|
## Contributing |
||||||
|
|
||||||
|
Tests were written to ensure everything work as expected. You can run the tests |
||||||
|
with: |
||||||
|
|
||||||
|
```bash |
||||||
|
# running tests for PostgreSQL |
||||||
|
go test -tags postgresql |
||||||
|
|
||||||
|
# running test for MySQL |
||||||
|
go test -tags mysql |
||||||
|
|
||||||
|
# running tests for SQLite |
||||||
|
go test -tags sqlite |
||||||
|
|
||||||
|
# running tests for SQL Server |
||||||
|
go test -tags sqlserver |
||||||
|
|
||||||
|
# running tests for Oracle |
||||||
|
go test -tags oracle |
||||||
|
|
||||||
|
# running test for multiple databases at once |
||||||
|
go test -tags 'sqlite postgresql mysql' |
||||||
|
|
||||||
|
# running tests + benchmark |
||||||
|
go test -v -bench=. -tags postgresql |
||||||
|
``` |
||||||
|
|
||||||
|
Travis runs tests for PostgreSQL, MySQL and SQLite. |
||||||
|
|
||||||
|
To set the connection string of tests for each database, edit the `.env` |
||||||
|
file, but do not include the changes a in pull request. |
||||||
|
|
||||||
|
## Changes in v2 |
||||||
|
|
||||||
|
A context was created to allow cache of some SQL statements. See in the |
||||||
|
documentation above how to use it. |
||||||
|
|
||||||
|
The helpers were renamed to have a smaller name: |
||||||
|
|
||||||
|
```go |
||||||
|
PostgreSQLHelper{} -> PostgreSQL{} |
||||||
|
MySQLHelper{} -> MySQL{} |
||||||
|
SQLiteHelper{} -> SQLite{} |
||||||
|
SQLServerHelper{} -> SQLServer{} |
||||||
|
OracleHelper{} -> Oracle{} |
||||||
|
``` |
||||||
|
|
||||||
|
The old functions and helpers are still available for backward compatibility. |
||||||
|
See the file [deprecated.go](https://github.com/go-testfixtures/testfixtures/blob/master/LICENSE) |
||||||
|
|
||||||
|
## Alternatives |
||||||
|
|
||||||
|
If you don't think using fixtures is a good idea, you can try one of these |
||||||
|
packages instead: |
||||||
|
|
||||||
|
- [factory-go][factorygo]: Factory for Go. Inspired by Python's Factory Boy |
||||||
|
and Ruby's Factory Girl |
||||||
|
- [go-txdb (Single transaction SQL driver for Go)][gotxdb]: Use a single |
||||||
|
database transaction for each functional test, so you can rollback to |
||||||
|
previous state between tests to have the same database state in all tests |
||||||
|
- [go-sqlmock][gosqlmock]: A mock for the sql.DB interface. This allow you to unit |
||||||
|
test database code without having to connect to a real database |
||||||
|
|
||||||
|
There's also these other implementations of test fixtures for Go: |
||||||
|
|
||||||
|
- [go-fixtures][gofixtures]: Django style fixtures for Go |
||||||
|
- [mongofixtures][mongofixtures]: Fixtures for MongoDB |
||||||
|
- [fixturer][fixturer]: Another fixture loader supporting MySQL |
||||||
|
|
||||||
|
[railstests]: http://guides.rubyonrails.org/testing.html#the-test-database |
||||||
|
[gotxdb]: https://github.com/DATA-DOG/go-txdb |
||||||
|
[gosqlmock]: https://github.com/DATA-DOG/go-sqlmock |
||||||
|
[gofixtures]: https://github.com/AreaHQ/go-fixtures |
||||||
|
[mongofixtures]: https://github.com/OwlyCode/mongofixtures |
||||||
|
[fixturer]: https://github.com/44hapa/fixturer |
||||||
|
[factorygo]: https://github.com/bluele/factory-go |
@ -0,0 +1,59 @@ |
|||||||
|
package testfixtures |
||||||
|
|
||||||
|
import ( |
||||||
|
"database/sql" |
||||||
|
) |
||||||
|
|
||||||
|
type ( |
||||||
|
DataBaseHelper Helper // Deprecated: Use Helper instead
|
||||||
|
|
||||||
|
PostgreSQLHelper struct { // Deprecated: Use PostgreSQL{} instead
|
||||||
|
PostgreSQL |
||||||
|
UseAlterConstraint bool |
||||||
|
} |
||||||
|
MySQLHelper struct { // Deprecated: Use MySQL{} instead
|
||||||
|
MySQL |
||||||
|
} |
||||||
|
SQLiteHelper struct { // Deprecated: Use SQLite{} instead
|
||||||
|
SQLite |
||||||
|
} |
||||||
|
SQLServerHelper struct { // Deprecated: Use SQLServer{} instead
|
||||||
|
SQLServer |
||||||
|
} |
||||||
|
OracleHelper struct { // Deprecated: Use Oracle{} instead
|
||||||
|
Oracle |
||||||
|
} |
||||||
|
) |
||||||
|
|
||||||
|
func (h *PostgreSQLHelper) disableReferentialIntegrity(db *sql.DB, loadFn loadFunction) error { |
||||||
|
h.PostgreSQL.UseAlterConstraint = h.UseAlterConstraint |
||||||
|
return h.PostgreSQL.disableReferentialIntegrity(db, loadFn) |
||||||
|
} |
||||||
|
|
||||||
|
// LoadFixtureFiles load all specified fixtures files into database:
|
||||||
|
// LoadFixtureFiles(db, &PostgreSQL{},
|
||||||
|
// "fixtures/customers.yml", "fixtures/orders.yml")
|
||||||
|
// // add as many files you want
|
||||||
|
//
|
||||||
|
// Deprecated: Use NewFiles() and Load() instead.
|
||||||
|
func LoadFixtureFiles(db *sql.DB, helper Helper, files ...string) error { |
||||||
|
c, err := NewFiles(db, helper, files...) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
return c.Load() |
||||||
|
} |
||||||
|
|
||||||
|
// LoadFixtures loads all fixtures in a given folder into the database:
|
||||||
|
// LoadFixtures("myfixturesfolder", db, &PostgreSQL{})
|
||||||
|
//
|
||||||
|
// Deprecated: Use NewFolder() and Load() instead.
|
||||||
|
func LoadFixtures(folderName string, db *sql.DB, helper Helper) error { |
||||||
|
c, err := NewFolder(db, helper, folderName) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
return c.Load() |
||||||
|
} |
@ -0,0 +1,38 @@ |
|||||||
|
package testfixtures |
||||||
|
|
||||||
|
import ( |
||||||
|
"database/sql" |
||||||
|
"fmt" |
||||||
|
) |
||||||
|
|
||||||
|
const ( |
||||||
|
paramTypeDollar = iota + 1 |
||||||
|
paramTypeQuestion |
||||||
|
paramTypeColon |
||||||
|
) |
||||||
|
|
||||||
|
type loadFunction func(tx *sql.Tx) error |
||||||
|
|
||||||
|
// Helper is the generic interface for the database helper
|
||||||
|
type Helper interface { |
||||||
|
init(*sql.DB) error |
||||||
|
disableReferentialIntegrity(*sql.DB, loadFunction) error |
||||||
|
paramType() int |
||||||
|
databaseName(*sql.DB) string |
||||||
|
quoteKeyword(string) string |
||||||
|
whileInsertOnTable(*sql.Tx, string, func() error) error |
||||||
|
} |
||||||
|
|
||||||
|
type baseHelper struct{} |
||||||
|
|
||||||
|
func (*baseHelper) init(_ *sql.DB) error { |
||||||
|
return nil |
||||||
|
} |
||||||
|
|
||||||
|
func (*baseHelper) quoteKeyword(str string) string { |
||||||
|
return fmt.Sprintf(`"%s"`, str) |
||||||
|
} |
||||||
|
|
||||||
|
func (*baseHelper) whileInsertOnTable(_ *sql.Tx, _ string, fn func() error) error { |
||||||
|
return fn() |
||||||
|
} |
@ -0,0 +1,45 @@ |
|||||||
|
package testfixtures |
||||||
|
|
||||||
|
import ( |
||||||
|
"database/sql" |
||||||
|
"fmt" |
||||||
|
) |
||||||
|
|
||||||
|
// MySQL is the MySQL helper for this package
|
||||||
|
type MySQL struct { |
||||||
|
baseHelper |
||||||
|
} |
||||||
|
|
||||||
|
func (*MySQL) paramType() int { |
||||||
|
return paramTypeQuestion |
||||||
|
} |
||||||
|
|
||||||
|
func (*MySQL) quoteKeyword(str string) string { |
||||||
|
return fmt.Sprintf("`%s`", str) |
||||||
|
} |
||||||
|
|
||||||
|
func (*MySQL) databaseName(db *sql.DB) (dbName string) { |
||||||
|
db.QueryRow("SELECT DATABASE()").Scan(&dbName) |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
func (h *MySQL) disableReferentialIntegrity(db *sql.DB, loadFn loadFunction) error { |
||||||
|
// re-enable after load
|
||||||
|
defer db.Exec("SET FOREIGN_KEY_CHECKS = 1") |
||||||
|
|
||||||
|
tx, err := db.Begin() |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
if _, err = tx.Exec("SET FOREIGN_KEY_CHECKS = 0"); err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
if err = loadFn(tx); err != nil { |
||||||
|
tx.Rollback() |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
return tx.Commit() |
||||||
|
} |
@ -0,0 +1,19 @@ |
|||||||
|
package testfixtures |
||||||
|
|
||||||
|
var ( |
||||||
|
skipDatabaseNameCheck bool |
||||||
|
resetSequencesTo int64 = 10000 |
||||||
|
) |
||||||
|
|
||||||
|
// SkipDatabaseNameCheck If true, loading fixtures will not check if the database
|
||||||
|
// name constaint "test". Use with caution!
|
||||||
|
func SkipDatabaseNameCheck(value bool) { |
||||||
|
skipDatabaseNameCheck = value |
||||||
|
} |
||||||
|
|
||||||
|
// ResetSequencesTo sets the value the sequences will be reset to.
|
||||||
|
// This is used by PostgreSQL and Oracle.
|
||||||
|
// Defaults to 10000.
|
||||||
|
func ResetSequencesTo(value int64) { |
||||||
|
resetSequencesTo = value |
||||||
|
} |
@ -0,0 +1,132 @@ |
|||||||
|
package testfixtures |
||||||
|
|
||||||
|
import ( |
||||||
|
"database/sql" |
||||||
|
"fmt" |
||||||
|
"strings" |
||||||
|
) |
||||||
|
|
||||||
|
// Oracle is the Oracle database helper for this package
|
||||||
|
type Oracle struct { |
||||||
|
baseHelper |
||||||
|
|
||||||
|
enabledConstraints []oracleConstraint |
||||||
|
sequences []string |
||||||
|
} |
||||||
|
|
||||||
|
type oracleConstraint struct { |
||||||
|
tableName string |
||||||
|
constraintName string |
||||||
|
} |
||||||
|
|
||||||
|
func (h *Oracle) init(db *sql.DB) error { |
||||||
|
var err error |
||||||
|
|
||||||
|
h.enabledConstraints, err = h.getEnabledConstraints(db) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
h.sequences, err = h.getSequences(db) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
return nil |
||||||
|
} |
||||||
|
|
||||||
|
func (*Oracle) paramType() int { |
||||||
|
return paramTypeColon |
||||||
|
} |
||||||
|
|
||||||
|
func (*Oracle) quoteKeyword(str string) string { |
||||||
|
return fmt.Sprintf("\"%s\"", strings.ToUpper(str)) |
||||||
|
} |
||||||
|
|
||||||
|
func (*Oracle) databaseName(db *sql.DB) (dbName string) { |
||||||
|
db.QueryRow("SELECT user FROM DUAL").Scan(&dbName) |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
func (*Oracle) getEnabledConstraints(db *sql.DB) ([]oracleConstraint, error) { |
||||||
|
constraints := make([]oracleConstraint, 0) |
||||||
|
rows, err := db.Query(` |
||||||
|
SELECT table_name, constraint_name |
||||||
|
FROM user_constraints |
||||||
|
WHERE constraint_type = 'R' |
||||||
|
AND status = 'ENABLED' |
||||||
|
`) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
defer rows.Close() |
||||||
|
for rows.Next() { |
||||||
|
var constraint oracleConstraint |
||||||
|
rows.Scan(&constraint.tableName, &constraint.constraintName) |
||||||
|
constraints = append(constraints, constraint) |
||||||
|
} |
||||||
|
return constraints, nil |
||||||
|
} |
||||||
|
|
||||||
|
func (*Oracle) getSequences(db *sql.DB) ([]string, error) { |
||||||
|
sequences := make([]string, 0) |
||||||
|
rows, err := db.Query("SELECT sequence_name FROM user_sequences") |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
|
||||||
|
defer rows.Close() |
||||||
|
for rows.Next() { |
||||||
|
var sequence string |
||||||
|
rows.Scan(&sequence) |
||||||
|
sequences = append(sequences, sequence) |
||||||
|
} |
||||||
|
return sequences, nil |
||||||
|
} |
||||||
|
|
||||||
|
func (h *Oracle) resetSequences(db *sql.DB) error { |
||||||
|
for _, sequence := range h.sequences { |
||||||
|
_, err := db.Exec(fmt.Sprintf("DROP SEQUENCE %s", h.quoteKeyword(sequence))) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
_, err = db.Exec(fmt.Sprintf("CREATE SEQUENCE %s START WITH %d", h.quoteKeyword(sequence), resetSequencesTo)) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
} |
||||||
|
return nil |
||||||
|
} |
||||||
|
|
||||||
|
func (h *Oracle) disableReferentialIntegrity(db *sql.DB, loadFn loadFunction) error { |
||||||
|
// re-enable after load
|
||||||
|
defer func() { |
||||||
|
for _, c := range h.enabledConstraints { |
||||||
|
db.Exec(fmt.Sprintf("ALTER TABLE %s ENABLE CONSTRAINT %s", h.quoteKeyword(c.tableName), h.quoteKeyword(c.constraintName))) |
||||||
|
} |
||||||
|
}() |
||||||
|
|
||||||
|
// disable foreign keys
|
||||||
|
for _, c := range h.enabledConstraints { |
||||||
|
_, err := db.Exec(fmt.Sprintf("ALTER TABLE %s DISABLE CONSTRAINT %s", h.quoteKeyword(c.tableName), h.quoteKeyword(c.constraintName))) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
tx, err := db.Begin() |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
if err = loadFn(tx); err != nil { |
||||||
|
tx.Rollback() |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
if err = tx.Commit(); err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
return h.resetSequences(db) |
||||||
|
} |
@ -0,0 +1,211 @@ |
|||||||
|
package testfixtures |
||||||
|
|
||||||
|
import ( |
||||||
|
"database/sql" |
||||||
|
"fmt" |
||||||
|
) |
||||||
|
|
||||||
|
// PostgreSQL is the PG helper for this package
|
||||||
|
type PostgreSQL struct { |
||||||
|
baseHelper |
||||||
|
|
||||||
|
// UseAlterConstraint If true, the contraint disabling will do
|
||||||
|
// using ALTER CONTRAINT sintax, only allowed in PG >= 9.4.
|
||||||
|
// If false, the constraint disabling will use DISABLE TRIGGER ALL,
|
||||||
|
// which requires SUPERUSER privileges.
|
||||||
|
UseAlterConstraint bool |
||||||
|
|
||||||
|
tables []string |
||||||
|
sequences []string |
||||||
|
nonDeferrableConstraints []pgConstraint |
||||||
|
} |
||||||
|
|
||||||
|
type pgConstraint struct { |
||||||
|
tableName string |
||||||
|
constraintName string |
||||||
|
} |
||||||
|
|
||||||
|
func (h *PostgreSQL) init(db *sql.DB) error { |
||||||
|
var err error |
||||||
|
|
||||||
|
h.tables, err = h.getTables(db) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
h.sequences, err = h.getSequences(db) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
h.nonDeferrableConstraints, err = h.getNonDeferrableConstraints(db) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
return nil |
||||||
|
} |
||||||
|
|
||||||
|
func (*PostgreSQL) paramType() int { |
||||||
|
return paramTypeDollar |
||||||
|
} |
||||||
|
|
||||||
|
func (*PostgreSQL) databaseName(db *sql.DB) (dbName string) { |
||||||
|
db.QueryRow("SELECT current_database()").Scan(&dbName) |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
func (h *PostgreSQL) getTables(db *sql.DB) ([]string, error) { |
||||||
|
var tables []string |
||||||
|
|
||||||
|
sql := ` |
||||||
|
SELECT table_name |
||||||
|
FROM information_schema.tables |
||||||
|
WHERE table_schema = 'public' |
||||||
|
AND table_type = 'BASE TABLE'; |
||||||
|
` |
||||||
|
rows, err := db.Query(sql) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
|
||||||
|
defer rows.Close() |
||||||
|
for rows.Next() { |
||||||
|
var table string |
||||||
|
rows.Scan(&table) |
||||||
|
tables = append(tables, table) |
||||||
|
} |
||||||
|
return tables, nil |
||||||
|
} |
||||||
|
|
||||||
|
func (h *PostgreSQL) getSequences(db *sql.DB) ([]string, error) { |
||||||
|
var sequences []string |
||||||
|
|
||||||
|
sql := "SELECT relname FROM pg_class WHERE relkind = 'S'" |
||||||
|
rows, err := db.Query(sql) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
|
||||||
|
defer rows.Close() |
||||||
|
for rows.Next() { |
||||||
|
var sequence string |
||||||
|
if err = rows.Scan(&sequence); err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
sequences = append(sequences, sequence) |
||||||
|
} |
||||||
|
return sequences, nil |
||||||
|
} |
||||||
|
|
||||||
|
func (*PostgreSQL) getNonDeferrableConstraints(db *sql.DB) ([]pgConstraint, error) { |
||||||
|
var constraints []pgConstraint |
||||||
|
|
||||||
|
sql := ` |
||||||
|
SELECT table_name, constraint_name |
||||||
|
FROM information_schema.table_constraints |
||||||
|
WHERE constraint_type = 'FOREIGN KEY' |
||||||
|
AND is_deferrable = 'NO'` |
||||||
|
rows, err := db.Query(sql) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
|
||||||
|
defer rows.Close() |
||||||
|
for rows.Next() { |
||||||
|
var constraint pgConstraint |
||||||
|
err = rows.Scan(&constraint.tableName, &constraint.constraintName) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
constraints = append(constraints, constraint) |
||||||
|
} |
||||||
|
return constraints, nil |
||||||
|
} |
||||||
|
|
||||||
|
func (h *PostgreSQL) disableTriggers(db *sql.DB, loadFn loadFunction) error { |
||||||
|
defer func() { |
||||||
|
// re-enable triggers after load
|
||||||
|
var sql string |
||||||
|
for _, table := range h.tables { |
||||||
|
sql += fmt.Sprintf("ALTER TABLE %s ENABLE TRIGGER ALL;", h.quoteKeyword(table)) |
||||||
|
} |
||||||
|
db.Exec(sql) |
||||||
|
}() |
||||||
|
|
||||||
|
tx, err := db.Begin() |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
var sql string |
||||||
|
for _, table := range h.tables { |
||||||
|
sql += fmt.Sprintf("ALTER TABLE %s DISABLE TRIGGER ALL;", h.quoteKeyword(table)) |
||||||
|
} |
||||||
|
if _, err = tx.Exec(sql); err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
if err = loadFn(tx); err != nil { |
||||||
|
tx.Rollback() |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
return tx.Commit() |
||||||
|
} |
||||||
|
|
||||||
|
func (h *PostgreSQL) makeConstraintsDeferrable(db *sql.DB, loadFn loadFunction) error { |
||||||
|
defer func() { |
||||||
|
// ensure constraint being not deferrable again after load
|
||||||
|
var sql string |
||||||
|
for _, constraint := range h.nonDeferrableConstraints { |
||||||
|
sql += fmt.Sprintf("ALTER TABLE %s ALTER CONSTRAINT %s NOT DEFERRABLE;", h.quoteKeyword(constraint.tableName), h.quoteKeyword(constraint.constraintName)) |
||||||
|
} |
||||||
|
db.Exec(sql) |
||||||
|
}() |
||||||
|
|
||||||
|
var sql string |
||||||
|
for _, constraint := range h.nonDeferrableConstraints { |
||||||
|
sql += fmt.Sprintf("ALTER TABLE %s ALTER CONSTRAINT %s DEFERRABLE;", h.quoteKeyword(constraint.tableName), h.quoteKeyword(constraint.constraintName)) |
||||||
|
} |
||||||
|
if _, err := db.Exec(sql); err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
tx, err := db.Begin() |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
if _, err = tx.Exec("SET CONSTRAINTS ALL DEFERRED"); err != nil { |
||||||
|
return nil |
||||||
|
} |
||||||
|
|
||||||
|
if err = loadFn(tx); err != nil { |
||||||
|
tx.Rollback() |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
return tx.Commit() |
||||||
|
} |
||||||
|
|
||||||
|
func (h *PostgreSQL) disableReferentialIntegrity(db *sql.DB, loadFn loadFunction) error { |
||||||
|
// ensure sequences being reset after load
|
||||||
|
defer h.resetSequences(db) |
||||||
|
|
||||||
|
if h.UseAlterConstraint { |
||||||
|
return h.makeConstraintsDeferrable(db, loadFn) |
||||||
|
} else { |
||||||
|
return h.disableTriggers(db, loadFn) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
func (h *PostgreSQL) resetSequences(db *sql.DB) error { |
||||||
|
for _, sequence := range h.sequences { |
||||||
|
_, err := db.Exec(fmt.Sprintf("SELECT SETVAL('%s', %d)", sequence, resetSequencesTo)) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
} |
||||||
|
return nil |
||||||
|
} |
@ -0,0 +1,40 @@ |
|||||||
|
package testfixtures |
||||||
|
|
||||||
|
import ( |
||||||
|
"database/sql" |
||||||
|
"path/filepath" |
||||||
|
) |
||||||
|
|
||||||
|
// SQLite is the SQLite Helper for this package
|
||||||
|
type SQLite struct { |
||||||
|
baseHelper |
||||||
|
} |
||||||
|
|
||||||
|
func (*SQLite) paramType() int { |
||||||
|
return paramTypeQuestion |
||||||
|
} |
||||||
|
|
||||||
|
func (*SQLite) databaseName(db *sql.DB) (dbName string) { |
||||||
|
var seq int |
||||||
|
var main string |
||||||
|
db.QueryRow("PRAGMA database_list").Scan(&seq, &main, &dbName) |
||||||
|
dbName = filepath.Base(dbName) |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
func (*SQLite) disableReferentialIntegrity(db *sql.DB, loadFn loadFunction) error { |
||||||
|
tx, err := db.Begin() |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
if _, err = tx.Exec("PRAGMA defer_foreign_keys = ON"); err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
if err = loadFn(tx); err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
return tx.Commit() |
||||||
|
} |
@ -0,0 +1,110 @@ |
|||||||
|
package testfixtures |
||||||
|
|
||||||
|
import ( |
||||||
|
"database/sql" |
||||||
|
"fmt" |
||||||
|
) |
||||||
|
|
||||||
|
// SQLServer is the helper for SQL Server for this package.
|
||||||
|
// SQL Server >= 2008 is required.
|
||||||
|
type SQLServer struct { |
||||||
|
baseHelper |
||||||
|
|
||||||
|
tables []string |
||||||
|
} |
||||||
|
|
||||||
|
func (h *SQLServer) init(db *sql.DB) error { |
||||||
|
var err error |
||||||
|
|
||||||
|
h.tables, err = h.getTables(db) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
return nil |
||||||
|
} |
||||||
|
|
||||||
|
func (*SQLServer) paramType() int { |
||||||
|
return paramTypeQuestion |
||||||
|
} |
||||||
|
|
||||||
|
func (*SQLServer) quoteKeyword(str string) string { |
||||||
|
return fmt.Sprintf("[%s]", str) |
||||||
|
} |
||||||
|
|
||||||
|
func (*SQLServer) databaseName(db *sql.DB) (dbname string) { |
||||||
|
db.QueryRow("SELECT DB_NAME()").Scan(&dbname) |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
func (*SQLServer) getTables(db *sql.DB) ([]string, error) { |
||||||
|
rows, err := db.Query("SELECT table_name FROM information_schema.tables") |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
|
||||||
|
tables := make([]string, 0) |
||||||
|
defer rows.Close() |
||||||
|
for rows.Next() { |
||||||
|
var table string |
||||||
|
rows.Scan(&table) |
||||||
|
tables = append(tables, table) |
||||||
|
} |
||||||
|
return tables, nil |
||||||
|
} |
||||||
|
|
||||||
|
func (*SQLServer) tableHasIdentityColumn(tx *sql.Tx, tableName string) bool { |
||||||
|
sql := ` |
||||||
|
SELECT COUNT(*) |
||||||
|
FROM SYS.IDENTITY_COLUMNS |
||||||
|
WHERE OBJECT_NAME(OBJECT_ID) = ? |
||||||
|
` |
||||||
|
var count int |
||||||
|
tx.QueryRow(sql, tableName).Scan(&count) |
||||||
|
return count > 0 |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
func (h *SQLServer) whileInsertOnTable(tx *sql.Tx, tableName string, fn func() error) error { |
||||||
|
if h.tableHasIdentityColumn(tx, tableName) { |
||||||
|
defer tx.Exec(fmt.Sprintf("SET IDENTITY_INSERT %s OFF", h.quoteKeyword(tableName))) |
||||||
|
_, err := tx.Exec(fmt.Sprintf("SET IDENTITY_INSERT %s ON", h.quoteKeyword(tableName))) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
} |
||||||
|
return fn() |
||||||
|
} |
||||||
|
|
||||||
|
func (h *SQLServer) disableReferentialIntegrity(db *sql.DB, loadFn loadFunction) error { |
||||||
|
// ensure the triggers are re-enable after all
|
||||||
|
defer func() { |
||||||
|
sql := "" |
||||||
|
for _, table := range h.tables { |
||||||
|
sql += fmt.Sprintf("ALTER TABLE %s WITH CHECK CHECK CONSTRAINT ALL;", h.quoteKeyword(table)) |
||||||
|
} |
||||||
|
if _, err := db.Exec(sql); err != nil { |
||||||
|
fmt.Printf("Error on re-enabling constraints: %v\n", err) |
||||||
|
} |
||||||
|
}() |
||||||
|
|
||||||
|
sql := "" |
||||||
|
for _, table := range h.tables { |
||||||
|
sql += fmt.Sprintf("ALTER TABLE %s NOCHECK CONSTRAINT ALL;", h.quoteKeyword(table)) |
||||||
|
} |
||||||
|
if _, err := db.Exec(sql); err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
tx, err := db.Begin() |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
if err = loadFn(tx); err != nil { |
||||||
|
tx.Rollback() |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
return tx.Commit() |
||||||
|
} |
@ -0,0 +1,279 @@ |
|||||||
|
package testfixtures |
||||||
|
|
||||||
|
import ( |
||||||
|
"database/sql" |
||||||
|
"errors" |
||||||
|
"fmt" |
||||||
|
"io/ioutil" |
||||||
|
"path" |
||||||
|
"path/filepath" |
||||||
|
"regexp" |
||||||
|
"strings" |
||||||
|
|
||||||
|
"gopkg.in/yaml.v2" |
||||||
|
) |
||||||
|
|
||||||
|
// Context holds the fixtures to be loaded in the database.
|
||||||
|
type Context struct { |
||||||
|
db *sql.DB |
||||||
|
helper Helper |
||||||
|
fixturesFiles []*fixtureFile |
||||||
|
} |
||||||
|
|
||||||
|
type fixtureFile struct { |
||||||
|
path string |
||||||
|
fileName string |
||||||
|
content []byte |
||||||
|
insertSQLs []insertSQL |
||||||
|
} |
||||||
|
|
||||||
|
type insertSQL struct { |
||||||
|
sql string |
||||||
|
params []interface{} |
||||||
|
} |
||||||
|
|
||||||
|
var ( |
||||||
|
// ErrWrongCastNotAMap is returned when a map is not a map[interface{}]interface{}
|
||||||
|
ErrWrongCastNotAMap = errors.New("Could not cast record: not a map[interface{}]interface{}") |
||||||
|
|
||||||
|
// ErrFileIsNotSliceOrMap is returned the the fixture file is not a slice or map.
|
||||||
|
ErrFileIsNotSliceOrMap = errors.New("The fixture file is not a slice or map") |
||||||
|
|
||||||
|
// ErrKeyIsNotString is returned when a record is not of type string
|
||||||
|
ErrKeyIsNotString = errors.New("Record map key is not string") |
||||||
|
|
||||||
|
// ErrNotTestDatabase is returned when the database name doesn't contains "test"
|
||||||
|
ErrNotTestDatabase = errors.New(`Loading aborted because the database name does not contains "test"`) |
||||||
|
|
||||||
|
dbnameRegexp = regexp.MustCompile("(?i)test") |
||||||
|
) |
||||||
|
|
||||||
|
// NewFolder craetes a context for all fixtures in a given folder into the database:
|
||||||
|
// NewFolder(db, &PostgreSQL{}, "my/fixtures/folder")
|
||||||
|
func NewFolder(db *sql.DB, helper Helper, folderName string) (*Context, error) { |
||||||
|
fixtures, err := fixturesFromFolder(folderName) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
|
||||||
|
c, err := newContext(db, helper, fixtures) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
|
||||||
|
return c, nil |
||||||
|
} |
||||||
|
|
||||||
|
// NewFiles craetes a context for all specified fixtures files into database:
|
||||||
|
// NewFiles(db, &PostgreSQL{},
|
||||||
|
// "fixtures/customers.yml",
|
||||||
|
// "fixtures/orders.yml"
|
||||||
|
// // add as many files you want
|
||||||
|
// )
|
||||||
|
func NewFiles(db *sql.DB, helper Helper, fileNames ...string) (*Context, error) { |
||||||
|
fixtures, err := fixturesFromFiles(fileNames...) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
|
||||||
|
c, err := newContext(db, helper, fixtures) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
|
||||||
|
return c, nil |
||||||
|
} |
||||||
|
|
||||||
|
func newContext(db *sql.DB, helper Helper, fixtures []*fixtureFile) (*Context, error) { |
||||||
|
c := &Context{ |
||||||
|
db: db, |
||||||
|
helper: helper, |
||||||
|
fixturesFiles: fixtures, |
||||||
|
} |
||||||
|
|
||||||
|
if err := c.helper.init(c.db); err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
|
||||||
|
if err := c.buildInsertSQLs(); err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
|
||||||
|
return c, nil |
||||||
|
} |
||||||
|
|
||||||
|
// Load wipes and after load all fixtures in the database.
|
||||||
|
// if err := fixtures.Load(); err != nil {
|
||||||
|
// log.Fatal(err)
|
||||||
|
// }
|
||||||
|
func (c *Context) Load() error { |
||||||
|
if !skipDatabaseNameCheck { |
||||||
|
if !dbnameRegexp.MatchString(c.helper.databaseName(c.db)) { |
||||||
|
return ErrNotTestDatabase |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
err := c.helper.disableReferentialIntegrity(c.db, func(tx *sql.Tx) error { |
||||||
|
for _, file := range c.fixturesFiles { |
||||||
|
if err := file.delete(tx, c.helper); err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
err := c.helper.whileInsertOnTable(tx, file.fileNameWithoutExtension(), func() error { |
||||||
|
for _, i := range file.insertSQLs { |
||||||
|
if _, err := tx.Exec(i.sql, i.params...); err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
} |
||||||
|
return nil |
||||||
|
}) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
} |
||||||
|
return nil |
||||||
|
}) |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
func (c *Context) buildInsertSQLs() error { |
||||||
|
for _, f := range c.fixturesFiles { |
||||||
|
var records interface{} |
||||||
|
if err := yaml.Unmarshal(f.content, &records); err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
switch records := records.(type) { |
||||||
|
case []interface{}: |
||||||
|
for _, record := range records { |
||||||
|
recordMap, ok := record.(map[interface{}]interface{}) |
||||||
|
if !ok { |
||||||
|
return ErrWrongCastNotAMap |
||||||
|
} |
||||||
|
|
||||||
|
sql, values, err := f.buildInsertSQL(c.helper, recordMap) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
f.insertSQLs = append(f.insertSQLs, insertSQL{sql, values}) |
||||||
|
} |
||||||
|
case map[interface{}]interface{}: |
||||||
|
for _, record := range records { |
||||||
|
recordMap, ok := record.(map[interface{}]interface{}) |
||||||
|
if !ok { |
||||||
|
return ErrWrongCastNotAMap |
||||||
|
} |
||||||
|
|
||||||
|
sql, values, err := f.buildInsertSQL(c.helper, recordMap) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
f.insertSQLs = append(f.insertSQLs, insertSQL{sql, values}) |
||||||
|
} |
||||||
|
default: |
||||||
|
return ErrFileIsNotSliceOrMap |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
return nil |
||||||
|
} |
||||||
|
|
||||||
|
func (f *fixtureFile) fileNameWithoutExtension() string { |
||||||
|
return strings.Replace(f.fileName, filepath.Ext(f.fileName), "", 1) |
||||||
|
} |
||||||
|
|
||||||
|
func (f *fixtureFile) delete(tx *sql.Tx, h Helper) error { |
||||||
|
_, err := tx.Exec(fmt.Sprintf("DELETE FROM %s", h.quoteKeyword(f.fileNameWithoutExtension()))) |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
func (f *fixtureFile) buildInsertSQL(h Helper, record map[interface{}]interface{}) (sqlStr string, values []interface{}, err error) { |
||||||
|
var ( |
||||||
|
sqlColumns []string |
||||||
|
sqlValues []string |
||||||
|
i = 1 |
||||||
|
) |
||||||
|
for key, value := range record { |
||||||
|
keyStr, ok := key.(string) |
||||||
|
if !ok { |
||||||
|
err = ErrKeyIsNotString |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
sqlColumns = append(sqlColumns, h.quoteKeyword(keyStr)) |
||||||
|
|
||||||
|
switch h.paramType() { |
||||||
|
case paramTypeDollar: |
||||||
|
sqlValues = append(sqlValues, fmt.Sprintf("$%d", i)) |
||||||
|
case paramTypeQuestion: |
||||||
|
sqlValues = append(sqlValues, "?") |
||||||
|
case paramTypeColon: |
||||||
|
switch { |
||||||
|
case isDateTime(value): |
||||||
|
sqlValues = append(sqlValues, fmt.Sprintf("to_date(:%d, 'YYYY-MM-DD HH24:MI:SS')", i)) |
||||||
|
case isDate(value): |
||||||
|
sqlValues = append(sqlValues, fmt.Sprintf("to_date(:%d, 'YYYY-MM-DD')", i)) |
||||||
|
case isTime(value): |
||||||
|
sqlValues = append(sqlValues, fmt.Sprintf("to_date(:%d, 'HH24:MI:SS')", i)) |
||||||
|
default: |
||||||
|
sqlValues = append(sqlValues, fmt.Sprintf(":%d", i)) |
||||||
|
} |
||||||
|
} |
||||||
|
i++ |
||||||
|
values = append(values, value) |
||||||
|
} |
||||||
|
|
||||||
|
sqlStr = fmt.Sprintf( |
||||||
|
"INSERT INTO %s (%s) VALUES (%s)", |
||||||
|
h.quoteKeyword(f.fileNameWithoutExtension()), |
||||||
|
strings.Join(sqlColumns, ", "), |
||||||
|
strings.Join(sqlValues, ", "), |
||||||
|
) |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
func fixturesFromFolder(folderName string) ([]*fixtureFile, error) { |
||||||
|
var files []*fixtureFile |
||||||
|
fileinfos, err := ioutil.ReadDir(folderName) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
|
||||||
|
for _, fileinfo := range fileinfos { |
||||||
|
if !fileinfo.IsDir() && filepath.Ext(fileinfo.Name()) == ".yml" { |
||||||
|
fixture := &fixtureFile{ |
||||||
|
path: path.Join(folderName, fileinfo.Name()), |
||||||
|
fileName: fileinfo.Name(), |
||||||
|
} |
||||||
|
fixture.content, err = ioutil.ReadFile(fixture.path) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
files = append(files, fixture) |
||||||
|
} |
||||||
|
} |
||||||
|
return files, nil |
||||||
|
} |
||||||
|
|
||||||
|
func fixturesFromFiles(fileNames ...string) ([]*fixtureFile, error) { |
||||||
|
var ( |
||||||
|
fixtureFiles []*fixtureFile |
||||||
|
err error |
||||||
|
) |
||||||
|
|
||||||
|
for _, f := range fileNames { |
||||||
|
fixture := &fixtureFile{ |
||||||
|
path: f, |
||||||
|
fileName: filepath.Base(f), |
||||||
|
} |
||||||
|
fixture.content, err = ioutil.ReadFile(fixture.path) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
fixtureFiles = append(fixtureFiles, fixture) |
||||||
|
} |
||||||
|
|
||||||
|
return fixtureFiles, nil |
||||||
|
} |
@ -0,0 +1,36 @@ |
|||||||
|
package testfixtures |
||||||
|
|
||||||
|
import "regexp" |
||||||
|
|
||||||
|
var ( |
||||||
|
regexpDate = regexp.MustCompile("\\d\\d\\d\\d-\\d\\d-\\d\\d") |
||||||
|
regexpDateTime = regexp.MustCompile("\\d\\d\\d\\d-\\d\\d-\\d\\d \\d\\d:\\d\\d:\\d\\d") |
||||||
|
regexpTime = regexp.MustCompile("\\d\\d:\\d\\d:\\d\\d") |
||||||
|
) |
||||||
|
|
||||||
|
func isDate(value interface{}) bool { |
||||||
|
str, isStr := value.(string) |
||||||
|
if !isStr { |
||||||
|
return false |
||||||
|
} |
||||||
|
|
||||||
|
return regexpDate.MatchString(str) |
||||||
|
} |
||||||
|
|
||||||
|
func isDateTime(value interface{}) bool { |
||||||
|
str, isStr := value.(string) |
||||||
|
if !isStr { |
||||||
|
return false |
||||||
|
} |
||||||
|
|
||||||
|
return regexpDateTime.MatchString(str) |
||||||
|
} |
||||||
|
|
||||||
|
func isTime(value interface{}) bool { |
||||||
|
str, isStr := value.(string) |
||||||
|
if !isStr { |
||||||
|
return false |
||||||
|
} |
||||||
|
|
||||||
|
return regexpTime.MatchString(str) |
||||||
|
} |
@ -0,0 +1,13 @@ |
|||||||
|
Copyright 2011-2016 Canonical Ltd. |
||||||
|
|
||||||
|
Licensed under the Apache License, Version 2.0 (the "License"); |
||||||
|
you may not use this file except in compliance with the License. |
||||||
|
You may obtain a copy of the License at |
||||||
|
|
||||||
|
http://www.apache.org/licenses/LICENSE-2.0 |
||||||
|
|
||||||
|
Unless required by applicable law or agreed to in writing, software |
||||||
|
distributed under the License is distributed on an "AS IS" BASIS, |
||||||
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||||
|
See the License for the specific language governing permissions and |
||||||
|
limitations under the License. |
@ -0,0 +1,31 @@ |
|||||||
|
The following files were ported to Go from C files of libyaml, and thus |
||||||
|
are still covered by their original copyright and license: |
||||||
|
|
||||||
|
apic.go |
||||||
|
emitterc.go |
||||||
|
parserc.go |
||||||
|
readerc.go |
||||||
|
scannerc.go |
||||||
|
writerc.go |
||||||
|
yamlh.go |
||||||
|
yamlprivateh.go |
||||||
|
|
||||||
|
Copyright (c) 2006 Kirill Simonov |
||||||
|
|
||||||
|
Permission is hereby granted, free of charge, to any person obtaining a copy of |
||||||
|
this software and associated documentation files (the "Software"), to deal in |
||||||
|
the Software without restriction, including without limitation the rights to |
||||||
|
use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies |
||||||
|
of the Software, and to permit persons to whom the Software is furnished to do |
||||||
|
so, subject to the following conditions: |
||||||
|
|
||||||
|
The above copyright notice and this permission notice shall be included in all |
||||||
|
copies or substantial portions of the Software. |
||||||
|
|
||||||
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
||||||
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
||||||
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE |
||||||
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
||||||
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
||||||
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE |
||||||
|
SOFTWARE. |
@ -0,0 +1,131 @@ |
|||||||
|
# YAML support for the Go language |
||||||
|
|
||||||
|
Introduction |
||||||
|
------------ |
||||||
|
|
||||||
|
The yaml package enables Go programs to comfortably encode and decode YAML |
||||||
|
values. It was developed within [Canonical](https://www.canonical.com) as |
||||||
|
part of the [juju](https://juju.ubuntu.com) project, and is based on a |
||||||
|
pure Go port of the well-known [libyaml](http://pyyaml.org/wiki/LibYAML) |
||||||
|
C library to parse and generate YAML data quickly and reliably. |
||||||
|
|
||||||
|
Compatibility |
||||||
|
------------- |
||||||
|
|
||||||
|
The yaml package supports most of YAML 1.1 and 1.2, including support for |
||||||
|
anchors, tags, map merging, etc. Multi-document unmarshalling is not yet |
||||||
|
implemented, and base-60 floats from YAML 1.1 are purposefully not |
||||||
|
supported since they're a poor design and are gone in YAML 1.2. |
||||||
|
|
||||||
|
Installation and usage |
||||||
|
---------------------- |
||||||
|
|
||||||
|
The import path for the package is *gopkg.in/yaml.v2*. |
||||||
|
|
||||||
|
To install it, run: |
||||||
|
|
||||||
|
go get gopkg.in/yaml.v2 |
||||||
|
|
||||||
|
API documentation |
||||||
|
----------------- |
||||||
|
|
||||||
|
If opened in a browser, the import path itself leads to the API documentation: |
||||||
|
|
||||||
|
* [https://gopkg.in/yaml.v2](https://gopkg.in/yaml.v2) |
||||||
|
|
||||||
|
API stability |
||||||
|
------------- |
||||||
|
|
||||||
|
The package API for yaml v2 will remain stable as described in [gopkg.in](https://gopkg.in). |
||||||
|
|
||||||
|
|
||||||
|
License |
||||||
|
------- |
||||||
|
|
||||||
|
The yaml package is licensed under the Apache License 2.0. Please see the LICENSE file for details. |
||||||
|
|
||||||
|
|
||||||
|
Example |
||||||
|
------- |
||||||
|
|
||||||
|
```Go |
||||||
|
package main |
||||||
|
|
||||||
|
import ( |
||||||
|
"fmt" |
||||||
|
"log" |
||||||
|
|
||||||
|
"gopkg.in/yaml.v2" |
||||||
|
) |
||||||
|
|
||||||
|
var data = ` |
||||||
|
a: Easy! |
||||||
|
b: |
||||||
|
c: 2 |
||||||
|
d: [3, 4] |
||||||
|
` |
||||||
|
|
||||||
|
type T struct { |
||||||
|
A string |
||||||
|
B struct { |
||||||
|
RenamedC int `yaml:"c"` |
||||||
|
D []int `yaml:",flow"` |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
func main() { |
||||||
|
t := T{} |
||||||
|
|
||||||
|
err := yaml.Unmarshal([]byte(data), &t) |
||||||
|
if err != nil { |
||||||
|
log.Fatalf("error: %v", err) |
||||||
|
} |
||||||
|
fmt.Printf("--- t:\n%v\n\n", t) |
||||||
|
|
||||||
|
d, err := yaml.Marshal(&t) |
||||||
|
if err != nil { |
||||||
|
log.Fatalf("error: %v", err) |
||||||
|
} |
||||||
|
fmt.Printf("--- t dump:\n%s\n\n", string(d)) |
||||||
|
|
||||||
|
m := make(map[interface{}]interface{}) |
||||||
|
|
||||||
|
err = yaml.Unmarshal([]byte(data), &m) |
||||||
|
if err != nil { |
||||||
|
log.Fatalf("error: %v", err) |
||||||
|
} |
||||||
|
fmt.Printf("--- m:\n%v\n\n", m) |
||||||
|
|
||||||
|
d, err = yaml.Marshal(&m) |
||||||
|
if err != nil { |
||||||
|
log.Fatalf("error: %v", err) |
||||||
|
} |
||||||
|
fmt.Printf("--- m dump:\n%s\n\n", string(d)) |
||||||
|
} |
||||||
|
``` |
||||||
|
|
||||||
|
This example will generate the following output: |
||||||
|
|
||||||
|
``` |
||||||
|
--- t: |
||||||
|
{Easy! {2 [3 4]}} |
||||||
|
|
||||||
|
--- t dump: |
||||||
|
a: Easy! |
||||||
|
b: |
||||||
|
c: 2 |
||||||
|
d: [3, 4] |
||||||
|
|
||||||
|
|
||||||
|
--- m: |
||||||
|
map[a:Easy! b:map[c:2 d:[3 4]]] |
||||||
|
|
||||||
|
--- m dump: |
||||||
|
a: Easy! |
||||||
|
b: |
||||||
|
c: 2 |
||||||
|
d: |
||||||
|
- 3 |
||||||
|
- 4 |
||||||
|
``` |
||||||
|
|
@ -0,0 +1,742 @@ |
|||||||
|
package yaml |
||||||
|
|
||||||
|
import ( |
||||||
|
"io" |
||||||
|
"os" |
||||||
|
) |
||||||
|
|
||||||
|
func yaml_insert_token(parser *yaml_parser_t, pos int, token *yaml_token_t) { |
||||||
|
//fmt.Println("yaml_insert_token", "pos:", pos, "typ:", token.typ, "head:", parser.tokens_head, "len:", len(parser.tokens))
|
||||||
|
|
||||||
|
// Check if we can move the queue at the beginning of the buffer.
|
||||||
|
if parser.tokens_head > 0 && len(parser.tokens) == cap(parser.tokens) { |
||||||
|
if parser.tokens_head != len(parser.tokens) { |
||||||
|
copy(parser.tokens, parser.tokens[parser.tokens_head:]) |
||||||
|
} |
||||||
|
parser.tokens = parser.tokens[:len(parser.tokens)-parser.tokens_head] |
||||||
|
parser.tokens_head = 0 |
||||||
|
} |
||||||
|
parser.tokens = append(parser.tokens, *token) |
||||||
|
if pos < 0 { |
||||||
|
return |
||||||
|
} |
||||||
|
copy(parser.tokens[parser.tokens_head+pos+1:], parser.tokens[parser.tokens_head+pos:]) |
||||||
|
parser.tokens[parser.tokens_head+pos] = *token |
||||||
|
} |
||||||
|
|
||||||
|
// Create a new parser object.
|
||||||
|
func yaml_parser_initialize(parser *yaml_parser_t) bool { |
||||||
|
*parser = yaml_parser_t{ |
||||||
|
raw_buffer: make([]byte, 0, input_raw_buffer_size), |
||||||
|
buffer: make([]byte, 0, input_buffer_size), |
||||||
|
} |
||||||
|
return true |
||||||
|
} |
||||||
|
|
||||||
|
// Destroy a parser object.
|
||||||
|
func yaml_parser_delete(parser *yaml_parser_t) { |
||||||
|
*parser = yaml_parser_t{} |
||||||
|
} |
||||||
|
|
||||||
|
// String read handler.
|
||||||
|
func yaml_string_read_handler(parser *yaml_parser_t, buffer []byte) (n int, err error) { |
||||||
|
if parser.input_pos == len(parser.input) { |
||||||
|
return 0, io.EOF |
||||||
|
} |
||||||
|
n = copy(buffer, parser.input[parser.input_pos:]) |
||||||
|
parser.input_pos += n |
||||||
|
return n, nil |
||||||
|
} |
||||||
|
|
||||||
|
// File read handler.
|
||||||
|
func yaml_file_read_handler(parser *yaml_parser_t, buffer []byte) (n int, err error) { |
||||||
|
return parser.input_file.Read(buffer) |
||||||
|
} |
||||||
|
|
||||||
|
// Set a string input.
|
||||||
|
func yaml_parser_set_input_string(parser *yaml_parser_t, input []byte) { |
||||||
|
if parser.read_handler != nil { |
||||||
|
panic("must set the input source only once") |
||||||
|
} |
||||||
|
parser.read_handler = yaml_string_read_handler |
||||||
|
parser.input = input |
||||||
|
parser.input_pos = 0 |
||||||
|
} |
||||||
|
|
||||||
|
// Set a file input.
|
||||||
|
func yaml_parser_set_input_file(parser *yaml_parser_t, file *os.File) { |
||||||
|
if parser.read_handler != nil { |
||||||
|
panic("must set the input source only once") |
||||||
|
} |
||||||
|
parser.read_handler = yaml_file_read_handler |
||||||
|
parser.input_file = file |
||||||
|
} |
||||||
|
|
||||||
|
// Set the source encoding.
|
||||||
|
func yaml_parser_set_encoding(parser *yaml_parser_t, encoding yaml_encoding_t) { |
||||||
|
if parser.encoding != yaml_ANY_ENCODING { |
||||||
|
panic("must set the encoding only once") |
||||||
|
} |
||||||
|
parser.encoding = encoding |
||||||
|
} |
||||||
|
|
||||||
|
// Create a new emitter object.
|
||||||
|
func yaml_emitter_initialize(emitter *yaml_emitter_t) bool { |
||||||
|
*emitter = yaml_emitter_t{ |
||||||
|
buffer: make([]byte, output_buffer_size), |
||||||
|
raw_buffer: make([]byte, 0, output_raw_buffer_size), |
||||||
|
states: make([]yaml_emitter_state_t, 0, initial_stack_size), |
||||||
|
events: make([]yaml_event_t, 0, initial_queue_size), |
||||||
|
} |
||||||
|
return true |
||||||
|
} |
||||||
|
|
||||||
|
// Destroy an emitter object.
|
||||||
|
func yaml_emitter_delete(emitter *yaml_emitter_t) { |
||||||
|
*emitter = yaml_emitter_t{} |
||||||
|
} |
||||||
|
|
||||||
|
// String write handler.
|
||||||
|
func yaml_string_write_handler(emitter *yaml_emitter_t, buffer []byte) error { |
||||||
|
*emitter.output_buffer = append(*emitter.output_buffer, buffer...) |
||||||
|
return nil |
||||||
|
} |
||||||
|
|
||||||
|
// File write handler.
|
||||||
|
func yaml_file_write_handler(emitter *yaml_emitter_t, buffer []byte) error { |
||||||
|
_, err := emitter.output_file.Write(buffer) |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
// Set a string output.
|
||||||
|
func yaml_emitter_set_output_string(emitter *yaml_emitter_t, output_buffer *[]byte) { |
||||||
|
if emitter.write_handler != nil { |
||||||
|
panic("must set the output target only once") |
||||||
|
} |
||||||
|
emitter.write_handler = yaml_string_write_handler |
||||||
|
emitter.output_buffer = output_buffer |
||||||
|
} |
||||||
|
|
||||||
|
// Set a file output.
|
||||||
|
func yaml_emitter_set_output_file(emitter *yaml_emitter_t, file io.Writer) { |
||||||
|
if emitter.write_handler != nil { |
||||||
|
panic("must set the output target only once") |
||||||
|
} |
||||||
|
emitter.write_handler = yaml_file_write_handler |
||||||
|
emitter.output_file = file |
||||||
|
} |
||||||
|
|
||||||
|
// Set the output encoding.
|
||||||
|
func yaml_emitter_set_encoding(emitter *yaml_emitter_t, encoding yaml_encoding_t) { |
||||||
|
if emitter.encoding != yaml_ANY_ENCODING { |
||||||
|
panic("must set the output encoding only once") |
||||||
|
} |
||||||
|
emitter.encoding = encoding |
||||||
|
} |
||||||
|
|
||||||
|
// Set the canonical output style.
|
||||||
|
func yaml_emitter_set_canonical(emitter *yaml_emitter_t, canonical bool) { |
||||||
|
emitter.canonical = canonical |
||||||
|
} |
||||||
|
|
||||||
|
//// Set the indentation increment.
|
||||||
|
func yaml_emitter_set_indent(emitter *yaml_emitter_t, indent int) { |
||||||
|
if indent < 2 || indent > 9 { |
||||||
|
indent = 2 |
||||||
|
} |
||||||
|
emitter.best_indent = indent |
||||||
|
} |
||||||
|
|
||||||
|
// Set the preferred line width.
|
||||||
|
func yaml_emitter_set_width(emitter *yaml_emitter_t, width int) { |
||||||
|
if width < 0 { |
||||||
|
width = -1 |
||||||
|
} |
||||||
|
emitter.best_width = width |
||||||
|
} |
||||||
|
|
||||||
|
// Set if unescaped non-ASCII characters are allowed.
|
||||||
|
func yaml_emitter_set_unicode(emitter *yaml_emitter_t, unicode bool) { |
||||||
|
emitter.unicode = unicode |
||||||
|
} |
||||||
|
|
||||||
|
// Set the preferred line break character.
|
||||||
|
func yaml_emitter_set_break(emitter *yaml_emitter_t, line_break yaml_break_t) { |
||||||
|
emitter.line_break = line_break |
||||||
|
} |
||||||
|
|
||||||
|
///*
|
||||||
|
// * Destroy a token object.
|
||||||
|
// */
|
||||||
|
//
|
||||||
|
//YAML_DECLARE(void)
|
||||||
|
//yaml_token_delete(yaml_token_t *token)
|
||||||
|
//{
|
||||||
|
// assert(token); // Non-NULL token object expected.
|
||||||
|
//
|
||||||
|
// switch (token.type)
|
||||||
|
// {
|
||||||
|
// case YAML_TAG_DIRECTIVE_TOKEN:
|
||||||
|
// yaml_free(token.data.tag_directive.handle);
|
||||||
|
// yaml_free(token.data.tag_directive.prefix);
|
||||||
|
// break;
|
||||||
|
//
|
||||||
|
// case YAML_ALIAS_TOKEN:
|
||||||
|
// yaml_free(token.data.alias.value);
|
||||||
|
// break;
|
||||||
|
//
|
||||||
|
// case YAML_ANCHOR_TOKEN:
|
||||||
|
// yaml_free(token.data.anchor.value);
|
||||||
|
// break;
|
||||||
|
//
|
||||||
|
// case YAML_TAG_TOKEN:
|
||||||
|
// yaml_free(token.data.tag.handle);
|
||||||
|
// yaml_free(token.data.tag.suffix);
|
||||||
|
// break;
|
||||||
|
//
|
||||||
|
// case YAML_SCALAR_TOKEN:
|
||||||
|
// yaml_free(token.data.scalar.value);
|
||||||
|
// break;
|
||||||
|
//
|
||||||
|
// default:
|
||||||
|
// break;
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// memset(token, 0, sizeof(yaml_token_t));
|
||||||
|
//}
|
||||||
|
//
|
||||||
|
///*
|
||||||
|
// * Check if a string is a valid UTF-8 sequence.
|
||||||
|
// *
|
||||||
|
// * Check 'reader.c' for more details on UTF-8 encoding.
|
||||||
|
// */
|
||||||
|
//
|
||||||
|
//static int
|
||||||
|
//yaml_check_utf8(yaml_char_t *start, size_t length)
|
||||||
|
//{
|
||||||
|
// yaml_char_t *end = start+length;
|
||||||
|
// yaml_char_t *pointer = start;
|
||||||
|
//
|
||||||
|
// while (pointer < end) {
|
||||||
|
// unsigned char octet;
|
||||||
|
// unsigned int width;
|
||||||
|
// unsigned int value;
|
||||||
|
// size_t k;
|
||||||
|
//
|
||||||
|
// octet = pointer[0];
|
||||||
|
// width = (octet & 0x80) == 0x00 ? 1 :
|
||||||
|
// (octet & 0xE0) == 0xC0 ? 2 :
|
||||||
|
// (octet & 0xF0) == 0xE0 ? 3 :
|
||||||
|
// (octet & 0xF8) == 0xF0 ? 4 : 0;
|
||||||
|
// value = (octet & 0x80) == 0x00 ? octet & 0x7F :
|
||||||
|
// (octet & 0xE0) == 0xC0 ? octet & 0x1F :
|
||||||
|
// (octet & 0xF0) == 0xE0 ? octet & 0x0F :
|
||||||
|
// (octet & 0xF8) == 0xF0 ? octet & 0x07 : 0;
|
||||||
|
// if (!width) return 0;
|
||||||
|
// if (pointer+width > end) return 0;
|
||||||
|
// for (k = 1; k < width; k ++) {
|
||||||
|
// octet = pointer[k];
|
||||||
|
// if ((octet & 0xC0) != 0x80) return 0;
|
||||||
|
// value = (value << 6) + (octet & 0x3F);
|
||||||
|
// }
|
||||||
|
// if (!((width == 1) ||
|
||||||
|
// (width == 2 && value >= 0x80) ||
|
||||||
|
// (width == 3 && value >= 0x800) ||
|
||||||
|
// (width == 4 && value >= 0x10000))) return 0;
|
||||||
|
//
|
||||||
|
// pointer += width;
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// return 1;
|
||||||
|
//}
|
||||||
|
//
|
||||||
|
|
||||||
|
// Create STREAM-START.
|
||||||
|
func yaml_stream_start_event_initialize(event *yaml_event_t, encoding yaml_encoding_t) bool { |
||||||
|
*event = yaml_event_t{ |
||||||
|
typ: yaml_STREAM_START_EVENT, |
||||||
|
encoding: encoding, |
||||||
|
} |
||||||
|
return true |
||||||
|
} |
||||||
|
|
||||||
|
// Create STREAM-END.
|
||||||
|
func yaml_stream_end_event_initialize(event *yaml_event_t) bool { |
||||||
|
*event = yaml_event_t{ |
||||||
|
typ: yaml_STREAM_END_EVENT, |
||||||
|
} |
||||||
|
return true |
||||||
|
} |
||||||
|
|
||||||
|
// Create DOCUMENT-START.
|
||||||
|
func yaml_document_start_event_initialize(event *yaml_event_t, version_directive *yaml_version_directive_t, |
||||||
|
tag_directives []yaml_tag_directive_t, implicit bool) bool { |
||||||
|
*event = yaml_event_t{ |
||||||
|
typ: yaml_DOCUMENT_START_EVENT, |
||||||
|
version_directive: version_directive, |
||||||
|
tag_directives: tag_directives, |
||||||
|
implicit: implicit, |
||||||
|
} |
||||||
|
return true |
||||||
|
} |
||||||
|
|
||||||
|
// Create DOCUMENT-END.
|
||||||
|
func yaml_document_end_event_initialize(event *yaml_event_t, implicit bool) bool { |
||||||
|
*event = yaml_event_t{ |
||||||
|
typ: yaml_DOCUMENT_END_EVENT, |
||||||
|
implicit: implicit, |
||||||
|
} |
||||||
|
return true |
||||||
|
} |
||||||
|
|
||||||
|
///*
|
||||||
|
// * Create ALIAS.
|
||||||
|
// */
|
||||||
|
//
|
||||||
|
//YAML_DECLARE(int)
|
||||||
|
//yaml_alias_event_initialize(event *yaml_event_t, anchor *yaml_char_t)
|
||||||
|
//{
|
||||||
|
// mark yaml_mark_t = { 0, 0, 0 }
|
||||||
|
// anchor_copy *yaml_char_t = NULL
|
||||||
|
//
|
||||||
|
// assert(event) // Non-NULL event object is expected.
|
||||||
|
// assert(anchor) // Non-NULL anchor is expected.
|
||||||
|
//
|
||||||
|
// if (!yaml_check_utf8(anchor, strlen((char *)anchor))) return 0
|
||||||
|
//
|
||||||
|
// anchor_copy = yaml_strdup(anchor)
|
||||||
|
// if (!anchor_copy)
|
||||||
|
// return 0
|
||||||
|
//
|
||||||
|
// ALIAS_EVENT_INIT(*event, anchor_copy, mark, mark)
|
||||||
|
//
|
||||||
|
// return 1
|
||||||
|
//}
|
||||||
|
|
||||||
|
// Create SCALAR.
|
||||||
|
func yaml_scalar_event_initialize(event *yaml_event_t, anchor, tag, value []byte, plain_implicit, quoted_implicit bool, style yaml_scalar_style_t) bool { |
||||||
|
*event = yaml_event_t{ |
||||||
|
typ: yaml_SCALAR_EVENT, |
||||||
|
anchor: anchor, |
||||||
|
tag: tag, |
||||||
|
value: value, |
||||||
|
implicit: plain_implicit, |
||||||
|
quoted_implicit: quoted_implicit, |
||||||
|
style: yaml_style_t(style), |
||||||
|
} |
||||||
|
return true |
||||||
|
} |
||||||
|
|
||||||
|
// Create SEQUENCE-START.
|
||||||
|
func yaml_sequence_start_event_initialize(event *yaml_event_t, anchor, tag []byte, implicit bool, style yaml_sequence_style_t) bool { |
||||||
|
*event = yaml_event_t{ |
||||||
|
typ: yaml_SEQUENCE_START_EVENT, |
||||||
|
anchor: anchor, |
||||||
|
tag: tag, |
||||||
|
implicit: implicit, |
||||||
|
style: yaml_style_t(style), |
||||||
|
} |
||||||
|
return true |
||||||
|
} |
||||||
|
|
||||||
|
// Create SEQUENCE-END.
|
||||||
|
func yaml_sequence_end_event_initialize(event *yaml_event_t) bool { |
||||||
|
*event = yaml_event_t{ |
||||||
|
typ: yaml_SEQUENCE_END_EVENT, |
||||||
|
} |
||||||
|
return true |
||||||
|
} |
||||||
|
|
||||||
|
// Create MAPPING-START.
|
||||||
|
func yaml_mapping_start_event_initialize(event *yaml_event_t, anchor, tag []byte, implicit bool, style yaml_mapping_style_t) bool { |
||||||
|
*event = yaml_event_t{ |
||||||
|
typ: yaml_MAPPING_START_EVENT, |
||||||
|
anchor: anchor, |
||||||
|
tag: tag, |
||||||
|
implicit: implicit, |
||||||
|
style: yaml_style_t(style), |
||||||
|
} |
||||||
|
return true |
||||||
|
} |
||||||
|
|
||||||
|
// Create MAPPING-END.
|
||||||
|
func yaml_mapping_end_event_initialize(event *yaml_event_t) bool { |
||||||
|
*event = yaml_event_t{ |
||||||
|
typ: yaml_MAPPING_END_EVENT, |
||||||
|
} |
||||||
|
return true |
||||||
|
} |
||||||
|
|
||||||
|
// Destroy an event object.
|
||||||
|
func yaml_event_delete(event *yaml_event_t) { |
||||||
|
*event = yaml_event_t{} |
||||||
|
} |
||||||
|
|
||||||
|
///*
|
||||||
|
// * Create a document object.
|
||||||
|
// */
|
||||||
|
//
|
||||||
|
//YAML_DECLARE(int)
|
||||||
|
//yaml_document_initialize(document *yaml_document_t,
|
||||||
|
// version_directive *yaml_version_directive_t,
|
||||||
|
// tag_directives_start *yaml_tag_directive_t,
|
||||||
|
// tag_directives_end *yaml_tag_directive_t,
|
||||||
|
// start_implicit int, end_implicit int)
|
||||||
|
//{
|
||||||
|
// struct {
|
||||||
|
// error yaml_error_type_t
|
||||||
|
// } context
|
||||||
|
// struct {
|
||||||
|
// start *yaml_node_t
|
||||||
|
// end *yaml_node_t
|
||||||
|
// top *yaml_node_t
|
||||||
|
// } nodes = { NULL, NULL, NULL }
|
||||||
|
// version_directive_copy *yaml_version_directive_t = NULL
|
||||||
|
// struct {
|
||||||
|
// start *yaml_tag_directive_t
|
||||||
|
// end *yaml_tag_directive_t
|
||||||
|
// top *yaml_tag_directive_t
|
||||||
|
// } tag_directives_copy = { NULL, NULL, NULL }
|
||||||
|
// value yaml_tag_directive_t = { NULL, NULL }
|
||||||
|
// mark yaml_mark_t = { 0, 0, 0 }
|
||||||
|
//
|
||||||
|
// assert(document) // Non-NULL document object is expected.
|
||||||
|
// assert((tag_directives_start && tag_directives_end) ||
|
||||||
|
// (tag_directives_start == tag_directives_end))
|
||||||
|
// // Valid tag directives are expected.
|
||||||
|
//
|
||||||
|
// if (!STACK_INIT(&context, nodes, INITIAL_STACK_SIZE)) goto error
|
||||||
|
//
|
||||||
|
// if (version_directive) {
|
||||||
|
// version_directive_copy = yaml_malloc(sizeof(yaml_version_directive_t))
|
||||||
|
// if (!version_directive_copy) goto error
|
||||||
|
// version_directive_copy.major = version_directive.major
|
||||||
|
// version_directive_copy.minor = version_directive.minor
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// if (tag_directives_start != tag_directives_end) {
|
||||||
|
// tag_directive *yaml_tag_directive_t
|
||||||
|
// if (!STACK_INIT(&context, tag_directives_copy, INITIAL_STACK_SIZE))
|
||||||
|
// goto error
|
||||||
|
// for (tag_directive = tag_directives_start
|
||||||
|
// tag_directive != tag_directives_end; tag_directive ++) {
|
||||||
|
// assert(tag_directive.handle)
|
||||||
|
// assert(tag_directive.prefix)
|
||||||
|
// if (!yaml_check_utf8(tag_directive.handle,
|
||||||
|
// strlen((char *)tag_directive.handle)))
|
||||||
|
// goto error
|
||||||
|
// if (!yaml_check_utf8(tag_directive.prefix,
|
||||||
|
// strlen((char *)tag_directive.prefix)))
|
||||||
|
// goto error
|
||||||
|
// value.handle = yaml_strdup(tag_directive.handle)
|
||||||
|
// value.prefix = yaml_strdup(tag_directive.prefix)
|
||||||
|
// if (!value.handle || !value.prefix) goto error
|
||||||
|
// if (!PUSH(&context, tag_directives_copy, value))
|
||||||
|
// goto error
|
||||||
|
// value.handle = NULL
|
||||||
|
// value.prefix = NULL
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// DOCUMENT_INIT(*document, nodes.start, nodes.end, version_directive_copy,
|
||||||
|
// tag_directives_copy.start, tag_directives_copy.top,
|
||||||
|
// start_implicit, end_implicit, mark, mark)
|
||||||
|
//
|
||||||
|
// return 1
|
||||||
|
//
|
||||||
|
//error:
|
||||||
|
// STACK_DEL(&context, nodes)
|
||||||
|
// yaml_free(version_directive_copy)
|
||||||
|
// while (!STACK_EMPTY(&context, tag_directives_copy)) {
|
||||||
|
// value yaml_tag_directive_t = POP(&context, tag_directives_copy)
|
||||||
|
// yaml_free(value.handle)
|
||||||
|
// yaml_free(value.prefix)
|
||||||
|
// }
|
||||||
|
// STACK_DEL(&context, tag_directives_copy)
|
||||||
|
// yaml_free(value.handle)
|
||||||
|
// yaml_free(value.prefix)
|
||||||
|
//
|
||||||
|
// return 0
|
||||||
|
//}
|
||||||
|
//
|
||||||
|
///*
|
||||||
|
// * Destroy a document object.
|
||||||
|
// */
|
||||||
|
//
|
||||||
|
//YAML_DECLARE(void)
|
||||||
|
//yaml_document_delete(document *yaml_document_t)
|
||||||
|
//{
|
||||||
|
// struct {
|
||||||
|
// error yaml_error_type_t
|
||||||
|
// } context
|
||||||
|
// tag_directive *yaml_tag_directive_t
|
||||||
|
//
|
||||||
|
// context.error = YAML_NO_ERROR // Eliminate a compliler warning.
|
||||||
|
//
|
||||||
|
// assert(document) // Non-NULL document object is expected.
|
||||||
|
//
|
||||||
|
// while (!STACK_EMPTY(&context, document.nodes)) {
|
||||||
|
// node yaml_node_t = POP(&context, document.nodes)
|
||||||
|
// yaml_free(node.tag)
|
||||||
|
// switch (node.type) {
|
||||||
|
// case YAML_SCALAR_NODE:
|
||||||
|
// yaml_free(node.data.scalar.value)
|
||||||
|
// break
|
||||||
|
// case YAML_SEQUENCE_NODE:
|
||||||
|
// STACK_DEL(&context, node.data.sequence.items)
|
||||||
|
// break
|
||||||
|
// case YAML_MAPPING_NODE:
|
||||||
|
// STACK_DEL(&context, node.data.mapping.pairs)
|
||||||
|
// break
|
||||||
|
// default:
|
||||||
|
// assert(0) // Should not happen.
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
// STACK_DEL(&context, document.nodes)
|
||||||
|
//
|
||||||
|
// yaml_free(document.version_directive)
|
||||||
|
// for (tag_directive = document.tag_directives.start
|
||||||
|
// tag_directive != document.tag_directives.end
|
||||||
|
// tag_directive++) {
|
||||||
|
// yaml_free(tag_directive.handle)
|
||||||
|
// yaml_free(tag_directive.prefix)
|
||||||
|
// }
|
||||||
|
// yaml_free(document.tag_directives.start)
|
||||||
|
//
|
||||||
|
// memset(document, 0, sizeof(yaml_document_t))
|
||||||
|
//}
|
||||||
|
//
|
||||||
|
///**
|
||||||
|
// * Get a document node.
|
||||||
|
// */
|
||||||
|
//
|
||||||
|
//YAML_DECLARE(yaml_node_t *)
|
||||||
|
//yaml_document_get_node(document *yaml_document_t, index int)
|
||||||
|
//{
|
||||||
|
// assert(document) // Non-NULL document object is expected.
|
||||||
|
//
|
||||||
|
// if (index > 0 && document.nodes.start + index <= document.nodes.top) {
|
||||||
|
// return document.nodes.start + index - 1
|
||||||
|
// }
|
||||||
|
// return NULL
|
||||||
|
//}
|
||||||
|
//
|
||||||
|
///**
|
||||||
|
// * Get the root object.
|
||||||
|
// */
|
||||||
|
//
|
||||||
|
//YAML_DECLARE(yaml_node_t *)
|
||||||
|
//yaml_document_get_root_node(document *yaml_document_t)
|
||||||
|
//{
|
||||||
|
// assert(document) // Non-NULL document object is expected.
|
||||||
|
//
|
||||||
|
// if (document.nodes.top != document.nodes.start) {
|
||||||
|
// return document.nodes.start
|
||||||
|
// }
|
||||||
|
// return NULL
|
||||||
|
//}
|
||||||
|
//
|
||||||
|
///*
|
||||||
|
// * Add a scalar node to a document.
|
||||||
|
// */
|
||||||
|
//
|
||||||
|
//YAML_DECLARE(int)
|
||||||
|
//yaml_document_add_scalar(document *yaml_document_t,
|
||||||
|
// tag *yaml_char_t, value *yaml_char_t, length int,
|
||||||
|
// style yaml_scalar_style_t)
|
||||||
|
//{
|
||||||
|
// struct {
|
||||||
|
// error yaml_error_type_t
|
||||||
|
// } context
|
||||||
|
// mark yaml_mark_t = { 0, 0, 0 }
|
||||||
|
// tag_copy *yaml_char_t = NULL
|
||||||
|
// value_copy *yaml_char_t = NULL
|
||||||
|
// node yaml_node_t
|
||||||
|
//
|
||||||
|
// assert(document) // Non-NULL document object is expected.
|
||||||
|
// assert(value) // Non-NULL value is expected.
|
||||||
|
//
|
||||||
|
// if (!tag) {
|
||||||
|
// tag = (yaml_char_t *)YAML_DEFAULT_SCALAR_TAG
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// if (!yaml_check_utf8(tag, strlen((char *)tag))) goto error
|
||||||
|
// tag_copy = yaml_strdup(tag)
|
||||||
|
// if (!tag_copy) goto error
|
||||||
|
//
|
||||||
|
// if (length < 0) {
|
||||||
|
// length = strlen((char *)value)
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// if (!yaml_check_utf8(value, length)) goto error
|
||||||
|
// value_copy = yaml_malloc(length+1)
|
||||||
|
// if (!value_copy) goto error
|
||||||
|
// memcpy(value_copy, value, length)
|
||||||
|
// value_copy[length] = '\0'
|
||||||
|
//
|
||||||
|
// SCALAR_NODE_INIT(node, tag_copy, value_copy, length, style, mark, mark)
|
||||||
|
// if (!PUSH(&context, document.nodes, node)) goto error
|
||||||
|
//
|
||||||
|
// return document.nodes.top - document.nodes.start
|
||||||
|
//
|
||||||
|
//error:
|
||||||
|
// yaml_free(tag_copy)
|
||||||
|
// yaml_free(value_copy)
|
||||||
|
//
|
||||||
|
// return 0
|
||||||
|
//}
|
||||||
|
//
|
||||||
|
///*
|
||||||
|
// * Add a sequence node to a document.
|
||||||
|
// */
|
||||||
|
//
|
||||||
|
//YAML_DECLARE(int)
|
||||||
|
//yaml_document_add_sequence(document *yaml_document_t,
|
||||||
|
// tag *yaml_char_t, style yaml_sequence_style_t)
|
||||||
|
//{
|
||||||
|
// struct {
|
||||||
|
// error yaml_error_type_t
|
||||||
|
// } context
|
||||||
|
// mark yaml_mark_t = { 0, 0, 0 }
|
||||||
|
// tag_copy *yaml_char_t = NULL
|
||||||
|
// struct {
|
||||||
|
// start *yaml_node_item_t
|
||||||
|
// end *yaml_node_item_t
|
||||||
|
// top *yaml_node_item_t
|
||||||
|
// } items = { NULL, NULL, NULL }
|
||||||
|
// node yaml_node_t
|
||||||
|
//
|
||||||
|
// assert(document) // Non-NULL document object is expected.
|
||||||
|
//
|
||||||
|
// if (!tag) {
|
||||||
|
// tag = (yaml_char_t *)YAML_DEFAULT_SEQUENCE_TAG
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// if (!yaml_check_utf8(tag, strlen((char *)tag))) goto error
|
||||||
|
// tag_copy = yaml_strdup(tag)
|
||||||
|
// if (!tag_copy) goto error
|
||||||
|
//
|
||||||
|
// if (!STACK_INIT(&context, items, INITIAL_STACK_SIZE)) goto error
|
||||||
|
//
|
||||||
|
// SEQUENCE_NODE_INIT(node, tag_copy, items.start, items.end,
|
||||||
|
// style, mark, mark)
|
||||||
|
// if (!PUSH(&context, document.nodes, node)) goto error
|
||||||
|
//
|
||||||
|
// return document.nodes.top - document.nodes.start
|
||||||
|
//
|
||||||
|
//error:
|
||||||
|
// STACK_DEL(&context, items)
|
||||||
|
// yaml_free(tag_copy)
|
||||||
|
//
|
||||||
|
// return 0
|
||||||
|
//}
|
||||||
|
//
|
||||||
|
///*
|
||||||
|
// * Add a mapping node to a document.
|
||||||
|
// */
|
||||||
|
//
|
||||||
|
//YAML_DECLARE(int)
|
||||||
|
//yaml_document_add_mapping(document *yaml_document_t,
|
||||||
|
// tag *yaml_char_t, style yaml_mapping_style_t)
|
||||||
|
//{
|
||||||
|
// struct {
|
||||||
|
// error yaml_error_type_t
|
||||||
|
// } context
|
||||||
|
// mark yaml_mark_t = { 0, 0, 0 }
|
||||||
|
// tag_copy *yaml_char_t = NULL
|
||||||
|
// struct {
|
||||||
|
// start *yaml_node_pair_t
|
||||||
|
// end *yaml_node_pair_t
|
||||||
|
// top *yaml_node_pair_t
|
||||||
|
// } pairs = { NULL, NULL, NULL }
|
||||||
|
// node yaml_node_t
|
||||||
|
//
|
||||||
|
// assert(document) // Non-NULL document object is expected.
|
||||||
|
//
|
||||||
|
// if (!tag) {
|
||||||
|
// tag = (yaml_char_t *)YAML_DEFAULT_MAPPING_TAG
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// if (!yaml_check_utf8(tag, strlen((char *)tag))) goto error
|
||||||
|
// tag_copy = yaml_strdup(tag)
|
||||||
|
// if (!tag_copy) goto error
|
||||||
|
//
|
||||||
|
// if (!STACK_INIT(&context, pairs, INITIAL_STACK_SIZE)) goto error
|
||||||
|
//
|
||||||
|
// MAPPING_NODE_INIT(node, tag_copy, pairs.start, pairs.end,
|
||||||
|
// style, mark, mark)
|
||||||
|
// if (!PUSH(&context, document.nodes, node)) goto error
|
||||||
|
//
|
||||||
|
// return document.nodes.top - document.nodes.start
|
||||||
|
//
|
||||||
|
//error:
|
||||||
|
// STACK_DEL(&context, pairs)
|
||||||
|
// yaml_free(tag_copy)
|
||||||
|
//
|
||||||
|
// return 0
|
||||||
|
//}
|
||||||
|
//
|
||||||
|
///*
|
||||||
|
// * Append an item to a sequence node.
|
||||||
|
// */
|
||||||
|
//
|
||||||
|
//YAML_DECLARE(int)
|
||||||
|
//yaml_document_append_sequence_item(document *yaml_document_t,
|
||||||
|
// sequence int, item int)
|
||||||
|
//{
|
||||||
|
// struct {
|
||||||
|
// error yaml_error_type_t
|
||||||
|
// } context
|
||||||
|
//
|
||||||
|
// assert(document) // Non-NULL document is required.
|
||||||
|
// assert(sequence > 0
|
||||||
|
// && document.nodes.start + sequence <= document.nodes.top)
|
||||||
|
// // Valid sequence id is required.
|
||||||
|
// assert(document.nodes.start[sequence-1].type == YAML_SEQUENCE_NODE)
|
||||||
|
// // A sequence node is required.
|
||||||
|
// assert(item > 0 && document.nodes.start + item <= document.nodes.top)
|
||||||
|
// // Valid item id is required.
|
||||||
|
//
|
||||||
|
// if (!PUSH(&context,
|
||||||
|
// document.nodes.start[sequence-1].data.sequence.items, item))
|
||||||
|
// return 0
|
||||||
|
//
|
||||||
|
// return 1
|
||||||
|
//}
|
||||||
|
//
|
||||||
|
///*
|
||||||
|
// * Append a pair of a key and a value to a mapping node.
|
||||||
|
// */
|
||||||
|
//
|
||||||
|
//YAML_DECLARE(int)
|
||||||
|
//yaml_document_append_mapping_pair(document *yaml_document_t,
|
||||||
|
// mapping int, key int, value int)
|
||||||
|
//{
|
||||||
|
// struct {
|
||||||
|
// error yaml_error_type_t
|
||||||
|
// } context
|
||||||
|
//
|
||||||
|
// pair yaml_node_pair_t
|
||||||
|
//
|
||||||
|
// assert(document) // Non-NULL document is required.
|
||||||
|
// assert(mapping > 0
|
||||||
|
// && document.nodes.start + mapping <= document.nodes.top)
|
||||||
|
// // Valid mapping id is required.
|
||||||
|
// assert(document.nodes.start[mapping-1].type == YAML_MAPPING_NODE)
|
||||||
|
// // A mapping node is required.
|
||||||
|
// assert(key > 0 && document.nodes.start + key <= document.nodes.top)
|
||||||
|
// // Valid key id is required.
|
||||||
|
// assert(value > 0 && document.nodes.start + value <= document.nodes.top)
|
||||||
|
// // Valid value id is required.
|
||||||
|
//
|
||||||
|
// pair.key = key
|
||||||
|
// pair.value = value
|
||||||
|
//
|
||||||
|
// if (!PUSH(&context,
|
||||||
|
// document.nodes.start[mapping-1].data.mapping.pairs, pair))
|
||||||
|
// return 0
|
||||||
|
//
|
||||||
|
// return 1
|
||||||
|
//}
|
||||||
|
//
|
||||||
|
//
|
@ -0,0 +1,683 @@ |
|||||||
|
package yaml |
||||||
|
|
||||||
|
import ( |
||||||
|
"encoding" |
||||||
|
"encoding/base64" |
||||||
|
"fmt" |
||||||
|
"math" |
||||||
|
"reflect" |
||||||
|
"strconv" |
||||||
|
"time" |
||||||
|
) |
||||||
|
|
||||||
|
const ( |
||||||
|
documentNode = 1 << iota |
||||||
|
mappingNode |
||||||
|
sequenceNode |
||||||
|
scalarNode |
||||||
|
aliasNode |
||||||
|
) |
||||||
|
|
||||||
|
type node struct { |
||||||
|
kind int |
||||||
|
line, column int |
||||||
|
tag string |
||||||
|
value string |
||||||
|
implicit bool |
||||||
|
children []*node |
||||||
|
anchors map[string]*node |
||||||
|
} |
||||||
|
|
||||||
|
// ----------------------------------------------------------------------------
|
||||||
|
// Parser, produces a node tree out of a libyaml event stream.
|
||||||
|
|
||||||
|
type parser struct { |
||||||
|
parser yaml_parser_t |
||||||
|
event yaml_event_t |
||||||
|
doc *node |
||||||
|
} |
||||||
|
|
||||||
|
func newParser(b []byte) *parser { |
||||||
|
p := parser{} |
||||||
|
if !yaml_parser_initialize(&p.parser) { |
||||||
|
panic("failed to initialize YAML emitter") |
||||||
|
} |
||||||
|
|
||||||
|
if len(b) == 0 { |
||||||
|
b = []byte{'\n'} |
||||||
|
} |
||||||
|
|
||||||
|
yaml_parser_set_input_string(&p.parser, b) |
||||||
|
|
||||||
|
p.skip() |
||||||
|
if p.event.typ != yaml_STREAM_START_EVENT { |
||||||
|
panic("expected stream start event, got " + strconv.Itoa(int(p.event.typ))) |
||||||
|
} |
||||||
|
p.skip() |
||||||
|
return &p |
||||||
|
} |
||||||
|
|
||||||
|
func (p *parser) destroy() { |
||||||
|
if p.event.typ != yaml_NO_EVENT { |
||||||
|
yaml_event_delete(&p.event) |
||||||
|
} |
||||||
|
yaml_parser_delete(&p.parser) |
||||||
|
} |
||||||
|
|
||||||
|
func (p *parser) skip() { |
||||||
|
if p.event.typ != yaml_NO_EVENT { |
||||||
|
if p.event.typ == yaml_STREAM_END_EVENT { |
||||||
|
failf("attempted to go past the end of stream; corrupted value?") |
||||||
|
} |
||||||
|
yaml_event_delete(&p.event) |
||||||
|
} |
||||||
|
if !yaml_parser_parse(&p.parser, &p.event) { |
||||||
|
p.fail() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
func (p *parser) fail() { |
||||||
|
var where string |
||||||
|
var line int |
||||||
|
if p.parser.problem_mark.line != 0 { |
||||||
|
line = p.parser.problem_mark.line |
||||||
|
} else if p.parser.context_mark.line != 0 { |
||||||
|
line = p.parser.context_mark.line |
||||||
|
} |
||||||
|
if line != 0 { |
||||||
|
where = "line " + strconv.Itoa(line) + ": " |
||||||
|
} |
||||||
|
var msg string |
||||||
|
if len(p.parser.problem) > 0 { |
||||||
|
msg = p.parser.problem |
||||||
|
} else { |
||||||
|
msg = "unknown problem parsing YAML content" |
||||||
|
} |
||||||
|
failf("%s%s", where, msg) |
||||||
|
} |
||||||
|
|
||||||
|
func (p *parser) anchor(n *node, anchor []byte) { |
||||||
|
if anchor != nil { |
||||||
|
p.doc.anchors[string(anchor)] = n |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
func (p *parser) parse() *node { |
||||||
|
switch p.event.typ { |
||||||
|
case yaml_SCALAR_EVENT: |
||||||
|
return p.scalar() |
||||||
|
case yaml_ALIAS_EVENT: |
||||||
|
return p.alias() |
||||||
|
case yaml_MAPPING_START_EVENT: |
||||||
|
return p.mapping() |
||||||
|
case yaml_SEQUENCE_START_EVENT: |
||||||
|
return p.sequence() |
||||||
|
case yaml_DOCUMENT_START_EVENT: |
||||||
|
return p.document() |
||||||
|
case yaml_STREAM_END_EVENT: |
||||||
|
// Happens when attempting to decode an empty buffer.
|
||||||
|
return nil |
||||||
|
default: |
||||||
|
panic("attempted to parse unknown event: " + strconv.Itoa(int(p.event.typ))) |
||||||
|
} |
||||||
|
panic("unreachable") |
||||||
|
} |
||||||
|
|
||||||
|
func (p *parser) node(kind int) *node { |
||||||
|
return &node{ |
||||||
|
kind: kind, |
||||||
|
line: p.event.start_mark.line, |
||||||
|
column: p.event.start_mark.column, |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
func (p *parser) document() *node { |
||||||
|
n := p.node(documentNode) |
||||||
|
n.anchors = make(map[string]*node) |
||||||
|
p.doc = n |
||||||
|
p.skip() |
||||||
|
n.children = append(n.children, p.parse()) |
||||||
|
if p.event.typ != yaml_DOCUMENT_END_EVENT { |
||||||
|
panic("expected end of document event but got " + strconv.Itoa(int(p.event.typ))) |
||||||
|
} |
||||||
|
p.skip() |
||||||
|
return n |
||||||
|
} |
||||||
|
|
||||||
|
func (p *parser) alias() *node { |
||||||
|
n := p.node(aliasNode) |
||||||
|
n.value = string(p.event.anchor) |
||||||
|
p.skip() |
||||||
|
return n |
||||||
|
} |
||||||
|
|
||||||
|
func (p *parser) scalar() *node { |
||||||
|
n := p.node(scalarNode) |
||||||
|
n.value = string(p.event.value) |
||||||
|
n.tag = string(p.event.tag) |
||||||
|
n.implicit = p.event.implicit |
||||||
|
p.anchor(n, p.event.anchor) |
||||||
|
p.skip() |
||||||
|
return n |
||||||
|
} |
||||||
|
|
||||||
|
func (p *parser) sequence() *node { |
||||||
|
n := p.node(sequenceNode) |
||||||
|
p.anchor(n, p.event.anchor) |
||||||
|
p.skip() |
||||||
|
for p.event.typ != yaml_SEQUENCE_END_EVENT { |
||||||
|
n.children = append(n.children, p.parse()) |
||||||
|
} |
||||||
|
p.skip() |
||||||
|
return n |
||||||
|
} |
||||||
|
|
||||||
|
func (p *parser) mapping() *node { |
||||||
|
n := p.node(mappingNode) |
||||||
|
p.anchor(n, p.event.anchor) |
||||||
|
p.skip() |
||||||
|
for p.event.typ != yaml_MAPPING_END_EVENT { |
||||||
|
n.children = append(n.children, p.parse(), p.parse()) |
||||||
|
} |
||||||
|
p.skip() |
||||||
|
return n |
||||||
|
} |
||||||
|
|
||||||
|
// ----------------------------------------------------------------------------
|
||||||
|
// Decoder, unmarshals a node into a provided value.
|
||||||
|
|
||||||
|
type decoder struct { |
||||||
|
doc *node |
||||||
|
aliases map[string]bool |
||||||
|
mapType reflect.Type |
||||||
|
terrors []string |
||||||
|
} |
||||||
|
|
||||||
|
var ( |
||||||
|
mapItemType = reflect.TypeOf(MapItem{}) |
||||||
|
durationType = reflect.TypeOf(time.Duration(0)) |
||||||
|
defaultMapType = reflect.TypeOf(map[interface{}]interface{}{}) |
||||||
|
ifaceType = defaultMapType.Elem() |
||||||
|
) |
||||||
|
|
||||||
|
func newDecoder() *decoder { |
||||||
|
d := &decoder{mapType: defaultMapType} |
||||||
|
d.aliases = make(map[string]bool) |
||||||
|
return d |
||||||
|
} |
||||||
|
|
||||||
|
func (d *decoder) terror(n *node, tag string, out reflect.Value) { |
||||||
|
if n.tag != "" { |
||||||
|
tag = n.tag |
||||||
|
} |
||||||
|
value := n.value |
||||||
|
if tag != yaml_SEQ_TAG && tag != yaml_MAP_TAG { |
||||||
|
if len(value) > 10 { |
||||||
|
value = " `" + value[:7] + "...`" |
||||||
|
} else { |
||||||
|
value = " `" + value + "`" |
||||||
|
} |
||||||
|
} |
||||||
|
d.terrors = append(d.terrors, fmt.Sprintf("line %d: cannot unmarshal %s%s into %s", n.line+1, shortTag(tag), value, out.Type())) |
||||||
|
} |
||||||
|
|
||||||
|
func (d *decoder) callUnmarshaler(n *node, u Unmarshaler) (good bool) { |
||||||
|
terrlen := len(d.terrors) |
||||||
|
err := u.UnmarshalYAML(func(v interface{}) (err error) { |
||||||
|
defer handleErr(&err) |
||||||
|
d.unmarshal(n, reflect.ValueOf(v)) |
||||||
|
if len(d.terrors) > terrlen { |
||||||
|
issues := d.terrors[terrlen:] |
||||||
|
d.terrors = d.terrors[:terrlen] |
||||||
|
return &TypeError{issues} |
||||||
|
} |
||||||
|
return nil |
||||||
|
}) |
||||||
|
if e, ok := err.(*TypeError); ok { |
||||||
|
d.terrors = append(d.terrors, e.Errors...) |
||||||
|
return false |
||||||
|
} |
||||||
|
if err != nil { |
||||||
|
fail(err) |
||||||
|
} |
||||||
|
return true |
||||||
|
} |
||||||
|
|
||||||
|
// d.prepare initializes and dereferences pointers and calls UnmarshalYAML
|
||||||
|
// if a value is found to implement it.
|
||||||
|
// It returns the initialized and dereferenced out value, whether
|
||||||
|
// unmarshalling was already done by UnmarshalYAML, and if so whether
|
||||||
|
// its types unmarshalled appropriately.
|
||||||
|
//
|
||||||
|
// If n holds a null value, prepare returns before doing anything.
|
||||||
|
func (d *decoder) prepare(n *node, out reflect.Value) (newout reflect.Value, unmarshaled, good bool) { |
||||||
|
if n.tag == yaml_NULL_TAG || n.kind == scalarNode && n.tag == "" && (n.value == "null" || n.value == "" && n.implicit) { |
||||||
|
return out, false, false |
||||||
|
} |
||||||
|
again := true |
||||||
|
for again { |
||||||
|
again = false |
||||||
|
if out.Kind() == reflect.Ptr { |
||||||
|
if out.IsNil() { |
||||||
|
out.Set(reflect.New(out.Type().Elem())) |
||||||
|
} |
||||||
|
out = out.Elem() |
||||||
|
again = true |
||||||
|
} |
||||||
|
if out.CanAddr() { |
||||||
|
if u, ok := out.Addr().Interface().(Unmarshaler); ok { |
||||||
|
good = d.callUnmarshaler(n, u) |
||||||
|
return out, true, good |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
return out, false, false |
||||||
|
} |
||||||
|
|
||||||
|
func (d *decoder) unmarshal(n *node, out reflect.Value) (good bool) { |
||||||
|
switch n.kind { |
||||||
|
case documentNode: |
||||||
|
return d.document(n, out) |
||||||
|
case aliasNode: |
||||||
|
return d.alias(n, out) |
||||||
|
} |
||||||
|
out, unmarshaled, good := d.prepare(n, out) |
||||||
|
if unmarshaled { |
||||||
|
return good |
||||||
|
} |
||||||
|
switch n.kind { |
||||||
|
case scalarNode: |
||||||
|
good = d.scalar(n, out) |
||||||
|
case mappingNode: |
||||||
|
good = d.mapping(n, out) |
||||||
|
case sequenceNode: |
||||||
|
good = d.sequence(n, out) |
||||||
|
default: |
||||||
|
panic("internal error: unknown node kind: " + strconv.Itoa(n.kind)) |
||||||
|
} |
||||||
|
return good |
||||||
|
} |
||||||
|
|
||||||
|
func (d *decoder) document(n *node, out reflect.Value) (good bool) { |
||||||
|
if len(n.children) == 1 { |
||||||
|
d.doc = n |
||||||
|
d.unmarshal(n.children[0], out) |
||||||
|
return true |
||||||
|
} |
||||||
|
return false |
||||||
|
} |
||||||
|
|
||||||
|
func (d *decoder) alias(n *node, out reflect.Value) (good bool) { |
||||||
|
an, ok := d.doc.anchors[n.value] |
||||||
|
if !ok { |
||||||
|
failf("unknown anchor '%s' referenced", n.value) |
||||||
|
} |
||||||
|
if d.aliases[n.value] { |
||||||
|
failf("anchor '%s' value contains itself", n.value) |
||||||
|
} |
||||||
|
d.aliases[n.value] = true |
||||||
|
good = d.unmarshal(an, out) |
||||||
|
delete(d.aliases, n.value) |
||||||
|
return good |
||||||
|
} |
||||||
|
|
||||||
|
var zeroValue reflect.Value |
||||||
|
|
||||||
|
func resetMap(out reflect.Value) { |
||||||
|
for _, k := range out.MapKeys() { |
||||||
|
out.SetMapIndex(k, zeroValue) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
func (d *decoder) scalar(n *node, out reflect.Value) (good bool) { |
||||||
|
var tag string |
||||||
|
var resolved interface{} |
||||||
|
if n.tag == "" && !n.implicit { |
||||||
|
tag = yaml_STR_TAG |
||||||
|
resolved = n.value |
||||||
|
} else { |
||||||
|
tag, resolved = resolve(n.tag, n.value) |
||||||
|
if tag == yaml_BINARY_TAG { |
||||||
|
data, err := base64.StdEncoding.DecodeString(resolved.(string)) |
||||||
|
if err != nil { |
||||||
|
failf("!!binary value contains invalid base64 data") |
||||||
|
} |
||||||
|
resolved = string(data) |
||||||
|
} |
||||||
|
} |
||||||
|
if resolved == nil { |
||||||
|
if out.Kind() == reflect.Map && !out.CanAddr() { |
||||||
|
resetMap(out) |
||||||
|
} else { |
||||||
|
out.Set(reflect.Zero(out.Type())) |
||||||
|
} |
||||||
|
return true |
||||||
|
} |
||||||
|
if s, ok := resolved.(string); ok && out.CanAddr() { |
||||||
|
if u, ok := out.Addr().Interface().(encoding.TextUnmarshaler); ok { |
||||||
|
err := u.UnmarshalText([]byte(s)) |
||||||
|
if err != nil { |
||||||
|
fail(err) |
||||||
|
} |
||||||
|
return true |
||||||
|
} |
||||||
|
} |
||||||
|
switch out.Kind() { |
||||||
|
case reflect.String: |
||||||
|
if tag == yaml_BINARY_TAG { |
||||||
|
out.SetString(resolved.(string)) |
||||||
|
good = true |
||||||
|
} else if resolved != nil { |
||||||
|
out.SetString(n.value) |
||||||
|
good = true |
||||||
|
} |
||||||
|
case reflect.Interface: |
||||||
|
if resolved == nil { |
||||||
|
out.Set(reflect.Zero(out.Type())) |
||||||
|
} else { |
||||||
|
out.Set(reflect.ValueOf(resolved)) |
||||||
|
} |
||||||
|
good = true |
||||||
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64: |
||||||
|
switch resolved := resolved.(type) { |
||||||
|
case int: |
||||||
|
if !out.OverflowInt(int64(resolved)) { |
||||||
|
out.SetInt(int64(resolved)) |
||||||
|
good = true |
||||||
|
} |
||||||
|
case int64: |
||||||
|
if !out.OverflowInt(resolved) { |
||||||
|
out.SetInt(resolved) |
||||||
|
good = true |
||||||
|
} |
||||||
|
case uint64: |
||||||
|
if resolved <= math.MaxInt64 && !out.OverflowInt(int64(resolved)) { |
||||||
|
out.SetInt(int64(resolved)) |
||||||
|
good = true |
||||||
|
} |
||||||
|
case float64: |
||||||
|
if resolved <= math.MaxInt64 && !out.OverflowInt(int64(resolved)) { |
||||||
|
out.SetInt(int64(resolved)) |
||||||
|
good = true |
||||||
|
} |
||||||
|
case string: |
||||||
|
if out.Type() == durationType { |
||||||
|
d, err := time.ParseDuration(resolved) |
||||||
|
if err == nil { |
||||||
|
out.SetInt(int64(d)) |
||||||
|
good = true |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, reflect.Uintptr: |
||||||
|
switch resolved := resolved.(type) { |
||||||
|
case int: |
||||||
|
if resolved >= 0 && !out.OverflowUint(uint64(resolved)) { |
||||||
|
out.SetUint(uint64(resolved)) |
||||||
|
good = true |
||||||
|
} |
||||||
|
case int64: |
||||||
|
if resolved >= 0 && !out.OverflowUint(uint64(resolved)) { |
||||||
|
out.SetUint(uint64(resolved)) |
||||||
|
good = true |
||||||
|
} |
||||||
|
case uint64: |
||||||
|
if !out.OverflowUint(uint64(resolved)) { |
||||||
|
out.SetUint(uint64(resolved)) |
||||||
|
good = true |
||||||
|
} |
||||||
|
case float64: |
||||||
|
if resolved <= math.MaxUint64 && !out.OverflowUint(uint64(resolved)) { |
||||||
|
out.SetUint(uint64(resolved)) |
||||||
|
good = true |
||||||
|
} |
||||||
|
} |
||||||
|
case reflect.Bool: |
||||||
|
switch resolved := resolved.(type) { |
||||||
|
case bool: |
||||||
|
out.SetBool(resolved) |
||||||
|
good = true |
||||||
|
} |
||||||
|
case reflect.Float32, reflect.Float64: |
||||||
|
switch resolved := resolved.(type) { |
||||||
|
case int: |
||||||
|
out.SetFloat(float64(resolved)) |
||||||
|
good = true |
||||||
|
case int64: |
||||||
|
out.SetFloat(float64(resolved)) |
||||||
|
good = true |
||||||
|
case uint64: |
||||||
|
out.SetFloat(float64(resolved)) |
||||||
|
good = true |
||||||
|
case float64: |
||||||
|
out.SetFloat(resolved) |
||||||
|
good = true |
||||||
|
} |
||||||
|
case reflect.Ptr: |
||||||
|
if out.Type().Elem() == reflect.TypeOf(resolved) { |
||||||
|
// TODO DOes this make sense? When is out a Ptr except when decoding a nil value?
|
||||||
|
elem := reflect.New(out.Type().Elem()) |
||||||
|
elem.Elem().Set(reflect.ValueOf(resolved)) |
||||||
|
out.Set(elem) |
||||||
|
good = true |
||||||
|
} |
||||||
|
} |
||||||
|
if !good { |
||||||
|
d.terror(n, tag, out) |
||||||
|
} |
||||||
|
return good |
||||||
|
} |
||||||
|
|
||||||
|
func settableValueOf(i interface{}) reflect.Value { |
||||||
|
v := reflect.ValueOf(i) |
||||||
|
sv := reflect.New(v.Type()).Elem() |
||||||
|
sv.Set(v) |
||||||
|
return sv |
||||||
|
} |
||||||
|
|
||||||
|
func (d *decoder) sequence(n *node, out reflect.Value) (good bool) { |
||||||
|
l := len(n.children) |
||||||
|
|
||||||
|
var iface reflect.Value |
||||||
|
switch out.Kind() { |
||||||
|
case reflect.Slice: |
||||||
|
out.Set(reflect.MakeSlice(out.Type(), l, l)) |
||||||
|
case reflect.Interface: |
||||||
|
// No type hints. Will have to use a generic sequence.
|
||||||
|
iface = out |
||||||
|
out = settableValueOf(make([]interface{}, l)) |
||||||
|
default: |
||||||
|
d.terror(n, yaml_SEQ_TAG, out) |
||||||
|
return false |
||||||
|
} |
||||||
|
et := out.Type().Elem() |
||||||
|
|
||||||
|
j := 0 |
||||||
|
for i := 0; i < l; i++ { |
||||||
|
e := reflect.New(et).Elem() |
||||||
|
if ok := d.unmarshal(n.children[i], e); ok { |
||||||
|
out.Index(j).Set(e) |
||||||
|
j++ |
||||||
|
} |
||||||
|
} |
||||||
|
out.Set(out.Slice(0, j)) |
||||||
|
if iface.IsValid() { |
||||||
|
iface.Set(out) |
||||||
|
} |
||||||
|
return true |
||||||
|
} |
||||||
|
|
||||||
|
func (d *decoder) mapping(n *node, out reflect.Value) (good bool) { |
||||||
|
switch out.Kind() { |
||||||
|
case reflect.Struct: |
||||||
|
return d.mappingStruct(n, out) |
||||||
|
case reflect.Slice: |
||||||
|
return d.mappingSlice(n, out) |
||||||
|
case reflect.Map: |
||||||
|
// okay
|
||||||
|
case reflect.Interface: |
||||||
|
if d.mapType.Kind() == reflect.Map { |
||||||
|
iface := out |
||||||
|
out = reflect.MakeMap(d.mapType) |
||||||
|
iface.Set(out) |
||||||
|
} else { |
||||||
|
slicev := reflect.New(d.mapType).Elem() |
||||||
|
if !d.mappingSlice(n, slicev) { |
||||||
|
return false |
||||||
|
} |
||||||
|
out.Set(slicev) |
||||||
|
return true |
||||||
|
} |
||||||
|
default: |
||||||
|
d.terror(n, yaml_MAP_TAG, out) |
||||||
|
return false |
||||||
|
} |
||||||
|
outt := out.Type() |
||||||
|
kt := outt.Key() |
||||||
|
et := outt.Elem() |
||||||
|
|
||||||
|
mapType := d.mapType |
||||||
|
if outt.Key() == ifaceType && outt.Elem() == ifaceType { |
||||||
|
d.mapType = outt |
||||||
|
} |
||||||
|
|
||||||
|
if out.IsNil() { |
||||||
|
out.Set(reflect.MakeMap(outt)) |
||||||
|
} |
||||||
|
l := len(n.children) |
||||||
|
for i := 0; i < l; i += 2 { |
||||||
|
if isMerge(n.children[i]) { |
||||||
|
d.merge(n.children[i+1], out) |
||||||
|
continue |
||||||
|
} |
||||||
|
k := reflect.New(kt).Elem() |
||||||
|
if d.unmarshal(n.children[i], k) { |
||||||
|
kkind := k.Kind() |
||||||
|
if kkind == reflect.Interface { |
||||||
|
kkind = k.Elem().Kind() |
||||||
|
} |
||||||
|
if kkind == reflect.Map || kkind == reflect.Slice { |
||||||
|
failf("invalid map key: %#v", k.Interface()) |
||||||
|
} |
||||||
|
e := reflect.New(et).Elem() |
||||||
|
if d.unmarshal(n.children[i+1], e) { |
||||||
|
out.SetMapIndex(k, e) |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
d.mapType = mapType |
||||||
|
return true |
||||||
|
} |
||||||
|
|
||||||
|
func (d *decoder) mappingSlice(n *node, out reflect.Value) (good bool) { |
||||||
|
outt := out.Type() |
||||||
|
if outt.Elem() != mapItemType { |
||||||
|
d.terror(n, yaml_MAP_TAG, out) |
||||||
|
return false |
||||||
|
} |
||||||
|
|
||||||
|
mapType := d.mapType |
||||||
|
d.mapType = outt |
||||||
|
|
||||||
|
var slice []MapItem |
||||||
|
var l = len(n.children) |
||||||
|
for i := 0; i < l; i += 2 { |
||||||
|
if isMerge(n.children[i]) { |
||||||
|
d.merge(n.children[i+1], out) |
||||||
|
continue |
||||||
|
} |
||||||
|
item := MapItem{} |
||||||
|
k := reflect.ValueOf(&item.Key).Elem() |
||||||
|
if d.unmarshal(n.children[i], k) { |
||||||
|
v := reflect.ValueOf(&item.Value).Elem() |
||||||
|
if d.unmarshal(n.children[i+1], v) { |
||||||
|
slice = append(slice, item) |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
out.Set(reflect.ValueOf(slice)) |
||||||
|
d.mapType = mapType |
||||||
|
return true |
||||||
|
} |
||||||
|
|
||||||
|
func (d *decoder) mappingStruct(n *node, out reflect.Value) (good bool) { |
||||||
|
sinfo, err := getStructInfo(out.Type()) |
||||||
|
if err != nil { |
||||||
|
panic(err) |
||||||
|
} |
||||||
|
name := settableValueOf("") |
||||||
|
l := len(n.children) |
||||||
|
|
||||||
|
var inlineMap reflect.Value |
||||||
|
var elemType reflect.Type |
||||||
|
if sinfo.InlineMap != -1 { |
||||||
|
inlineMap = out.Field(sinfo.InlineMap) |
||||||
|
inlineMap.Set(reflect.New(inlineMap.Type()).Elem()) |
||||||
|
elemType = inlineMap.Type().Elem() |
||||||
|
} |
||||||
|
|
||||||
|
for i := 0; i < l; i += 2 { |
||||||
|
ni := n.children[i] |
||||||
|
if isMerge(ni) { |
||||||
|
d.merge(n.children[i+1], out) |
||||||
|
continue |
||||||
|
} |
||||||
|
if !d.unmarshal(ni, name) { |
||||||
|
continue |
||||||
|
} |
||||||
|
if info, ok := sinfo.FieldsMap[name.String()]; ok { |
||||||
|
var field reflect.Value |
||||||
|
if info.Inline == nil { |
||||||
|
field = out.Field(info.Num) |
||||||
|
} else { |
||||||
|
field = out.FieldByIndex(info.Inline) |
||||||
|
} |
||||||
|
d.unmarshal(n.children[i+1], field) |
||||||
|
} else if sinfo.InlineMap != -1 { |
||||||
|
if inlineMap.IsNil() { |
||||||
|
inlineMap.Set(reflect.MakeMap(inlineMap.Type())) |
||||||
|
} |
||||||
|
value := reflect.New(elemType).Elem() |
||||||
|
d.unmarshal(n.children[i+1], value) |
||||||
|
inlineMap.SetMapIndex(name, value) |
||||||
|
} |
||||||
|
} |
||||||
|
return true |
||||||
|
} |
||||||
|
|
||||||
|
func failWantMap() { |
||||||
|
failf("map merge requires map or sequence of maps as the value") |
||||||
|
} |
||||||
|
|
||||||
|
func (d *decoder) merge(n *node, out reflect.Value) { |
||||||
|
switch n.kind { |
||||||
|
case mappingNode: |
||||||
|
d.unmarshal(n, out) |
||||||
|
case aliasNode: |
||||||
|
an, ok := d.doc.anchors[n.value] |
||||||
|
if ok && an.kind != mappingNode { |
||||||
|
failWantMap() |
||||||
|
} |
||||||
|
d.unmarshal(n, out) |
||||||
|
case sequenceNode: |
||||||
|
// Step backwards as earlier nodes take precedence.
|
||||||
|
for i := len(n.children) - 1; i >= 0; i-- { |
||||||
|
ni := n.children[i] |
||||||
|
if ni.kind == aliasNode { |
||||||
|
an, ok := d.doc.anchors[ni.value] |
||||||
|
if ok && an.kind != mappingNode { |
||||||
|
failWantMap() |
||||||
|
} |
||||||
|
} else if ni.kind != mappingNode { |
||||||
|
failWantMap() |
||||||
|
} |
||||||
|
d.unmarshal(ni, out) |
||||||
|
} |
||||||
|
default: |
||||||
|
failWantMap() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
func isMerge(n *node) bool { |
||||||
|
return n.kind == scalarNode && n.value == "<<" && (n.implicit == true || n.tag == yaml_MERGE_TAG) |
||||||
|
} |
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,306 @@ |
|||||||
|
package yaml |
||||||
|
|
||||||
|
import ( |
||||||
|
"encoding" |
||||||
|
"fmt" |
||||||
|
"reflect" |
||||||
|
"regexp" |
||||||
|
"sort" |
||||||
|
"strconv" |
||||||
|
"strings" |
||||||
|
"time" |
||||||
|
) |
||||||
|
|
||||||
|
type encoder struct { |
||||||
|
emitter yaml_emitter_t |
||||||
|
event yaml_event_t |
||||||
|
out []byte |
||||||
|
flow bool |
||||||
|
} |
||||||
|
|
||||||
|
func newEncoder() (e *encoder) { |
||||||
|
e = &encoder{} |
||||||
|
e.must(yaml_emitter_initialize(&e.emitter)) |
||||||
|
yaml_emitter_set_output_string(&e.emitter, &e.out) |
||||||
|
yaml_emitter_set_unicode(&e.emitter, true) |
||||||
|
e.must(yaml_stream_start_event_initialize(&e.event, yaml_UTF8_ENCODING)) |
||||||
|
e.emit() |
||||||
|
e.must(yaml_document_start_event_initialize(&e.event, nil, nil, true)) |
||||||
|
e.emit() |
||||||
|
return e |
||||||
|
} |
||||||
|
|
||||||
|
func (e *encoder) finish() { |
||||||
|
e.must(yaml_document_end_event_initialize(&e.event, true)) |
||||||
|
e.emit() |
||||||
|
e.emitter.open_ended = false |
||||||
|
e.must(yaml_stream_end_event_initialize(&e.event)) |
||||||
|
e.emit() |
||||||
|
} |
||||||
|
|
||||||
|
func (e *encoder) destroy() { |
||||||
|
yaml_emitter_delete(&e.emitter) |
||||||
|
} |
||||||
|
|
||||||
|
func (e *encoder) emit() { |
||||||
|
// This will internally delete the e.event value.
|
||||||
|
if !yaml_emitter_emit(&e.emitter, &e.event) && e.event.typ != yaml_DOCUMENT_END_EVENT && e.event.typ != yaml_STREAM_END_EVENT { |
||||||
|
e.must(false) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
func (e *encoder) must(ok bool) { |
||||||
|
if !ok { |
||||||
|
msg := e.emitter.problem |
||||||
|
if msg == "" { |
||||||
|
msg = "unknown problem generating YAML content" |
||||||
|
} |
||||||
|
failf("%s", msg) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
func (e *encoder) marshal(tag string, in reflect.Value) { |
||||||
|
if !in.IsValid() { |
||||||
|
e.nilv() |
||||||
|
return |
||||||
|
} |
||||||
|
iface := in.Interface() |
||||||
|
if m, ok := iface.(Marshaler); ok { |
||||||
|
v, err := m.MarshalYAML() |
||||||
|
if err != nil { |
||||||
|
fail(err) |
||||||
|
} |
||||||
|
if v == nil { |
||||||
|
e.nilv() |
||||||
|
return |
||||||
|
} |
||||||
|
in = reflect.ValueOf(v) |
||||||
|
} else if m, ok := iface.(encoding.TextMarshaler); ok { |
||||||
|
text, err := m.MarshalText() |
||||||
|
if err != nil { |
||||||
|
fail(err) |
||||||
|
} |
||||||
|
in = reflect.ValueOf(string(text)) |
||||||
|
} |
||||||
|
switch in.Kind() { |
||||||
|
case reflect.Interface: |
||||||
|
if in.IsNil() { |
||||||
|
e.nilv() |
||||||
|
} else { |
||||||
|
e.marshal(tag, in.Elem()) |
||||||
|
} |
||||||
|
case reflect.Map: |
||||||
|
e.mapv(tag, in) |
||||||
|
case reflect.Ptr: |
||||||
|
if in.IsNil() { |
||||||
|
e.nilv() |
||||||
|
} else { |
||||||
|
e.marshal(tag, in.Elem()) |
||||||
|
} |
||||||
|
case reflect.Struct: |
||||||
|
e.structv(tag, in) |
||||||
|
case reflect.Slice: |
||||||
|
if in.Type().Elem() == mapItemType { |
||||||
|
e.itemsv(tag, in) |
||||||
|
} else { |
||||||
|
e.slicev(tag, in) |
||||||
|
} |
||||||
|
case reflect.String: |
||||||
|
e.stringv(tag, in) |
||||||
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64: |
||||||
|
if in.Type() == durationType { |
||||||
|
e.stringv(tag, reflect.ValueOf(iface.(time.Duration).String())) |
||||||
|
} else { |
||||||
|
e.intv(tag, in) |
||||||
|
} |
||||||
|
case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, reflect.Uintptr: |
||||||
|
e.uintv(tag, in) |
||||||
|
case reflect.Float32, reflect.Float64: |
||||||
|
e.floatv(tag, in) |
||||||
|
case reflect.Bool: |
||||||
|
e.boolv(tag, in) |
||||||
|
default: |
||||||
|
panic("cannot marshal type: " + in.Type().String()) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
func (e *encoder) mapv(tag string, in reflect.Value) { |
||||||
|
e.mappingv(tag, func() { |
||||||
|
keys := keyList(in.MapKeys()) |
||||||
|
sort.Sort(keys) |
||||||
|
for _, k := range keys { |
||||||
|
e.marshal("", k) |
||||||
|
e.marshal("", in.MapIndex(k)) |
||||||
|
} |
||||||
|
}) |
||||||
|
} |
||||||
|
|
||||||
|
func (e *encoder) itemsv(tag string, in reflect.Value) { |
||||||
|
e.mappingv(tag, func() { |
||||||
|
slice := in.Convert(reflect.TypeOf([]MapItem{})).Interface().([]MapItem) |
||||||
|
for _, item := range slice { |
||||||
|
e.marshal("", reflect.ValueOf(item.Key)) |
||||||
|
e.marshal("", reflect.ValueOf(item.Value)) |
||||||
|
} |
||||||
|
}) |
||||||
|
} |
||||||
|
|
||||||
|
func (e *encoder) structv(tag string, in reflect.Value) { |
||||||
|
sinfo, err := getStructInfo(in.Type()) |
||||||
|
if err != nil { |
||||||
|
panic(err) |
||||||
|
} |
||||||
|
e.mappingv(tag, func() { |
||||||
|
for _, info := range sinfo.FieldsList { |
||||||
|
var value reflect.Value |
||||||
|
if info.Inline == nil { |
||||||
|
value = in.Field(info.Num) |
||||||
|
} else { |
||||||
|
value = in.FieldByIndex(info.Inline) |
||||||
|
} |
||||||
|
if info.OmitEmpty && isZero(value) { |
||||||
|
continue |
||||||
|
} |
||||||
|
e.marshal("", reflect.ValueOf(info.Key)) |
||||||
|
e.flow = info.Flow |
||||||
|
e.marshal("", value) |
||||||
|
} |
||||||
|
if sinfo.InlineMap >= 0 { |
||||||
|
m := in.Field(sinfo.InlineMap) |
||||||
|
if m.Len() > 0 { |
||||||
|
e.flow = false |
||||||
|
keys := keyList(m.MapKeys()) |
||||||
|
sort.Sort(keys) |
||||||
|
for _, k := range keys { |
||||||
|
if _, found := sinfo.FieldsMap[k.String()]; found { |
||||||
|
panic(fmt.Sprintf("Can't have key %q in inlined map; conflicts with struct field", k.String())) |
||||||
|
} |
||||||
|
e.marshal("", k) |
||||||
|
e.flow = false |
||||||
|
e.marshal("", m.MapIndex(k)) |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
}) |
||||||
|
} |
||||||
|
|
||||||
|
func (e *encoder) mappingv(tag string, f func()) { |
||||||
|
implicit := tag == "" |
||||||
|
style := yaml_BLOCK_MAPPING_STYLE |
||||||
|
if e.flow { |
||||||
|
e.flow = false |
||||||
|
style = yaml_FLOW_MAPPING_STYLE |
||||||
|
} |
||||||
|
e.must(yaml_mapping_start_event_initialize(&e.event, nil, []byte(tag), implicit, style)) |
||||||
|
e.emit() |
||||||
|
f() |
||||||
|
e.must(yaml_mapping_end_event_initialize(&e.event)) |
||||||
|
e.emit() |
||||||
|
} |
||||||
|
|
||||||
|
func (e *encoder) slicev(tag string, in reflect.Value) { |
||||||
|
implicit := tag == "" |
||||||
|
style := yaml_BLOCK_SEQUENCE_STYLE |
||||||
|
if e.flow { |
||||||
|
e.flow = false |
||||||
|
style = yaml_FLOW_SEQUENCE_STYLE |
||||||
|
} |
||||||
|
e.must(yaml_sequence_start_event_initialize(&e.event, nil, []byte(tag), implicit, style)) |
||||||
|
e.emit() |
||||||
|
n := in.Len() |
||||||
|
for i := 0; i < n; i++ { |
||||||
|
e.marshal("", in.Index(i)) |
||||||
|
} |
||||||
|
e.must(yaml_sequence_end_event_initialize(&e.event)) |
||||||
|
e.emit() |
||||||
|
} |
||||||
|
|
||||||
|
// isBase60 returns whether s is in base 60 notation as defined in YAML 1.1.
|
||||||
|
//
|
||||||
|
// The base 60 float notation in YAML 1.1 is a terrible idea and is unsupported
|
||||||
|
// in YAML 1.2 and by this package, but these should be marshalled quoted for
|
||||||
|
// the time being for compatibility with other parsers.
|
||||||
|
func isBase60Float(s string) (result bool) { |
||||||
|
// Fast path.
|
||||||
|
if s == "" { |
||||||
|
return false |
||||||
|
} |
||||||
|
c := s[0] |
||||||
|
if !(c == '+' || c == '-' || c >= '0' && c <= '9') || strings.IndexByte(s, ':') < 0 { |
||||||
|
return false |
||||||
|
} |
||||||
|
// Do the full match.
|
||||||
|
return base60float.MatchString(s) |
||||||
|
} |
||||||
|
|
||||||
|
// From http://yaml.org/type/float.html, except the regular expression there
|
||||||
|
// is bogus. In practice parsers do not enforce the "\.[0-9_]*" suffix.
|
||||||
|
var base60float = regexp.MustCompile(`^[-+]?[0-9][0-9_]*(?::[0-5]?[0-9])+(?:\.[0-9_]*)?$`) |
||||||
|
|
||||||
|
func (e *encoder) stringv(tag string, in reflect.Value) { |
||||||
|
var style yaml_scalar_style_t |
||||||
|
s := in.String() |
||||||
|
rtag, rs := resolve("", s) |
||||||
|
if rtag == yaml_BINARY_TAG { |
||||||
|
if tag == "" || tag == yaml_STR_TAG { |
||||||
|
tag = rtag |
||||||
|
s = rs.(string) |
||||||
|
} else if tag == yaml_BINARY_TAG { |
||||||
|
failf("explicitly tagged !!binary data must be base64-encoded") |
||||||
|
} else { |
||||||
|
failf("cannot marshal invalid UTF-8 data as %s", shortTag(tag)) |
||||||
|
} |
||||||
|
} |
||||||
|
if tag == "" && (rtag != yaml_STR_TAG || isBase60Float(s)) { |
||||||
|
style = yaml_DOUBLE_QUOTED_SCALAR_STYLE |
||||||
|
} else if strings.Contains(s, "\n") { |
||||||
|
style = yaml_LITERAL_SCALAR_STYLE |
||||||
|
} else { |
||||||
|
style = yaml_PLAIN_SCALAR_STYLE |
||||||
|
} |
||||||
|
e.emitScalar(s, "", tag, style) |
||||||
|
} |
||||||
|
|
||||||
|
func (e *encoder) boolv(tag string, in reflect.Value) { |
||||||
|
var s string |
||||||
|
if in.Bool() { |
||||||
|
s = "true" |
||||||
|
} else { |
||||||
|
s = "false" |
||||||
|
} |
||||||
|
e.emitScalar(s, "", tag, yaml_PLAIN_SCALAR_STYLE) |
||||||
|
} |
||||||
|
|
||||||
|
func (e *encoder) intv(tag string, in reflect.Value) { |
||||||
|
s := strconv.FormatInt(in.Int(), 10) |
||||||
|
e.emitScalar(s, "", tag, yaml_PLAIN_SCALAR_STYLE) |
||||||
|
} |
||||||
|
|
||||||
|
func (e *encoder) uintv(tag string, in reflect.Value) { |
||||||
|
s := strconv.FormatUint(in.Uint(), 10) |
||||||
|
e.emitScalar(s, "", tag, yaml_PLAIN_SCALAR_STYLE) |
||||||
|
} |
||||||
|
|
||||||
|
func (e *encoder) floatv(tag string, in reflect.Value) { |
||||||
|
// FIXME: Handle 64 bits here.
|
||||||
|
s := strconv.FormatFloat(float64(in.Float()), 'g', -1, 32) |
||||||
|
switch s { |
||||||
|
case "+Inf": |
||||||
|
s = ".inf" |
||||||
|
case "-Inf": |
||||||
|
s = "-.inf" |
||||||
|
case "NaN": |
||||||
|
s = ".nan" |
||||||
|
} |
||||||
|
e.emitScalar(s, "", tag, yaml_PLAIN_SCALAR_STYLE) |
||||||
|
} |
||||||
|
|
||||||
|
func (e *encoder) nilv() { |
||||||
|
e.emitScalar("null", "", "", yaml_PLAIN_SCALAR_STYLE) |
||||||
|
} |
||||||
|
|
||||||
|
func (e *encoder) emitScalar(value, anchor, tag string, style yaml_scalar_style_t) { |
||||||
|
implicit := tag == "" |
||||||
|
e.must(yaml_scalar_event_initialize(&e.event, []byte(anchor), []byte(tag), []byte(value), implicit, implicit, style)) |
||||||
|
e.emit() |
||||||
|
} |
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,394 @@ |
|||||||
|
package yaml |
||||||
|
|
||||||
|
import ( |
||||||
|
"io" |
||||||
|
) |
||||||
|
|
||||||
|
// Set the reader error and return 0.
|
||||||
|
func yaml_parser_set_reader_error(parser *yaml_parser_t, problem string, offset int, value int) bool { |
||||||
|
parser.error = yaml_READER_ERROR |
||||||
|
parser.problem = problem |
||||||
|
parser.problem_offset = offset |
||||||
|
parser.problem_value = value |
||||||
|
return false |
||||||
|
} |
||||||
|
|
||||||
|
// Byte order marks.
|
||||||
|
const ( |
||||||
|
bom_UTF8 = "\xef\xbb\xbf" |
||||||
|
bom_UTF16LE = "\xff\xfe" |
||||||
|
bom_UTF16BE = "\xfe\xff" |
||||||
|
) |
||||||
|
|
||||||
|
// Determine the input stream encoding by checking the BOM symbol. If no BOM is
|
||||||
|
// found, the UTF-8 encoding is assumed. Return 1 on success, 0 on failure.
|
||||||
|
func yaml_parser_determine_encoding(parser *yaml_parser_t) bool { |
||||||
|
// Ensure that we had enough bytes in the raw buffer.
|
||||||
|
for !parser.eof && len(parser.raw_buffer)-parser.raw_buffer_pos < 3 { |
||||||
|
if !yaml_parser_update_raw_buffer(parser) { |
||||||
|
return false |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// Determine the encoding.
|
||||||
|
buf := parser.raw_buffer |
||||||
|
pos := parser.raw_buffer_pos |
||||||
|
avail := len(buf) - pos |
||||||
|
if avail >= 2 && buf[pos] == bom_UTF16LE[0] && buf[pos+1] == bom_UTF16LE[1] { |
||||||
|
parser.encoding = yaml_UTF16LE_ENCODING |
||||||
|
parser.raw_buffer_pos += 2 |
||||||
|
parser.offset += 2 |
||||||
|
} else if avail >= 2 && buf[pos] == bom_UTF16BE[0] && buf[pos+1] == bom_UTF16BE[1] { |
||||||
|
parser.encoding = yaml_UTF16BE_ENCODING |
||||||
|
parser.raw_buffer_pos += 2 |
||||||
|
parser.offset += 2 |
||||||
|
} else if avail >= 3 && buf[pos] == bom_UTF8[0] && buf[pos+1] == bom_UTF8[1] && buf[pos+2] == bom_UTF8[2] { |
||||||
|
parser.encoding = yaml_UTF8_ENCODING |
||||||
|
parser.raw_buffer_pos += 3 |
||||||
|
parser.offset += 3 |
||||||
|
} else { |
||||||
|
parser.encoding = yaml_UTF8_ENCODING |
||||||
|
} |
||||||
|
return true |
||||||
|
} |
||||||
|
|
||||||
|
// Update the raw buffer.
|
||||||
|
func yaml_parser_update_raw_buffer(parser *yaml_parser_t) bool { |
||||||
|
size_read := 0 |
||||||
|
|
||||||
|
// Return if the raw buffer is full.
|
||||||
|
if parser.raw_buffer_pos == 0 && len(parser.raw_buffer) == cap(parser.raw_buffer) { |
||||||
|
return true |
||||||
|
} |
||||||
|
|
||||||
|
// Return on EOF.
|
||||||
|
if parser.eof { |
||||||
|
return true |
||||||
|
} |
||||||
|
|
||||||
|
// Move the remaining bytes in the raw buffer to the beginning.
|
||||||
|
if parser.raw_buffer_pos > 0 && parser.raw_buffer_pos < len(parser.raw_buffer) { |
||||||
|
copy(parser.raw_buffer, parser.raw_buffer[parser.raw_buffer_pos:]) |
||||||
|
} |
||||||
|
parser.raw_buffer = parser.raw_buffer[:len(parser.raw_buffer)-parser.raw_buffer_pos] |
||||||
|
parser.raw_buffer_pos = 0 |
||||||
|
|
||||||
|
// Call the read handler to fill the buffer.
|
||||||
|
size_read, err := parser.read_handler(parser, parser.raw_buffer[len(parser.raw_buffer):cap(parser.raw_buffer)]) |
||||||
|
parser.raw_buffer = parser.raw_buffer[:len(parser.raw_buffer)+size_read] |
||||||
|
if err == io.EOF { |
||||||
|
parser.eof = true |
||||||
|
} else if err != nil { |
||||||
|
return yaml_parser_set_reader_error(parser, "input error: "+err.Error(), parser.offset, -1) |
||||||
|
} |
||||||
|
return true |
||||||
|
} |
||||||
|
|
||||||
|
// Ensure that the buffer contains at least `length` characters.
|
||||||
|
// Return true on success, false on failure.
|
||||||
|
//
|
||||||
|
// The length is supposed to be significantly less that the buffer size.
|
||||||
|
func yaml_parser_update_buffer(parser *yaml_parser_t, length int) bool { |
||||||
|
if parser.read_handler == nil { |
||||||
|
panic("read handler must be set") |
||||||
|
} |
||||||
|
|
||||||
|
// If the EOF flag is set and the raw buffer is empty, do nothing.
|
||||||
|
if parser.eof && parser.raw_buffer_pos == len(parser.raw_buffer) { |
||||||
|
return true |
||||||
|
} |
||||||
|
|
||||||
|
// Return if the buffer contains enough characters.
|
||||||
|
if parser.unread >= length { |
||||||
|
return true |
||||||
|
} |
||||||
|
|
||||||
|
// Determine the input encoding if it is not known yet.
|
||||||
|
if parser.encoding == yaml_ANY_ENCODING { |
||||||
|
if !yaml_parser_determine_encoding(parser) { |
||||||
|
return false |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// Move the unread characters to the beginning of the buffer.
|
||||||
|
buffer_len := len(parser.buffer) |
||||||
|
if parser.buffer_pos > 0 && parser.buffer_pos < buffer_len { |
||||||
|
copy(parser.buffer, parser.buffer[parser.buffer_pos:]) |
||||||
|
buffer_len -= parser.buffer_pos |
||||||
|
parser.buffer_pos = 0 |
||||||
|
} else if parser.buffer_pos == buffer_len { |
||||||
|
buffer_len = 0 |
||||||
|
parser.buffer_pos = 0 |
||||||
|
} |
||||||
|
|
||||||
|
// Open the whole buffer for writing, and cut it before returning.
|
||||||
|
parser.buffer = parser.buffer[:cap(parser.buffer)] |
||||||
|
|
||||||
|
// Fill the buffer until it has enough characters.
|
||||||
|
first := true |
||||||
|
for parser.unread < length { |
||||||
|
|
||||||
|
// Fill the raw buffer if necessary.
|
||||||
|
if !first || parser.raw_buffer_pos == len(parser.raw_buffer) { |
||||||
|
if !yaml_parser_update_raw_buffer(parser) { |
||||||
|
parser.buffer = parser.buffer[:buffer_len] |
||||||
|
return false |
||||||
|
} |
||||||
|
} |
||||||
|
first = false |
||||||
|
|
||||||
|
// Decode the raw buffer.
|
||||||
|
inner: |
||||||
|
for parser.raw_buffer_pos != len(parser.raw_buffer) { |
||||||
|
var value rune |
||||||
|
var width int |
||||||
|
|
||||||
|
raw_unread := len(parser.raw_buffer) - parser.raw_buffer_pos |
||||||
|
|
||||||
|
// Decode the next character.
|
||||||
|
switch parser.encoding { |
||||||
|
case yaml_UTF8_ENCODING: |
||||||
|
// Decode a UTF-8 character. Check RFC 3629
|
||||||
|
// (http://www.ietf.org/rfc/rfc3629.txt) for more details.
|
||||||
|
//
|
||||||
|
// The following table (taken from the RFC) is used for
|
||||||
|
// decoding.
|
||||||
|
//
|
||||||
|
// Char. number range | UTF-8 octet sequence
|
||||||
|
// (hexadecimal) | (binary)
|
||||||
|
// --------------------+------------------------------------
|
||||||
|
// 0000 0000-0000 007F | 0xxxxxxx
|
||||||
|
// 0000 0080-0000 07FF | 110xxxxx 10xxxxxx
|
||||||
|
// 0000 0800-0000 FFFF | 1110xxxx 10xxxxxx 10xxxxxx
|
||||||
|
// 0001 0000-0010 FFFF | 11110xxx 10xxxxxx 10xxxxxx 10xxxxxx
|
||||||
|
//
|
||||||
|
// Additionally, the characters in the range 0xD800-0xDFFF
|
||||||
|
// are prohibited as they are reserved for use with UTF-16
|
||||||
|
// surrogate pairs.
|
||||||
|
|
||||||
|
// Determine the length of the UTF-8 sequence.
|
||||||
|
octet := parser.raw_buffer[parser.raw_buffer_pos] |
||||||
|
switch { |
||||||
|
case octet&0x80 == 0x00: |
||||||
|
width = 1 |
||||||
|
case octet&0xE0 == 0xC0: |
||||||
|
width = 2 |
||||||
|
case octet&0xF0 == 0xE0: |
||||||
|
width = 3 |
||||||
|
case octet&0xF8 == 0xF0: |
||||||
|
width = 4 |
||||||
|
default: |
||||||
|
// The leading octet is invalid.
|
||||||
|
return yaml_parser_set_reader_error(parser, |
||||||
|
"invalid leading UTF-8 octet", |
||||||
|
parser.offset, int(octet)) |
||||||
|
} |
||||||
|
|
||||||
|
// Check if the raw buffer contains an incomplete character.
|
||||||
|
if width > raw_unread { |
||||||
|
if parser.eof { |
||||||
|
return yaml_parser_set_reader_error(parser, |
||||||
|
"incomplete UTF-8 octet sequence", |
||||||
|
parser.offset, -1) |
||||||
|
} |
||||||
|
break inner |
||||||
|
} |
||||||
|
|
||||||
|
// Decode the leading octet.
|
||||||
|
switch { |
||||||
|
case octet&0x80 == 0x00: |
||||||
|
value = rune(octet & 0x7F) |
||||||
|
case octet&0xE0 == 0xC0: |
||||||
|
value = rune(octet & 0x1F) |
||||||
|
case octet&0xF0 == 0xE0: |
||||||
|
value = rune(octet & 0x0F) |
||||||
|
case octet&0xF8 == 0xF0: |
||||||
|
value = rune(octet & 0x07) |
||||||
|
default: |
||||||
|
value = 0 |
||||||
|
} |
||||||
|
|
||||||
|
// Check and decode the trailing octets.
|
||||||
|
for k := 1; k < width; k++ { |
||||||
|
octet = parser.raw_buffer[parser.raw_buffer_pos+k] |
||||||
|
|
||||||
|
// Check if the octet is valid.
|
||||||
|
if (octet & 0xC0) != 0x80 { |
||||||
|
return yaml_parser_set_reader_error(parser, |
||||||
|
"invalid trailing UTF-8 octet", |
||||||
|
parser.offset+k, int(octet)) |
||||||
|
} |
||||||
|
|
||||||
|
// Decode the octet.
|
||||||
|
value = (value << 6) + rune(octet&0x3F) |
||||||
|
} |
||||||
|
|
||||||
|
// Check the length of the sequence against the value.
|
||||||
|
switch { |
||||||
|
case width == 1: |
||||||
|
case width == 2 && value >= 0x80: |
||||||
|
case width == 3 && value >= 0x800: |
||||||
|
case width == 4 && value >= 0x10000: |
||||||
|
default: |
||||||
|
return yaml_parser_set_reader_error(parser, |
||||||
|
"invalid length of a UTF-8 sequence", |
||||||
|
parser.offset, -1) |
||||||
|
} |
||||||
|
|
||||||
|
// Check the range of the value.
|
||||||
|
if value >= 0xD800 && value <= 0xDFFF || value > 0x10FFFF { |
||||||
|
return yaml_parser_set_reader_error(parser, |
||||||
|
"invalid Unicode character", |
||||||
|
parser.offset, int(value)) |
||||||
|
} |
||||||
|
|
||||||
|
case yaml_UTF16LE_ENCODING, yaml_UTF16BE_ENCODING: |
||||||
|
var low, high int |
||||||
|
if parser.encoding == yaml_UTF16LE_ENCODING { |
||||||
|
low, high = 0, 1 |
||||||
|
} else { |
||||||
|
low, high = 1, 0 |
||||||
|
} |
||||||
|
|
||||||
|
// The UTF-16 encoding is not as simple as one might
|
||||||
|
// naively think. Check RFC 2781
|
||||||
|
// (http://www.ietf.org/rfc/rfc2781.txt).
|
||||||
|
//
|
||||||
|
// Normally, two subsequent bytes describe a Unicode
|
||||||
|
// character. However a special technique (called a
|
||||||
|
// surrogate pair) is used for specifying character
|
||||||
|
// values larger than 0xFFFF.
|
||||||
|
//
|
||||||
|
// A surrogate pair consists of two pseudo-characters:
|
||||||
|
// high surrogate area (0xD800-0xDBFF)
|
||||||
|
// low surrogate area (0xDC00-0xDFFF)
|
||||||
|
//
|
||||||
|
// The following formulas are used for decoding
|
||||||
|
// and encoding characters using surrogate pairs:
|
||||||
|
//
|
||||||
|
// U = U' + 0x10000 (0x01 00 00 <= U <= 0x10 FF FF)
|
||||||
|
// U' = yyyyyyyyyyxxxxxxxxxx (0 <= U' <= 0x0F FF FF)
|
||||||
|
// W1 = 110110yyyyyyyyyy
|
||||||
|
// W2 = 110111xxxxxxxxxx
|
||||||
|
//
|
||||||
|
// where U is the character value, W1 is the high surrogate
|
||||||
|
// area, W2 is the low surrogate area.
|
||||||
|
|
||||||
|
// Check for incomplete UTF-16 character.
|
||||||
|
if raw_unread < 2 { |
||||||
|
if parser.eof { |
||||||
|
return yaml_parser_set_reader_error(parser, |
||||||
|
"incomplete UTF-16 character", |
||||||
|
parser.offset, -1) |
||||||
|
} |
||||||
|
break inner |
||||||
|
} |
||||||
|
|
||||||
|
// Get the character.
|
||||||
|
value = rune(parser.raw_buffer[parser.raw_buffer_pos+low]) + |
||||||
|
(rune(parser.raw_buffer[parser.raw_buffer_pos+high]) << 8) |
||||||
|
|
||||||
|
// Check for unexpected low surrogate area.
|
||||||
|
if value&0xFC00 == 0xDC00 { |
||||||
|
return yaml_parser_set_reader_error(parser, |
||||||
|
"unexpected low surrogate area", |
||||||
|
parser.offset, int(value)) |
||||||
|
} |
||||||
|
|
||||||
|
// Check for a high surrogate area.
|
||||||
|
if value&0xFC00 == 0xD800 { |
||||||
|
width = 4 |
||||||
|
|
||||||
|
// Check for incomplete surrogate pair.
|
||||||
|
if raw_unread < 4 { |
||||||
|
if parser.eof { |
||||||
|
return yaml_parser_set_reader_error(parser, |
||||||
|
"incomplete UTF-16 surrogate pair", |
||||||
|
parser.offset, -1) |
||||||
|
} |
||||||
|
break inner |
||||||
|
} |
||||||
|
|
||||||
|
// Get the next character.
|
||||||
|
value2 := rune(parser.raw_buffer[parser.raw_buffer_pos+low+2]) + |
||||||
|
(rune(parser.raw_buffer[parser.raw_buffer_pos+high+2]) << 8) |
||||||
|
|
||||||
|
// Check for a low surrogate area.
|
||||||
|
if value2&0xFC00 != 0xDC00 { |
||||||
|
return yaml_parser_set_reader_error(parser, |
||||||
|
"expected low surrogate area", |
||||||
|
parser.offset+2, int(value2)) |
||||||
|
} |
||||||
|
|
||||||
|
// Generate the value of the surrogate pair.
|
||||||
|
value = 0x10000 + ((value & 0x3FF) << 10) + (value2 & 0x3FF) |
||||||
|
} else { |
||||||
|
width = 2 |
||||||
|
} |
||||||
|
|
||||||
|
default: |
||||||
|
panic("impossible") |
||||||
|
} |
||||||
|
|
||||||
|
// Check if the character is in the allowed range:
|
||||||
|
// #x9 | #xA | #xD | [#x20-#x7E] (8 bit)
|
||||||
|
// | #x85 | [#xA0-#xD7FF] | [#xE000-#xFFFD] (16 bit)
|
||||||
|
// | [#x10000-#x10FFFF] (32 bit)
|
||||||
|
switch { |
||||||
|
case value == 0x09: |
||||||
|
case value == 0x0A: |
||||||
|
case value == 0x0D: |
||||||
|
case value >= 0x20 && value <= 0x7E: |
||||||
|
case value == 0x85: |
||||||
|
case value >= 0xA0 && value <= 0xD7FF: |
||||||
|
case value >= 0xE000 && value <= 0xFFFD: |
||||||
|
case value >= 0x10000 && value <= 0x10FFFF: |
||||||
|
default: |
||||||
|
return yaml_parser_set_reader_error(parser, |
||||||
|
"control characters are not allowed", |
||||||
|
parser.offset, int(value)) |
||||||
|
} |
||||||
|
|
||||||
|
// Move the raw pointers.
|
||||||
|
parser.raw_buffer_pos += width |
||||||
|
parser.offset += width |
||||||
|
|
||||||
|
// Finally put the character into the buffer.
|
||||||
|
if value <= 0x7F { |
||||||
|
// 0000 0000-0000 007F . 0xxxxxxx
|
||||||
|
parser.buffer[buffer_len+0] = byte(value) |
||||||
|
buffer_len += 1 |
||||||
|
} else if value <= 0x7FF { |
||||||
|
// 0000 0080-0000 07FF . 110xxxxx 10xxxxxx
|
||||||
|
parser.buffer[buffer_len+0] = byte(0xC0 + (value >> 6)) |
||||||
|
parser.buffer[buffer_len+1] = byte(0x80 + (value & 0x3F)) |
||||||
|
buffer_len += 2 |
||||||
|
} else if value <= 0xFFFF { |
||||||
|
// 0000 0800-0000 FFFF . 1110xxxx 10xxxxxx 10xxxxxx
|
||||||
|
parser.buffer[buffer_len+0] = byte(0xE0 + (value >> 12)) |
||||||
|
parser.buffer[buffer_len+1] = byte(0x80 + ((value >> 6) & 0x3F)) |
||||||
|
parser.buffer[buffer_len+2] = byte(0x80 + (value & 0x3F)) |
||||||
|
buffer_len += 3 |
||||||
|
} else { |
||||||
|
// 0001 0000-0010 FFFF . 11110xxx 10xxxxxx 10xxxxxx 10xxxxxx
|
||||||
|
parser.buffer[buffer_len+0] = byte(0xF0 + (value >> 18)) |
||||||
|
parser.buffer[buffer_len+1] = byte(0x80 + ((value >> 12) & 0x3F)) |
||||||
|
parser.buffer[buffer_len+2] = byte(0x80 + ((value >> 6) & 0x3F)) |
||||||
|
parser.buffer[buffer_len+3] = byte(0x80 + (value & 0x3F)) |
||||||
|
buffer_len += 4 |
||||||
|
} |
||||||
|
|
||||||
|
parser.unread++ |
||||||
|
} |
||||||
|
|
||||||
|
// On EOF, put NUL into the buffer and return.
|
||||||
|
if parser.eof { |
||||||
|
parser.buffer[buffer_len] = 0 |
||||||
|
buffer_len++ |
||||||
|
parser.unread++ |
||||||
|
break |
||||||
|
} |
||||||
|
} |
||||||
|
parser.buffer = parser.buffer[:buffer_len] |
||||||
|
return true |
||||||
|
} |
@ -0,0 +1,203 @@ |
|||||||
|
package yaml |
||||||
|
|
||||||
|
import ( |
||||||
|
"encoding/base64" |
||||||
|
"math" |
||||||
|
"strconv" |
||||||
|
"strings" |
||||||
|
"unicode/utf8" |
||||||
|
) |
||||||
|
|
||||||
|
type resolveMapItem struct { |
||||||
|
value interface{} |
||||||
|
tag string |
||||||
|
} |
||||||
|
|
||||||
|
var resolveTable = make([]byte, 256) |
||||||
|
var resolveMap = make(map[string]resolveMapItem) |
||||||
|
|
||||||
|
func init() { |
||||||
|
t := resolveTable |
||||||
|
t[int('+')] = 'S' // Sign
|
||||||
|
t[int('-')] = 'S' |
||||||
|
for _, c := range "0123456789" { |
||||||
|
t[int(c)] = 'D' // Digit
|
||||||
|
} |
||||||
|
for _, c := range "yYnNtTfFoO~" { |
||||||
|
t[int(c)] = 'M' // In map
|
||||||
|
} |
||||||
|
t[int('.')] = '.' // Float (potentially in map)
|
||||||
|
|
||||||
|
var resolveMapList = []struct { |
||||||
|
v interface{} |
||||||
|
tag string |
||||||
|
l []string |
||||||
|
}{ |
||||||
|
{true, yaml_BOOL_TAG, []string{"y", "Y", "yes", "Yes", "YES"}}, |
||||||
|
{true, yaml_BOOL_TAG, []string{"true", "True", "TRUE"}}, |
||||||
|
{true, yaml_BOOL_TAG, []string{"on", "On", "ON"}}, |
||||||
|
{false, yaml_BOOL_TAG, []string{"n", "N", "no", "No", "NO"}}, |
||||||
|
{false, yaml_BOOL_TAG, []string{"false", "False", "FALSE"}}, |
||||||
|
{false, yaml_BOOL_TAG, []string{"off", "Off", "OFF"}}, |
||||||
|
{nil, yaml_NULL_TAG, []string{"", "~", "null", "Null", "NULL"}}, |
||||||
|
{math.NaN(), yaml_FLOAT_TAG, []string{".nan", ".NaN", ".NAN"}}, |
||||||
|
{math.Inf(+1), yaml_FLOAT_TAG, []string{".inf", ".Inf", ".INF"}}, |
||||||
|
{math.Inf(+1), yaml_FLOAT_TAG, []string{"+.inf", "+.Inf", "+.INF"}}, |
||||||
|
{math.Inf(-1), yaml_FLOAT_TAG, []string{"-.inf", "-.Inf", "-.INF"}}, |
||||||
|
{"<<", yaml_MERGE_TAG, []string{"<<"}}, |
||||||
|
} |
||||||
|
|
||||||
|
m := resolveMap |
||||||
|
for _, item := range resolveMapList { |
||||||
|
for _, s := range item.l { |
||||||
|
m[s] = resolveMapItem{item.v, item.tag} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
const longTagPrefix = "tag:yaml.org,2002:" |
||||||
|
|
||||||
|
func shortTag(tag string) string { |
||||||
|
// TODO This can easily be made faster and produce less garbage.
|
||||||
|
if strings.HasPrefix(tag, longTagPrefix) { |
||||||
|
return "!!" + tag[len(longTagPrefix):] |
||||||
|
} |
||||||
|
return tag |
||||||
|
} |
||||||
|
|
||||||
|
func longTag(tag string) string { |
||||||
|
if strings.HasPrefix(tag, "!!") { |
||||||
|
return longTagPrefix + tag[2:] |
||||||
|
} |
||||||
|
return tag |
||||||
|
} |
||||||
|
|
||||||
|
func resolvableTag(tag string) bool { |
||||||
|
switch tag { |
||||||
|
case "", yaml_STR_TAG, yaml_BOOL_TAG, yaml_INT_TAG, yaml_FLOAT_TAG, yaml_NULL_TAG: |
||||||
|
return true |
||||||
|
} |
||||||
|
return false |
||||||
|
} |
||||||
|
|
||||||
|
func resolve(tag string, in string) (rtag string, out interface{}) { |
||||||
|
if !resolvableTag(tag) { |
||||||
|
return tag, in |
||||||
|
} |
||||||
|
|
||||||
|
defer func() { |
||||||
|
switch tag { |
||||||
|
case "", rtag, yaml_STR_TAG, yaml_BINARY_TAG: |
||||||
|
return |
||||||
|
} |
||||||
|
failf("cannot decode %s `%s` as a %s", shortTag(rtag), in, shortTag(tag)) |
||||||
|
}() |
||||||
|
|
||||||
|
// Any data is accepted as a !!str or !!binary.
|
||||||
|
// Otherwise, the prefix is enough of a hint about what it might be.
|
||||||
|
hint := byte('N') |
||||||
|
if in != "" { |
||||||
|
hint = resolveTable[in[0]] |
||||||
|
} |
||||||
|
if hint != 0 && tag != yaml_STR_TAG && tag != yaml_BINARY_TAG { |
||||||
|
// Handle things we can lookup in a map.
|
||||||
|
if item, ok := resolveMap[in]; ok { |
||||||
|
return item.tag, item.value |
||||||
|
} |
||||||
|
|
||||||
|
// Base 60 floats are a bad idea, were dropped in YAML 1.2, and
|
||||||
|
// are purposefully unsupported here. They're still quoted on
|
||||||
|
// the way out for compatibility with other parser, though.
|
||||||
|
|
||||||
|
switch hint { |
||||||
|
case 'M': |
||||||
|
// We've already checked the map above.
|
||||||
|
|
||||||
|
case '.': |
||||||
|
// Not in the map, so maybe a normal float.
|
||||||
|
floatv, err := strconv.ParseFloat(in, 64) |
||||||
|
if err == nil { |
||||||
|
return yaml_FLOAT_TAG, floatv |
||||||
|
} |
||||||
|
|
||||||
|
case 'D', 'S': |
||||||
|
// Int, float, or timestamp.
|
||||||
|
plain := strings.Replace(in, "_", "", -1) |
||||||
|
intv, err := strconv.ParseInt(plain, 0, 64) |
||||||
|
if err == nil { |
||||||
|
if intv == int64(int(intv)) { |
||||||
|
return yaml_INT_TAG, int(intv) |
||||||
|
} else { |
||||||
|
return yaml_INT_TAG, intv |
||||||
|
} |
||||||
|
} |
||||||
|
uintv, err := strconv.ParseUint(plain, 0, 64) |
||||||
|
if err == nil { |
||||||
|
return yaml_INT_TAG, uintv |
||||||
|
} |
||||||
|
floatv, err := strconv.ParseFloat(plain, 64) |
||||||
|
if err == nil { |
||||||
|
return yaml_FLOAT_TAG, floatv |
||||||
|
} |
||||||
|
if strings.HasPrefix(plain, "0b") { |
||||||
|
intv, err := strconv.ParseInt(plain[2:], 2, 64) |
||||||
|
if err == nil { |
||||||
|
if intv == int64(int(intv)) { |
||||||
|
return yaml_INT_TAG, int(intv) |
||||||
|
} else { |
||||||
|
return yaml_INT_TAG, intv |
||||||
|
} |
||||||
|
} |
||||||
|
uintv, err := strconv.ParseUint(plain[2:], 2, 64) |
||||||
|
if err == nil { |
||||||
|
return yaml_INT_TAG, uintv |
||||||
|
} |
||||||
|
} else if strings.HasPrefix(plain, "-0b") { |
||||||
|
intv, err := strconv.ParseInt(plain[3:], 2, 64) |
||||||
|
if err == nil { |
||||||
|
if intv == int64(int(intv)) { |
||||||
|
return yaml_INT_TAG, -int(intv) |
||||||
|
} else { |
||||||
|
return yaml_INT_TAG, -intv |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
// XXX Handle timestamps here.
|
||||||
|
|
||||||
|
default: |
||||||
|
panic("resolveTable item not yet handled: " + string(rune(hint)) + " (with " + in + ")") |
||||||
|
} |
||||||
|
} |
||||||
|
if tag == yaml_BINARY_TAG { |
||||||
|
return yaml_BINARY_TAG, in |
||||||
|
} |
||||||
|
if utf8.ValidString(in) { |
||||||
|
return yaml_STR_TAG, in |
||||||
|
} |
||||||
|
return yaml_BINARY_TAG, encodeBase64(in) |
||||||
|
} |
||||||
|
|
||||||
|
// encodeBase64 encodes s as base64 that is broken up into multiple lines
|
||||||
|
// as appropriate for the resulting length.
|
||||||
|
func encodeBase64(s string) string { |
||||||
|
const lineLen = 70 |
||||||
|
encLen := base64.StdEncoding.EncodedLen(len(s)) |
||||||
|
lines := encLen/lineLen + 1 |
||||||
|
buf := make([]byte, encLen*2+lines) |
||||||
|
in := buf[0:encLen] |
||||||
|
out := buf[encLen:] |
||||||
|
base64.StdEncoding.Encode(in, []byte(s)) |
||||||
|
k := 0 |
||||||
|
for i := 0; i < len(in); i += lineLen { |
||||||
|
j := i + lineLen |
||||||
|
if j > len(in) { |
||||||
|
j = len(in) |
||||||
|
} |
||||||
|
k += copy(out[k:], in[i:j]) |
||||||
|
if lines > 1 { |
||||||
|
out[k] = '\n' |
||||||
|
k++ |
||||||
|
} |
||||||
|
} |
||||||
|
return string(out[:k]) |
||||||
|
} |
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,104 @@ |
|||||||
|
package yaml |
||||||
|
|
||||||
|
import ( |
||||||
|
"reflect" |
||||||
|
"unicode" |
||||||
|
) |
||||||
|
|
||||||
|
type keyList []reflect.Value |
||||||
|
|
||||||
|
func (l keyList) Len() int { return len(l) } |
||||||
|
func (l keyList) Swap(i, j int) { l[i], l[j] = l[j], l[i] } |
||||||
|
func (l keyList) Less(i, j int) bool { |
||||||
|
a := l[i] |
||||||
|
b := l[j] |
||||||
|
ak := a.Kind() |
||||||
|
bk := b.Kind() |
||||||
|
for (ak == reflect.Interface || ak == reflect.Ptr) && !a.IsNil() { |
||||||
|
a = a.Elem() |
||||||
|
ak = a.Kind() |
||||||
|
} |
||||||
|
for (bk == reflect.Interface || bk == reflect.Ptr) && !b.IsNil() { |
||||||
|
b = b.Elem() |
||||||
|
bk = b.Kind() |
||||||
|
} |
||||||
|
af, aok := keyFloat(a) |
||||||
|
bf, bok := keyFloat(b) |
||||||
|
if aok && bok { |
||||||
|
if af != bf { |
||||||
|
return af < bf |
||||||
|
} |
||||||
|
if ak != bk { |
||||||
|
return ak < bk |
||||||
|
} |
||||||
|
return numLess(a, b) |
||||||
|
} |
||||||
|
if ak != reflect.String || bk != reflect.String { |
||||||
|
return ak < bk |
||||||
|
} |
||||||
|
ar, br := []rune(a.String()), []rune(b.String()) |
||||||
|
for i := 0; i < len(ar) && i < len(br); i++ { |
||||||
|
if ar[i] == br[i] { |
||||||
|
continue |
||||||
|
} |
||||||
|
al := unicode.IsLetter(ar[i]) |
||||||
|
bl := unicode.IsLetter(br[i]) |
||||||
|
if al && bl { |
||||||
|
return ar[i] < br[i] |
||||||
|
} |
||||||
|
if al || bl { |
||||||
|
return bl |
||||||
|
} |
||||||
|
var ai, bi int |
||||||
|
var an, bn int64 |
||||||
|
for ai = i; ai < len(ar) && unicode.IsDigit(ar[ai]); ai++ { |
||||||
|
an = an*10 + int64(ar[ai]-'0') |
||||||
|
} |
||||||
|
for bi = i; bi < len(br) && unicode.IsDigit(br[bi]); bi++ { |
||||||
|
bn = bn*10 + int64(br[bi]-'0') |
||||||
|
} |
||||||
|
if an != bn { |
||||||
|
return an < bn |
||||||
|
} |
||||||
|
if ai != bi { |
||||||
|
return ai < bi |
||||||
|
} |
||||||
|
return ar[i] < br[i] |
||||||
|
} |
||||||
|
return len(ar) < len(br) |
||||||
|
} |
||||||
|
|
||||||
|
// keyFloat returns a float value for v if it is a number/bool
|
||||||
|
// and whether it is a number/bool or not.
|
||||||
|
func keyFloat(v reflect.Value) (f float64, ok bool) { |
||||||
|
switch v.Kind() { |
||||||
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64: |
||||||
|
return float64(v.Int()), true |
||||||
|
case reflect.Float32, reflect.Float64: |
||||||
|
return v.Float(), true |
||||||
|
case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, reflect.Uintptr: |
||||||
|
return float64(v.Uint()), true |
||||||
|
case reflect.Bool: |
||||||
|
if v.Bool() { |
||||||
|
return 1, true |
||||||
|
} |
||||||
|
return 0, true |
||||||
|
} |
||||||
|
return 0, false |
||||||
|
} |
||||||
|
|
||||||
|
// numLess returns whether a < b.
|
||||||
|
// a and b must necessarily have the same kind.
|
||||||
|
func numLess(a, b reflect.Value) bool { |
||||||
|
switch a.Kind() { |
||||||
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64: |
||||||
|
return a.Int() < b.Int() |
||||||
|
case reflect.Float32, reflect.Float64: |
||||||
|
return a.Float() < b.Float() |
||||||
|
case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, reflect.Uintptr: |
||||||
|
return a.Uint() < b.Uint() |
||||||
|
case reflect.Bool: |
||||||
|
return !a.Bool() && b.Bool() |
||||||
|
} |
||||||
|
panic("not a number") |
||||||
|
} |
@ -0,0 +1,89 @@ |
|||||||
|
package yaml |
||||||
|
|
||||||
|
// Set the writer error and return false.
|
||||||
|
func yaml_emitter_set_writer_error(emitter *yaml_emitter_t, problem string) bool { |
||||||
|
emitter.error = yaml_WRITER_ERROR |
||||||
|
emitter.problem = problem |
||||||
|
return false |
||||||
|
} |
||||||
|
|
||||||
|
// Flush the output buffer.
|
||||||
|
func yaml_emitter_flush(emitter *yaml_emitter_t) bool { |
||||||
|
if emitter.write_handler == nil { |
||||||
|
panic("write handler not set") |
||||||
|
} |
||||||
|
|
||||||
|
// Check if the buffer is empty.
|
||||||
|
if emitter.buffer_pos == 0 { |
||||||
|
return true |
||||||
|
} |
||||||
|
|
||||||
|
// If the output encoding is UTF-8, we don't need to recode the buffer.
|
||||||
|
if emitter.encoding == yaml_UTF8_ENCODING { |
||||||
|
if err := emitter.write_handler(emitter, emitter.buffer[:emitter.buffer_pos]); err != nil { |
||||||
|
return yaml_emitter_set_writer_error(emitter, "write error: "+err.Error()) |
||||||
|
} |
||||||
|
emitter.buffer_pos = 0 |
||||||
|
return true |
||||||
|
} |
||||||
|
|
||||||
|
// Recode the buffer into the raw buffer.
|
||||||
|
var low, high int |
||||||
|
if emitter.encoding == yaml_UTF16LE_ENCODING { |
||||||
|
low, high = 0, 1 |
||||||
|
} else { |
||||||
|
high, low = 1, 0 |
||||||
|
} |
||||||
|
|
||||||
|
pos := 0 |
||||||
|
for pos < emitter.buffer_pos { |
||||||
|
// See the "reader.c" code for more details on UTF-8 encoding. Note
|
||||||
|
// that we assume that the buffer contains a valid UTF-8 sequence.
|
||||||
|
|
||||||
|
// Read the next UTF-8 character.
|
||||||
|
octet := emitter.buffer[pos] |
||||||
|
|
||||||
|
var w int |
||||||
|
var value rune |
||||||
|
switch { |
||||||
|
case octet&0x80 == 0x00: |
||||||
|
w, value = 1, rune(octet&0x7F) |
||||||
|
case octet&0xE0 == 0xC0: |
||||||
|
w, value = 2, rune(octet&0x1F) |
||||||
|
case octet&0xF0 == 0xE0: |
||||||
|
w, value = 3, rune(octet&0x0F) |
||||||
|
case octet&0xF8 == 0xF0: |
||||||
|
w, value = 4, rune(octet&0x07) |
||||||
|
} |
||||||
|
for k := 1; k < w; k++ { |
||||||
|
octet = emitter.buffer[pos+k] |
||||||
|
value = (value << 6) + (rune(octet) & 0x3F) |
||||||
|
} |
||||||
|
pos += w |
||||||
|
|
||||||
|
// Write the character.
|
||||||
|
if value < 0x10000 { |
||||||
|
var b [2]byte |
||||||
|
b[high] = byte(value >> 8) |
||||||
|
b[low] = byte(value & 0xFF) |
||||||
|
emitter.raw_buffer = append(emitter.raw_buffer, b[0], b[1]) |
||||||
|
} else { |
||||||
|
// Write the character using a surrogate pair (check "reader.c").
|
||||||
|
var b [4]byte |
||||||
|
value -= 0x10000 |
||||||
|
b[high] = byte(0xD8 + (value >> 18)) |
||||||
|
b[low] = byte((value >> 10) & 0xFF) |
||||||
|
b[high+2] = byte(0xDC + ((value >> 8) & 0xFF)) |
||||||
|
b[low+2] = byte(value & 0xFF) |
||||||
|
emitter.raw_buffer = append(emitter.raw_buffer, b[0], b[1], b[2], b[3]) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// Write the raw buffer.
|
||||||
|
if err := emitter.write_handler(emitter, emitter.raw_buffer); err != nil { |
||||||
|
return yaml_emitter_set_writer_error(emitter, "write error: "+err.Error()) |
||||||
|
} |
||||||
|
emitter.buffer_pos = 0 |
||||||
|
emitter.raw_buffer = emitter.raw_buffer[:0] |
||||||
|
return true |
||||||
|
} |
@ -0,0 +1,346 @@ |
|||||||
|
// Package yaml implements YAML support for the Go language.
|
||||||
|
//
|
||||||
|
// Source code and other details for the project are available at GitHub:
|
||||||
|
//
|
||||||
|
// https://github.com/go-yaml/yaml
|
||||||
|
//
|
||||||
|
package yaml |
||||||
|
|
||||||
|
import ( |
||||||
|
"errors" |
||||||
|
"fmt" |
||||||
|
"reflect" |
||||||
|
"strings" |
||||||
|
"sync" |
||||||
|
) |
||||||
|
|
||||||
|
// MapSlice encodes and decodes as a YAML map.
|
||||||
|
// The order of keys is preserved when encoding and decoding.
|
||||||
|
type MapSlice []MapItem |
||||||
|
|
||||||
|
// MapItem is an item in a MapSlice.
|
||||||
|
type MapItem struct { |
||||||
|
Key, Value interface{} |
||||||
|
} |
||||||
|
|
||||||
|
// The Unmarshaler interface may be implemented by types to customize their
|
||||||
|
// behavior when being unmarshaled from a YAML document. The UnmarshalYAML
|
||||||
|
// method receives a function that may be called to unmarshal the original
|
||||||
|
// YAML value into a field or variable. It is safe to call the unmarshal
|
||||||
|
// function parameter more than once if necessary.
|
||||||
|
type Unmarshaler interface { |
||||||
|
UnmarshalYAML(unmarshal func(interface{}) error) error |
||||||
|
} |
||||||
|
|
||||||
|
// The Marshaler interface may be implemented by types to customize their
|
||||||
|
// behavior when being marshaled into a YAML document. The returned value
|
||||||
|
// is marshaled in place of the original value implementing Marshaler.
|
||||||
|
//
|
||||||
|
// If an error is returned by MarshalYAML, the marshaling procedure stops
|
||||||
|
// and returns with the provided error.
|
||||||
|
type Marshaler interface { |
||||||
|
MarshalYAML() (interface{}, error) |
||||||
|
} |
||||||
|
|
||||||
|
// Unmarshal decodes the first document found within the in byte slice
|
||||||
|
// and assigns decoded values into the out value.
|
||||||
|
//
|
||||||
|
// Maps and pointers (to a struct, string, int, etc) are accepted as out
|
||||||
|
// values. If an internal pointer within a struct is not initialized,
|
||||||
|
// the yaml package will initialize it if necessary for unmarshalling
|
||||||
|
// the provided data. The out parameter must not be nil.
|
||||||
|
//
|
||||||
|
// The type of the decoded values should be compatible with the respective
|
||||||
|
// values in out. If one or more values cannot be decoded due to a type
|
||||||
|
// mismatches, decoding continues partially until the end of the YAML
|
||||||
|
// content, and a *yaml.TypeError is returned with details for all
|
||||||
|
// missed values.
|
||||||
|
//
|
||||||
|
// Struct fields are only unmarshalled if they are exported (have an
|
||||||
|
// upper case first letter), and are unmarshalled using the field name
|
||||||
|
// lowercased as the default key. Custom keys may be defined via the
|
||||||
|
// "yaml" name in the field tag: the content preceding the first comma
|
||||||
|
// is used as the key, and the following comma-separated options are
|
||||||
|
// used to tweak the marshalling process (see Marshal).
|
||||||
|
// Conflicting names result in a runtime error.
|
||||||
|
//
|
||||||
|
// For example:
|
||||||
|
//
|
||||||
|
// type T struct {
|
||||||
|
// F int `yaml:"a,omitempty"`
|
||||||
|
// B int
|
||||||
|
// }
|
||||||
|
// var t T
|
||||||
|
// yaml.Unmarshal([]byte("a: 1\nb: 2"), &t)
|
||||||
|
//
|
||||||
|
// See the documentation of Marshal for the format of tags and a list of
|
||||||
|
// supported tag options.
|
||||||
|
//
|
||||||
|
func Unmarshal(in []byte, out interface{}) (err error) { |
||||||
|
defer handleErr(&err) |
||||||
|
d := newDecoder() |
||||||
|
p := newParser(in) |
||||||
|
defer p.destroy() |
||||||
|
node := p.parse() |
||||||
|
if node != nil { |
||||||
|
v := reflect.ValueOf(out) |
||||||
|
if v.Kind() == reflect.Ptr && !v.IsNil() { |
||||||
|
v = v.Elem() |
||||||
|
} |
||||||
|
d.unmarshal(node, v) |
||||||
|
} |
||||||
|
if len(d.terrors) > 0 { |
||||||
|
return &TypeError{d.terrors} |
||||||
|
} |
||||||
|
return nil |
||||||
|
} |
||||||
|
|
||||||
|
// Marshal serializes the value provided into a YAML document. The structure
|
||||||
|
// of the generated document will reflect the structure of the value itself.
|
||||||
|
// Maps and pointers (to struct, string, int, etc) are accepted as the in value.
|
||||||
|
//
|
||||||
|
// Struct fields are only unmarshalled if they are exported (have an upper case
|
||||||
|
// first letter), and are unmarshalled using the field name lowercased as the
|
||||||
|
// default key. Custom keys may be defined via the "yaml" name in the field
|
||||||
|
// tag: the content preceding the first comma is used as the key, and the
|
||||||
|
// following comma-separated options are used to tweak the marshalling process.
|
||||||
|
// Conflicting names result in a runtime error.
|
||||||
|
//
|
||||||
|
// The field tag format accepted is:
|
||||||
|
//
|
||||||
|
// `(...) yaml:"[<key>][,<flag1>[,<flag2>]]" (...)`
|
||||||
|
//
|
||||||
|
// The following flags are currently supported:
|
||||||
|
//
|
||||||
|
// omitempty Only include the field if it's not set to the zero
|
||||||
|
// value for the type or to empty slices or maps.
|
||||||
|
// Does not apply to zero valued structs.
|
||||||
|
//
|
||||||
|
// flow Marshal using a flow style (useful for structs,
|
||||||
|
// sequences and maps).
|
||||||
|
//
|
||||||
|
// inline Inline the field, which must be a struct or a map,
|
||||||
|
// causing all of its fields or keys to be processed as if
|
||||||
|
// they were part of the outer struct. For maps, keys must
|
||||||
|
// not conflict with the yaml keys of other struct fields.
|
||||||
|
//
|
||||||
|
// In addition, if the key is "-", the field is ignored.
|
||||||
|
//
|
||||||
|
// For example:
|
||||||
|
//
|
||||||
|
// type T struct {
|
||||||
|
// F int "a,omitempty"
|
||||||
|
// B int
|
||||||
|
// }
|
||||||
|
// yaml.Marshal(&T{B: 2}) // Returns "b: 2\n"
|
||||||
|
// yaml.Marshal(&T{F: 1}} // Returns "a: 1\nb: 0\n"
|
||||||
|
//
|
||||||
|
func Marshal(in interface{}) (out []byte, err error) { |
||||||
|
defer handleErr(&err) |
||||||
|
e := newEncoder() |
||||||
|
defer e.destroy() |
||||||
|
e.marshal("", reflect.ValueOf(in)) |
||||||
|
e.finish() |
||||||
|
out = e.out |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
func handleErr(err *error) { |
||||||
|
if v := recover(); v != nil { |
||||||
|
if e, ok := v.(yamlError); ok { |
||||||
|
*err = e.err |
||||||
|
} else { |
||||||
|
panic(v) |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
type yamlError struct { |
||||||
|
err error |
||||||
|
} |
||||||
|
|
||||||
|
func fail(err error) { |
||||||
|
panic(yamlError{err}) |
||||||
|
} |
||||||
|
|
||||||
|
func failf(format string, args ...interface{}) { |
||||||
|
panic(yamlError{fmt.Errorf("yaml: "+format, args...)}) |
||||||
|
} |
||||||
|
|
||||||
|
// A TypeError is returned by Unmarshal when one or more fields in
|
||||||
|
// the YAML document cannot be properly decoded into the requested
|
||||||
|
// types. When this error is returned, the value is still
|
||||||
|
// unmarshaled partially.
|
||||||
|
type TypeError struct { |
||||||
|
Errors []string |
||||||
|
} |
||||||
|
|
||||||
|
func (e *TypeError) Error() string { |
||||||
|
return fmt.Sprintf("yaml: unmarshal errors:\n %s", strings.Join(e.Errors, "\n ")) |
||||||
|
} |
||||||
|
|
||||||
|
// --------------------------------------------------------------------------
|
||||||
|
// Maintain a mapping of keys to structure field indexes
|
||||||
|
|
||||||
|
// The code in this section was copied from mgo/bson.
|
||||||
|
|
||||||
|
// structInfo holds details for the serialization of fields of
|
||||||
|
// a given struct.
|
||||||
|
type structInfo struct { |
||||||
|
FieldsMap map[string]fieldInfo |
||||||
|
FieldsList []fieldInfo |
||||||
|
|
||||||
|
// InlineMap is the number of the field in the struct that
|
||||||
|
// contains an ,inline map, or -1 if there's none.
|
||||||
|
InlineMap int |
||||||
|
} |
||||||
|
|
||||||
|
type fieldInfo struct { |
||||||
|
Key string |
||||||
|
Num int |
||||||
|
OmitEmpty bool |
||||||
|
Flow bool |
||||||
|
|
||||||
|
// Inline holds the field index if the field is part of an inlined struct.
|
||||||
|
Inline []int |
||||||
|
} |
||||||
|
|
||||||
|
var structMap = make(map[reflect.Type]*structInfo) |
||||||
|
var fieldMapMutex sync.RWMutex |
||||||
|
|
||||||
|
func getStructInfo(st reflect.Type) (*structInfo, error) { |
||||||
|
fieldMapMutex.RLock() |
||||||
|
sinfo, found := structMap[st] |
||||||
|
fieldMapMutex.RUnlock() |
||||||
|
if found { |
||||||
|
return sinfo, nil |
||||||
|
} |
||||||
|
|
||||||
|
n := st.NumField() |
||||||
|
fieldsMap := make(map[string]fieldInfo) |
||||||
|
fieldsList := make([]fieldInfo, 0, n) |
||||||
|
inlineMap := -1 |
||||||
|
for i := 0; i != n; i++ { |
||||||
|
field := st.Field(i) |
||||||
|
if field.PkgPath != "" && !field.Anonymous { |
||||||
|
continue // Private field
|
||||||
|
} |
||||||
|
|
||||||
|
info := fieldInfo{Num: i} |
||||||
|
|
||||||
|
tag := field.Tag.Get("yaml") |
||||||
|
if tag == "" && strings.Index(string(field.Tag), ":") < 0 { |
||||||
|
tag = string(field.Tag) |
||||||
|
} |
||||||
|
if tag == "-" { |
||||||
|
continue |
||||||
|
} |
||||||
|
|
||||||
|
inline := false |
||||||
|
fields := strings.Split(tag, ",") |
||||||
|
if len(fields) > 1 { |
||||||
|
for _, flag := range fields[1:] { |
||||||
|
switch flag { |
||||||
|
case "omitempty": |
||||||
|
info.OmitEmpty = true |
||||||
|
case "flow": |
||||||
|
info.Flow = true |
||||||
|
case "inline": |
||||||
|
inline = true |
||||||
|
default: |
||||||
|
return nil, errors.New(fmt.Sprintf("Unsupported flag %q in tag %q of type %s", flag, tag, st)) |
||||||
|
} |
||||||
|
} |
||||||
|
tag = fields[0] |
||||||
|
} |
||||||
|
|
||||||
|
if inline { |
||||||
|
switch field.Type.Kind() { |
||||||
|
case reflect.Map: |
||||||
|
if inlineMap >= 0 { |
||||||
|
return nil, errors.New("Multiple ,inline maps in struct " + st.String()) |
||||||
|
} |
||||||
|
if field.Type.Key() != reflect.TypeOf("") { |
||||||
|
return nil, errors.New("Option ,inline needs a map with string keys in struct " + st.String()) |
||||||
|
} |
||||||
|
inlineMap = info.Num |
||||||
|
case reflect.Struct: |
||||||
|
sinfo, err := getStructInfo(field.Type) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
for _, finfo := range sinfo.FieldsList { |
||||||
|
if _, found := fieldsMap[finfo.Key]; found { |
||||||
|
msg := "Duplicated key '" + finfo.Key + "' in struct " + st.String() |
||||||
|
return nil, errors.New(msg) |
||||||
|
} |
||||||
|
if finfo.Inline == nil { |
||||||
|
finfo.Inline = []int{i, finfo.Num} |
||||||
|
} else { |
||||||
|
finfo.Inline = append([]int{i}, finfo.Inline...) |
||||||
|
} |
||||||
|
fieldsMap[finfo.Key] = finfo |
||||||
|
fieldsList = append(fieldsList, finfo) |
||||||
|
} |
||||||
|
default: |
||||||
|
//return nil, errors.New("Option ,inline needs a struct value or map field")
|
||||||
|
return nil, errors.New("Option ,inline needs a struct value field") |
||||||
|
} |
||||||
|
continue |
||||||
|
} |
||||||
|
|
||||||
|
if tag != "" { |
||||||
|
info.Key = tag |
||||||
|
} else { |
||||||
|
info.Key = strings.ToLower(field.Name) |
||||||
|
} |
||||||
|
|
||||||
|
if _, found = fieldsMap[info.Key]; found { |
||||||
|
msg := "Duplicated key '" + info.Key + "' in struct " + st.String() |
||||||
|
return nil, errors.New(msg) |
||||||
|
} |
||||||
|
|
||||||
|
fieldsList = append(fieldsList, info) |
||||||
|
fieldsMap[info.Key] = info |
||||||
|
} |
||||||
|
|
||||||
|
sinfo = &structInfo{fieldsMap, fieldsList, inlineMap} |
||||||
|
|
||||||
|
fieldMapMutex.Lock() |
||||||
|
structMap[st] = sinfo |
||||||
|
fieldMapMutex.Unlock() |
||||||
|
return sinfo, nil |
||||||
|
} |
||||||
|
|
||||||
|
func isZero(v reflect.Value) bool { |
||||||
|
switch v.Kind() { |
||||||
|
case reflect.String: |
||||||
|
return len(v.String()) == 0 |
||||||
|
case reflect.Interface, reflect.Ptr: |
||||||
|
return v.IsNil() |
||||||
|
case reflect.Slice: |
||||||
|
return v.Len() == 0 |
||||||
|
case reflect.Map: |
||||||
|
return v.Len() == 0 |
||||||
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64: |
||||||
|
return v.Int() == 0 |
||||||
|
case reflect.Float32, reflect.Float64: |
||||||
|
return v.Float() == 0 |
||||||
|
case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, reflect.Uintptr: |
||||||
|
return v.Uint() == 0 |
||||||
|
case reflect.Bool: |
||||||
|
return !v.Bool() |
||||||
|
case reflect.Struct: |
||||||
|
vt := v.Type() |
||||||
|
for i := v.NumField() - 1; i >= 0; i-- { |
||||||
|
if vt.Field(i).PkgPath != "" { |
||||||
|
continue // Private field
|
||||||
|
} |
||||||
|
if !isZero(v.Field(i)) { |
||||||
|
return false |
||||||
|
} |
||||||
|
} |
||||||
|
return true |
||||||
|
} |
||||||
|
return false |
||||||
|
} |
@ -0,0 +1,716 @@ |
|||||||
|
package yaml |
||||||
|
|
||||||
|
import ( |
||||||
|
"io" |
||||||
|
) |
||||||
|
|
||||||
|
// The version directive data.
|
||||||
|
type yaml_version_directive_t struct { |
||||||
|
major int8 // The major version number.
|
||||||
|
minor int8 // The minor version number.
|
||||||
|
} |
||||||
|
|
||||||
|
// The tag directive data.
|
||||||
|
type yaml_tag_directive_t struct { |
||||||
|
handle []byte // The tag handle.
|
||||||
|
prefix []byte // The tag prefix.
|
||||||
|
} |
||||||
|
|
||||||
|
type yaml_encoding_t int |
||||||
|
|
||||||
|
// The stream encoding.
|
||||||
|
const ( |
||||||
|
// Let the parser choose the encoding.
|
||||||
|
yaml_ANY_ENCODING yaml_encoding_t = iota |
||||||
|
|
||||||
|
yaml_UTF8_ENCODING // The default UTF-8 encoding.
|
||||||
|
yaml_UTF16LE_ENCODING // The UTF-16-LE encoding with BOM.
|
||||||
|
yaml_UTF16BE_ENCODING // The UTF-16-BE encoding with BOM.
|
||||||
|
) |
||||||
|
|
||||||
|
type yaml_break_t int |
||||||
|
|
||||||
|
// Line break types.
|
||||||
|
const ( |
||||||
|
// Let the parser choose the break type.
|
||||||
|
yaml_ANY_BREAK yaml_break_t = iota |
||||||
|
|
||||||
|
yaml_CR_BREAK // Use CR for line breaks (Mac style).
|
||||||
|
yaml_LN_BREAK // Use LN for line breaks (Unix style).
|
||||||
|
yaml_CRLN_BREAK // Use CR LN for line breaks (DOS style).
|
||||||
|
) |
||||||
|
|
||||||
|
type yaml_error_type_t int |
||||||
|
|
||||||
|
// Many bad things could happen with the parser and emitter.
|
||||||
|
const ( |
||||||
|
// No error is produced.
|
||||||
|
yaml_NO_ERROR yaml_error_type_t = iota |
||||||
|
|
||||||
|
yaml_MEMORY_ERROR // Cannot allocate or reallocate a block of memory.
|
||||||
|
yaml_READER_ERROR // Cannot read or decode the input stream.
|
||||||
|
yaml_SCANNER_ERROR // Cannot scan the input stream.
|
||||||
|
yaml_PARSER_ERROR // Cannot parse the input stream.
|
||||||
|
yaml_COMPOSER_ERROR // Cannot compose a YAML document.
|
||||||
|
yaml_WRITER_ERROR // Cannot write to the output stream.
|
||||||
|
yaml_EMITTER_ERROR // Cannot emit a YAML stream.
|
||||||
|
) |
||||||
|
|
||||||
|
// The pointer position.
|
||||||
|
type yaml_mark_t struct { |
||||||
|
index int // The position index.
|
||||||
|
line int // The position line.
|
||||||
|
column int // The position column.
|
||||||
|
} |
||||||
|
|
||||||
|
// Node Styles
|
||||||
|
|
||||||
|
type yaml_style_t int8 |
||||||
|
|
||||||
|
type yaml_scalar_style_t yaml_style_t |
||||||
|
|
||||||
|
// Scalar styles.
|
||||||
|
const ( |
||||||
|
// Let the emitter choose the style.
|
||||||
|
yaml_ANY_SCALAR_STYLE yaml_scalar_style_t = iota |
||||||
|
|
||||||
|
yaml_PLAIN_SCALAR_STYLE // The plain scalar style.
|
||||||
|
yaml_SINGLE_QUOTED_SCALAR_STYLE // The single-quoted scalar style.
|
||||||
|
yaml_DOUBLE_QUOTED_SCALAR_STYLE // The double-quoted scalar style.
|
||||||
|
yaml_LITERAL_SCALAR_STYLE // The literal scalar style.
|
||||||
|
yaml_FOLDED_SCALAR_STYLE // The folded scalar style.
|
||||||
|
) |
||||||
|
|
||||||
|
type yaml_sequence_style_t yaml_style_t |
||||||
|
|
||||||
|
// Sequence styles.
|
||||||
|
const ( |
||||||
|
// Let the emitter choose the style.
|
||||||
|
yaml_ANY_SEQUENCE_STYLE yaml_sequence_style_t = iota |
||||||
|
|
||||||
|
yaml_BLOCK_SEQUENCE_STYLE // The block sequence style.
|
||||||
|
yaml_FLOW_SEQUENCE_STYLE // The flow sequence style.
|
||||||
|
) |
||||||
|
|
||||||
|
type yaml_mapping_style_t yaml_style_t |
||||||
|
|
||||||
|
// Mapping styles.
|
||||||
|
const ( |
||||||
|
// Let the emitter choose the style.
|
||||||
|
yaml_ANY_MAPPING_STYLE yaml_mapping_style_t = iota |
||||||
|
|
||||||
|
yaml_BLOCK_MAPPING_STYLE // The block mapping style.
|
||||||
|
yaml_FLOW_MAPPING_STYLE // The flow mapping style.
|
||||||
|
) |
||||||
|
|
||||||
|
// Tokens
|
||||||
|
|
||||||
|
type yaml_token_type_t int |
||||||
|
|
||||||
|
// Token types.
|
||||||
|
const ( |
||||||
|
// An empty token.
|
||||||
|
yaml_NO_TOKEN yaml_token_type_t = iota |
||||||
|
|
||||||
|
yaml_STREAM_START_TOKEN // A STREAM-START token.
|
||||||
|
yaml_STREAM_END_TOKEN // A STREAM-END token.
|
||||||
|
|
||||||
|
yaml_VERSION_DIRECTIVE_TOKEN // A VERSION-DIRECTIVE token.
|
||||||
|
yaml_TAG_DIRECTIVE_TOKEN // A TAG-DIRECTIVE token.
|
||||||
|
yaml_DOCUMENT_START_TOKEN // A DOCUMENT-START token.
|
||||||
|
yaml_DOCUMENT_END_TOKEN // A DOCUMENT-END token.
|
||||||
|
|
||||||
|
yaml_BLOCK_SEQUENCE_START_TOKEN // A BLOCK-SEQUENCE-START token.
|
||||||
|
yaml_BLOCK_MAPPING_START_TOKEN // A BLOCK-SEQUENCE-END token.
|
||||||
|
yaml_BLOCK_END_TOKEN // A BLOCK-END token.
|
||||||
|
|
||||||
|
yaml_FLOW_SEQUENCE_START_TOKEN // A FLOW-SEQUENCE-START token.
|
||||||
|
yaml_FLOW_SEQUENCE_END_TOKEN // A FLOW-SEQUENCE-END token.
|
||||||
|
yaml_FLOW_MAPPING_START_TOKEN // A FLOW-MAPPING-START token.
|
||||||
|
yaml_FLOW_MAPPING_END_TOKEN // A FLOW-MAPPING-END token.
|
||||||
|
|
||||||
|
yaml_BLOCK_ENTRY_TOKEN // A BLOCK-ENTRY token.
|
||||||
|
yaml_FLOW_ENTRY_TOKEN // A FLOW-ENTRY token.
|
||||||
|
yaml_KEY_TOKEN // A KEY token.
|
||||||
|
yaml_VALUE_TOKEN // A VALUE token.
|
||||||
|
|
||||||
|
yaml_ALIAS_TOKEN // An ALIAS token.
|
||||||
|
yaml_ANCHOR_TOKEN // An ANCHOR token.
|
||||||
|
yaml_TAG_TOKEN // A TAG token.
|
||||||
|
yaml_SCALAR_TOKEN // A SCALAR token.
|
||||||
|
) |
||||||
|
|
||||||
|
func (tt yaml_token_type_t) String() string { |
||||||
|
switch tt { |
||||||
|
case yaml_NO_TOKEN: |
||||||
|
return "yaml_NO_TOKEN" |
||||||
|
case yaml_STREAM_START_TOKEN: |
||||||
|
return "yaml_STREAM_START_TOKEN" |
||||||
|
case yaml_STREAM_END_TOKEN: |
||||||
|
return "yaml_STREAM_END_TOKEN" |
||||||
|
case yaml_VERSION_DIRECTIVE_TOKEN: |
||||||
|
return "yaml_VERSION_DIRECTIVE_TOKEN" |
||||||
|
case yaml_TAG_DIRECTIVE_TOKEN: |
||||||
|
return "yaml_TAG_DIRECTIVE_TOKEN" |
||||||
|
case yaml_DOCUMENT_START_TOKEN: |
||||||
|
return "yaml_DOCUMENT_START_TOKEN" |
||||||
|
case yaml_DOCUMENT_END_TOKEN: |
||||||
|
return "yaml_DOCUMENT_END_TOKEN" |
||||||
|
case yaml_BLOCK_SEQUENCE_START_TOKEN: |
||||||
|
return "yaml_BLOCK_SEQUENCE_START_TOKEN" |
||||||
|
case yaml_BLOCK_MAPPING_START_TOKEN: |
||||||
|
return "yaml_BLOCK_MAPPING_START_TOKEN" |
||||||
|
case yaml_BLOCK_END_TOKEN: |
||||||
|
return "yaml_BLOCK_END_TOKEN" |
||||||
|
case yaml_FLOW_SEQUENCE_START_TOKEN: |
||||||
|
return "yaml_FLOW_SEQUENCE_START_TOKEN" |
||||||
|
case yaml_FLOW_SEQUENCE_END_TOKEN: |
||||||
|
return "yaml_FLOW_SEQUENCE_END_TOKEN" |
||||||
|
case yaml_FLOW_MAPPING_START_TOKEN: |
||||||
|
return "yaml_FLOW_MAPPING_START_TOKEN" |
||||||
|
case yaml_FLOW_MAPPING_END_TOKEN: |
||||||
|
return "yaml_FLOW_MAPPING_END_TOKEN" |
||||||
|
case yaml_BLOCK_ENTRY_TOKEN: |
||||||
|
return "yaml_BLOCK_ENTRY_TOKEN" |
||||||
|
case yaml_FLOW_ENTRY_TOKEN: |
||||||
|
return "yaml_FLOW_ENTRY_TOKEN" |
||||||
|
case yaml_KEY_TOKEN: |
||||||
|
return "yaml_KEY_TOKEN" |
||||||
|
case yaml_VALUE_TOKEN: |
||||||
|
return "yaml_VALUE_TOKEN" |
||||||
|
case yaml_ALIAS_TOKEN: |
||||||
|
return "yaml_ALIAS_TOKEN" |
||||||
|
case yaml_ANCHOR_TOKEN: |
||||||
|
return "yaml_ANCHOR_TOKEN" |
||||||
|
case yaml_TAG_TOKEN: |
||||||
|
return "yaml_TAG_TOKEN" |
||||||
|
case yaml_SCALAR_TOKEN: |
||||||
|
return "yaml_SCALAR_TOKEN" |
||||||
|
} |
||||||
|
return "<unknown token>" |
||||||
|
} |
||||||
|
|
||||||
|
// The token structure.
|
||||||
|
type yaml_token_t struct { |
||||||
|
// The token type.
|
||||||
|
typ yaml_token_type_t |
||||||
|
|
||||||
|
// The start/end of the token.
|
||||||
|
start_mark, end_mark yaml_mark_t |
||||||
|
|
||||||
|
// The stream encoding (for yaml_STREAM_START_TOKEN).
|
||||||
|
encoding yaml_encoding_t |
||||||
|
|
||||||
|
// The alias/anchor/scalar value or tag/tag directive handle
|
||||||
|
// (for yaml_ALIAS_TOKEN, yaml_ANCHOR_TOKEN, yaml_SCALAR_TOKEN, yaml_TAG_TOKEN, yaml_TAG_DIRECTIVE_TOKEN).
|
||||||
|
value []byte |
||||||
|
|
||||||
|
// The tag suffix (for yaml_TAG_TOKEN).
|
||||||
|
suffix []byte |
||||||
|
|
||||||
|
// The tag directive prefix (for yaml_TAG_DIRECTIVE_TOKEN).
|
||||||
|
prefix []byte |
||||||
|
|
||||||
|
// The scalar style (for yaml_SCALAR_TOKEN).
|
||||||
|
style yaml_scalar_style_t |
||||||
|
|
||||||
|
// The version directive major/minor (for yaml_VERSION_DIRECTIVE_TOKEN).
|
||||||
|
major, minor int8 |
||||||
|
} |
||||||
|
|
||||||
|
// Events
|
||||||
|
|
||||||
|
type yaml_event_type_t int8 |
||||||
|
|
||||||
|
// Event types.
|
||||||
|
const ( |
||||||
|
// An empty event.
|
||||||
|
yaml_NO_EVENT yaml_event_type_t = iota |
||||||
|
|
||||||
|
yaml_STREAM_START_EVENT // A STREAM-START event.
|
||||||
|
yaml_STREAM_END_EVENT // A STREAM-END event.
|
||||||
|
yaml_DOCUMENT_START_EVENT // A DOCUMENT-START event.
|
||||||
|
yaml_DOCUMENT_END_EVENT // A DOCUMENT-END event.
|
||||||
|
yaml_ALIAS_EVENT // An ALIAS event.
|
||||||
|
yaml_SCALAR_EVENT // A SCALAR event.
|
||||||
|
yaml_SEQUENCE_START_EVENT // A SEQUENCE-START event.
|
||||||
|
yaml_SEQUENCE_END_EVENT // A SEQUENCE-END event.
|
||||||
|
yaml_MAPPING_START_EVENT // A MAPPING-START event.
|
||||||
|
yaml_MAPPING_END_EVENT // A MAPPING-END event.
|
||||||
|
) |
||||||
|
|
||||||
|
// The event structure.
|
||||||
|
type yaml_event_t struct { |
||||||
|
|
||||||
|
// The event type.
|
||||||
|
typ yaml_event_type_t |
||||||
|
|
||||||
|
// The start and end of the event.
|
||||||
|
start_mark, end_mark yaml_mark_t |
||||||
|
|
||||||
|
// The document encoding (for yaml_STREAM_START_EVENT).
|
||||||
|
encoding yaml_encoding_t |
||||||
|
|
||||||
|
// The version directive (for yaml_DOCUMENT_START_EVENT).
|
||||||
|
version_directive *yaml_version_directive_t |
||||||
|
|
||||||
|
// The list of tag directives (for yaml_DOCUMENT_START_EVENT).
|
||||||
|
tag_directives []yaml_tag_directive_t |
||||||
|
|
||||||
|
// The anchor (for yaml_SCALAR_EVENT, yaml_SEQUENCE_START_EVENT, yaml_MAPPING_START_EVENT, yaml_ALIAS_EVENT).
|
||||||
|
anchor []byte |
||||||
|
|
||||||
|
// The tag (for yaml_SCALAR_EVENT, yaml_SEQUENCE_START_EVENT, yaml_MAPPING_START_EVENT).
|
||||||
|
tag []byte |
||||||
|
|
||||||
|
// The scalar value (for yaml_SCALAR_EVENT).
|
||||||
|
value []byte |
||||||
|
|
||||||
|
// Is the document start/end indicator implicit, or the tag optional?
|
||||||
|
// (for yaml_DOCUMENT_START_EVENT, yaml_DOCUMENT_END_EVENT, yaml_SEQUENCE_START_EVENT, yaml_MAPPING_START_EVENT, yaml_SCALAR_EVENT).
|
||||||
|
implicit bool |
||||||
|
|
||||||
|
// Is the tag optional for any non-plain style? (for yaml_SCALAR_EVENT).
|
||||||
|
quoted_implicit bool |
||||||
|
|
||||||
|
// The style (for yaml_SCALAR_EVENT, yaml_SEQUENCE_START_EVENT, yaml_MAPPING_START_EVENT).
|
||||||
|
style yaml_style_t |
||||||
|
} |
||||||
|
|
||||||
|
func (e *yaml_event_t) scalar_style() yaml_scalar_style_t { return yaml_scalar_style_t(e.style) } |
||||||
|
func (e *yaml_event_t) sequence_style() yaml_sequence_style_t { return yaml_sequence_style_t(e.style) } |
||||||
|
func (e *yaml_event_t) mapping_style() yaml_mapping_style_t { return yaml_mapping_style_t(e.style) } |
||||||
|
|
||||||
|
// Nodes
|
||||||
|
|
||||||
|
const ( |
||||||
|
yaml_NULL_TAG = "tag:yaml.org,2002:null" // The tag !!null with the only possible value: null.
|
||||||
|
yaml_BOOL_TAG = "tag:yaml.org,2002:bool" // The tag !!bool with the values: true and false.
|
||||||
|
yaml_STR_TAG = "tag:yaml.org,2002:str" // The tag !!str for string values.
|
||||||
|
yaml_INT_TAG = "tag:yaml.org,2002:int" // The tag !!int for integer values.
|
||||||
|
yaml_FLOAT_TAG = "tag:yaml.org,2002:float" // The tag !!float for float values.
|
||||||
|
yaml_TIMESTAMP_TAG = "tag:yaml.org,2002:timestamp" // The tag !!timestamp for date and time values.
|
||||||
|
|
||||||
|
yaml_SEQ_TAG = "tag:yaml.org,2002:seq" // The tag !!seq is used to denote sequences.
|
||||||
|
yaml_MAP_TAG = "tag:yaml.org,2002:map" // The tag !!map is used to denote mapping.
|
||||||
|
|
||||||
|
// Not in original libyaml.
|
||||||
|
yaml_BINARY_TAG = "tag:yaml.org,2002:binary" |
||||||
|
yaml_MERGE_TAG = "tag:yaml.org,2002:merge" |
||||||
|
|
||||||
|
yaml_DEFAULT_SCALAR_TAG = yaml_STR_TAG // The default scalar tag is !!str.
|
||||||
|
yaml_DEFAULT_SEQUENCE_TAG = yaml_SEQ_TAG // The default sequence tag is !!seq.
|
||||||
|
yaml_DEFAULT_MAPPING_TAG = yaml_MAP_TAG // The default mapping tag is !!map.
|
||||||
|
) |
||||||
|
|
||||||
|
type yaml_node_type_t int |
||||||
|
|
||||||
|
// Node types.
|
||||||
|
const ( |
||||||
|
// An empty node.
|
||||||
|
yaml_NO_NODE yaml_node_type_t = iota |
||||||
|
|
||||||
|
yaml_SCALAR_NODE // A scalar node.
|
||||||
|
yaml_SEQUENCE_NODE // A sequence node.
|
||||||
|
yaml_MAPPING_NODE // A mapping node.
|
||||||
|
) |
||||||
|
|
||||||
|
// An element of a sequence node.
|
||||||
|
type yaml_node_item_t int |
||||||
|
|
||||||
|
// An element of a mapping node.
|
||||||
|
type yaml_node_pair_t struct { |
||||||
|
key int // The key of the element.
|
||||||
|
value int // The value of the element.
|
||||||
|
} |
||||||
|
|
||||||
|
// The node structure.
|
||||||
|
type yaml_node_t struct { |
||||||
|
typ yaml_node_type_t // The node type.
|
||||||
|
tag []byte // The node tag.
|
||||||
|
|
||||||
|
// The node data.
|
||||||
|
|
||||||
|
// The scalar parameters (for yaml_SCALAR_NODE).
|
||||||
|
scalar struct { |
||||||
|
value []byte // The scalar value.
|
||||||
|
length int // The length of the scalar value.
|
||||||
|
style yaml_scalar_style_t // The scalar style.
|
||||||
|
} |
||||||
|
|
||||||
|
// The sequence parameters (for YAML_SEQUENCE_NODE).
|
||||||
|
sequence struct { |
||||||
|
items_data []yaml_node_item_t // The stack of sequence items.
|
||||||
|
style yaml_sequence_style_t // The sequence style.
|
||||||
|
} |
||||||
|
|
||||||
|
// The mapping parameters (for yaml_MAPPING_NODE).
|
||||||
|
mapping struct { |
||||||
|
pairs_data []yaml_node_pair_t // The stack of mapping pairs (key, value).
|
||||||
|
pairs_start *yaml_node_pair_t // The beginning of the stack.
|
||||||
|
pairs_end *yaml_node_pair_t // The end of the stack.
|
||||||
|
pairs_top *yaml_node_pair_t // The top of the stack.
|
||||||
|
style yaml_mapping_style_t // The mapping style.
|
||||||
|
} |
||||||
|
|
||||||
|
start_mark yaml_mark_t // The beginning of the node.
|
||||||
|
end_mark yaml_mark_t // The end of the node.
|
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
// The document structure.
|
||||||
|
type yaml_document_t struct { |
||||||
|
|
||||||
|
// The document nodes.
|
||||||
|
nodes []yaml_node_t |
||||||
|
|
||||||
|
// The version directive.
|
||||||
|
version_directive *yaml_version_directive_t |
||||||
|
|
||||||
|
// The list of tag directives.
|
||||||
|
tag_directives_data []yaml_tag_directive_t |
||||||
|
tag_directives_start int // The beginning of the tag directives list.
|
||||||
|
tag_directives_end int // The end of the tag directives list.
|
||||||
|
|
||||||
|
start_implicit int // Is the document start indicator implicit?
|
||||||
|
end_implicit int // Is the document end indicator implicit?
|
||||||
|
|
||||||
|
// The start/end of the document.
|
||||||
|
start_mark, end_mark yaml_mark_t |
||||||
|
} |
||||||
|
|
||||||
|
// The prototype of a read handler.
|
||||||
|
//
|
||||||
|
// The read handler is called when the parser needs to read more bytes from the
|
||||||
|
// source. The handler should write not more than size bytes to the buffer.
|
||||||
|
// The number of written bytes should be set to the size_read variable.
|
||||||
|
//
|
||||||
|
// [in,out] data A pointer to an application data specified by
|
||||||
|
// yaml_parser_set_input().
|
||||||
|
// [out] buffer The buffer to write the data from the source.
|
||||||
|
// [in] size The size of the buffer.
|
||||||
|
// [out] size_read The actual number of bytes read from the source.
|
||||||
|
//
|
||||||
|
// On success, the handler should return 1. If the handler failed,
|
||||||
|
// the returned value should be 0. On EOF, the handler should set the
|
||||||
|
// size_read to 0 and return 1.
|
||||||
|
type yaml_read_handler_t func(parser *yaml_parser_t, buffer []byte) (n int, err error) |
||||||
|
|
||||||
|
// This structure holds information about a potential simple key.
|
||||||
|
type yaml_simple_key_t struct { |
||||||
|
possible bool // Is a simple key possible?
|
||||||
|
required bool // Is a simple key required?
|
||||||
|
token_number int // The number of the token.
|
||||||
|
mark yaml_mark_t // The position mark.
|
||||||
|
} |
||||||
|
|
||||||
|
// The states of the parser.
|
||||||
|
type yaml_parser_state_t int |
||||||
|
|
||||||
|
const ( |
||||||
|
yaml_PARSE_STREAM_START_STATE yaml_parser_state_t = iota |
||||||
|
|
||||||
|
yaml_PARSE_IMPLICIT_DOCUMENT_START_STATE // Expect the beginning of an implicit document.
|
||||||
|
yaml_PARSE_DOCUMENT_START_STATE // Expect DOCUMENT-START.
|
||||||
|
yaml_PARSE_DOCUMENT_CONTENT_STATE // Expect the content of a document.
|
||||||
|
yaml_PARSE_DOCUMENT_END_STATE // Expect DOCUMENT-END.
|
||||||
|
yaml_PARSE_BLOCK_NODE_STATE // Expect a block node.
|
||||||
|
yaml_PARSE_BLOCK_NODE_OR_INDENTLESS_SEQUENCE_STATE // Expect a block node or indentless sequence.
|
||||||
|
yaml_PARSE_FLOW_NODE_STATE // Expect a flow node.
|
||||||
|
yaml_PARSE_BLOCK_SEQUENCE_FIRST_ENTRY_STATE // Expect the first entry of a block sequence.
|
||||||
|
yaml_PARSE_BLOCK_SEQUENCE_ENTRY_STATE // Expect an entry of a block sequence.
|
||||||
|
yaml_PARSE_INDENTLESS_SEQUENCE_ENTRY_STATE // Expect an entry of an indentless sequence.
|
||||||
|
yaml_PARSE_BLOCK_MAPPING_FIRST_KEY_STATE // Expect the first key of a block mapping.
|
||||||
|
yaml_PARSE_BLOCK_MAPPING_KEY_STATE // Expect a block mapping key.
|
||||||
|
yaml_PARSE_BLOCK_MAPPING_VALUE_STATE // Expect a block mapping value.
|
||||||
|
yaml_PARSE_FLOW_SEQUENCE_FIRST_ENTRY_STATE // Expect the first entry of a flow sequence.
|
||||||
|
yaml_PARSE_FLOW_SEQUENCE_ENTRY_STATE // Expect an entry of a flow sequence.
|
||||||
|
yaml_PARSE_FLOW_SEQUENCE_ENTRY_MAPPING_KEY_STATE // Expect a key of an ordered mapping.
|
||||||
|
yaml_PARSE_FLOW_SEQUENCE_ENTRY_MAPPING_VALUE_STATE // Expect a value of an ordered mapping.
|
||||||
|
yaml_PARSE_FLOW_SEQUENCE_ENTRY_MAPPING_END_STATE // Expect the and of an ordered mapping entry.
|
||||||
|
yaml_PARSE_FLOW_MAPPING_FIRST_KEY_STATE // Expect the first key of a flow mapping.
|
||||||
|
yaml_PARSE_FLOW_MAPPING_KEY_STATE // Expect a key of a flow mapping.
|
||||||
|
yaml_PARSE_FLOW_MAPPING_VALUE_STATE // Expect a value of a flow mapping.
|
||||||
|
yaml_PARSE_FLOW_MAPPING_EMPTY_VALUE_STATE // Expect an empty value of a flow mapping.
|
||||||
|
yaml_PARSE_END_STATE // Expect nothing.
|
||||||
|
) |
||||||
|
|
||||||
|
func (ps yaml_parser_state_t) String() string { |
||||||
|
switch ps { |
||||||
|
case yaml_PARSE_STREAM_START_STATE: |
||||||
|
return "yaml_PARSE_STREAM_START_STATE" |
||||||
|
case yaml_PARSE_IMPLICIT_DOCUMENT_START_STATE: |
||||||
|
return "yaml_PARSE_IMPLICIT_DOCUMENT_START_STATE" |
||||||
|
case yaml_PARSE_DOCUMENT_START_STATE: |
||||||
|
return "yaml_PARSE_DOCUMENT_START_STATE" |
||||||
|
case yaml_PARSE_DOCUMENT_CONTENT_STATE: |
||||||
|
return "yaml_PARSE_DOCUMENT_CONTENT_STATE" |
||||||
|
case yaml_PARSE_DOCUMENT_END_STATE: |
||||||
|
return "yaml_PARSE_DOCUMENT_END_STATE" |
||||||
|
case yaml_PARSE_BLOCK_NODE_STATE: |
||||||
|
return "yaml_PARSE_BLOCK_NODE_STATE" |
||||||
|
case yaml_PARSE_BLOCK_NODE_OR_INDENTLESS_SEQUENCE_STATE: |
||||||
|
return "yaml_PARSE_BLOCK_NODE_OR_INDENTLESS_SEQUENCE_STATE" |
||||||
|
case yaml_PARSE_FLOW_NODE_STATE: |
||||||
|
return "yaml_PARSE_FLOW_NODE_STATE" |
||||||
|
case yaml_PARSE_BLOCK_SEQUENCE_FIRST_ENTRY_STATE: |
||||||
|
return "yaml_PARSE_BLOCK_SEQUENCE_FIRST_ENTRY_STATE" |
||||||
|
case yaml_PARSE_BLOCK_SEQUENCE_ENTRY_STATE: |
||||||
|
return "yaml_PARSE_BLOCK_SEQUENCE_ENTRY_STATE" |
||||||
|
case yaml_PARSE_INDENTLESS_SEQUENCE_ENTRY_STATE: |
||||||
|
return "yaml_PARSE_INDENTLESS_SEQUENCE_ENTRY_STATE" |
||||||
|
case yaml_PARSE_BLOCK_MAPPING_FIRST_KEY_STATE: |
||||||
|
return "yaml_PARSE_BLOCK_MAPPING_FIRST_KEY_STATE" |
||||||
|
case yaml_PARSE_BLOCK_MAPPING_KEY_STATE: |
||||||
|
return "yaml_PARSE_BLOCK_MAPPING_KEY_STATE" |
||||||
|
case yaml_PARSE_BLOCK_MAPPING_VALUE_STATE: |
||||||
|
return "yaml_PARSE_BLOCK_MAPPING_VALUE_STATE" |
||||||
|
case yaml_PARSE_FLOW_SEQUENCE_FIRST_ENTRY_STATE: |
||||||
|
return "yaml_PARSE_FLOW_SEQUENCE_FIRST_ENTRY_STATE" |
||||||
|
case yaml_PARSE_FLOW_SEQUENCE_ENTRY_STATE: |
||||||
|
return "yaml_PARSE_FLOW_SEQUENCE_ENTRY_STATE" |
||||||
|
case yaml_PARSE_FLOW_SEQUENCE_ENTRY_MAPPING_KEY_STATE: |
||||||
|
return "yaml_PARSE_FLOW_SEQUENCE_ENTRY_MAPPING_KEY_STATE" |
||||||
|
case yaml_PARSE_FLOW_SEQUENCE_ENTRY_MAPPING_VALUE_STATE: |
||||||
|
return "yaml_PARSE_FLOW_SEQUENCE_ENTRY_MAPPING_VALUE_STATE" |
||||||
|
case yaml_PARSE_FLOW_SEQUENCE_ENTRY_MAPPING_END_STATE: |
||||||
|
return "yaml_PARSE_FLOW_SEQUENCE_ENTRY_MAPPING_END_STATE" |
||||||
|
case yaml_PARSE_FLOW_MAPPING_FIRST_KEY_STATE: |
||||||
|
return "yaml_PARSE_FLOW_MAPPING_FIRST_KEY_STATE" |
||||||
|
case yaml_PARSE_FLOW_MAPPING_KEY_STATE: |
||||||
|
return "yaml_PARSE_FLOW_MAPPING_KEY_STATE" |
||||||
|
case yaml_PARSE_FLOW_MAPPING_VALUE_STATE: |
||||||
|
return "yaml_PARSE_FLOW_MAPPING_VALUE_STATE" |
||||||
|
case yaml_PARSE_FLOW_MAPPING_EMPTY_VALUE_STATE: |
||||||
|
return "yaml_PARSE_FLOW_MAPPING_EMPTY_VALUE_STATE" |
||||||
|
case yaml_PARSE_END_STATE: |
||||||
|
return "yaml_PARSE_END_STATE" |
||||||
|
} |
||||||
|
return "<unknown parser state>" |
||||||
|
} |
||||||
|
|
||||||
|
// This structure holds aliases data.
|
||||||
|
type yaml_alias_data_t struct { |
||||||
|
anchor []byte // The anchor.
|
||||||
|
index int // The node id.
|
||||||
|
mark yaml_mark_t // The anchor mark.
|
||||||
|
} |
||||||
|
|
||||||
|
// The parser structure.
|
||||||
|
//
|
||||||
|
// All members are internal. Manage the structure using the
|
||||||
|
// yaml_parser_ family of functions.
|
||||||
|
type yaml_parser_t struct { |
||||||
|
|
||||||
|
// Error handling
|
||||||
|
|
||||||
|
error yaml_error_type_t // Error type.
|
||||||
|
|
||||||
|
problem string // Error description.
|
||||||
|
|
||||||
|
// The byte about which the problem occured.
|
||||||
|
problem_offset int |
||||||
|
problem_value int |
||||||
|
problem_mark yaml_mark_t |
||||||
|
|
||||||
|
// The error context.
|
||||||
|
context string |
||||||
|
context_mark yaml_mark_t |
||||||
|
|
||||||
|
// Reader stuff
|
||||||
|
|
||||||
|
read_handler yaml_read_handler_t // Read handler.
|
||||||
|
|
||||||
|
input_file io.Reader // File input data.
|
||||||
|
input []byte // String input data.
|
||||||
|
input_pos int |
||||||
|
|
||||||
|
eof bool // EOF flag
|
||||||
|
|
||||||
|
buffer []byte // The working buffer.
|
||||||
|
buffer_pos int // The current position of the buffer.
|
||||||
|
|
||||||
|
unread int // The number of unread characters in the buffer.
|
||||||
|
|
||||||
|
raw_buffer []byte // The raw buffer.
|
||||||
|
raw_buffer_pos int // The current position of the buffer.
|
||||||
|
|
||||||
|
encoding yaml_encoding_t // The input encoding.
|
||||||
|
|
||||||
|
offset int // The offset of the current position (in bytes).
|
||||||
|
mark yaml_mark_t // The mark of the current position.
|
||||||
|
|
||||||
|
// Scanner stuff
|
||||||
|
|
||||||
|
stream_start_produced bool // Have we started to scan the input stream?
|
||||||
|
stream_end_produced bool // Have we reached the end of the input stream?
|
||||||
|
|
||||||
|
flow_level int // The number of unclosed '[' and '{' indicators.
|
||||||
|
|
||||||
|
tokens []yaml_token_t // The tokens queue.
|
||||||
|
tokens_head int // The head of the tokens queue.
|
||||||
|
tokens_parsed int // The number of tokens fetched from the queue.
|
||||||
|
token_available bool // Does the tokens queue contain a token ready for dequeueing.
|
||||||
|
|
||||||
|
indent int // The current indentation level.
|
||||||
|
indents []int // The indentation levels stack.
|
||||||
|
|
||||||
|
simple_key_allowed bool // May a simple key occur at the current position?
|
||||||
|
simple_keys []yaml_simple_key_t // The stack of simple keys.
|
||||||
|
|
||||||
|
// Parser stuff
|
||||||
|
|
||||||
|
state yaml_parser_state_t // The current parser state.
|
||||||
|
states []yaml_parser_state_t // The parser states stack.
|
||||||
|
marks []yaml_mark_t // The stack of marks.
|
||||||
|
tag_directives []yaml_tag_directive_t // The list of TAG directives.
|
||||||
|
|
||||||
|
// Dumper stuff
|
||||||
|
|
||||||
|
aliases []yaml_alias_data_t // The alias data.
|
||||||
|
|
||||||
|
document *yaml_document_t // The currently parsed document.
|
||||||
|
} |
||||||
|
|
||||||
|
// Emitter Definitions
|
||||||
|
|
||||||
|
// The prototype of a write handler.
|
||||||
|
//
|
||||||
|
// The write handler is called when the emitter needs to flush the accumulated
|
||||||
|
// characters to the output. The handler should write @a size bytes of the
|
||||||
|
// @a buffer to the output.
|
||||||
|
//
|
||||||
|
// @param[in,out] data A pointer to an application data specified by
|
||||||
|
// yaml_emitter_set_output().
|
||||||
|
// @param[in] buffer The buffer with bytes to be written.
|
||||||
|
// @param[in] size The size of the buffer.
|
||||||
|
//
|
||||||
|
// @returns On success, the handler should return @c 1. If the handler failed,
|
||||||
|
// the returned value should be @c 0.
|
||||||
|
//
|
||||||
|
type yaml_write_handler_t func(emitter *yaml_emitter_t, buffer []byte) error |
||||||
|
|
||||||
|
type yaml_emitter_state_t int |
||||||
|
|
||||||
|
// The emitter states.
|
||||||
|
const ( |
||||||
|
// Expect STREAM-START.
|
||||||
|
yaml_EMIT_STREAM_START_STATE yaml_emitter_state_t = iota |
||||||
|
|
||||||
|
yaml_EMIT_FIRST_DOCUMENT_START_STATE // Expect the first DOCUMENT-START or STREAM-END.
|
||||||
|
yaml_EMIT_DOCUMENT_START_STATE // Expect DOCUMENT-START or STREAM-END.
|
||||||
|
yaml_EMIT_DOCUMENT_CONTENT_STATE // Expect the content of a document.
|
||||||
|
yaml_EMIT_DOCUMENT_END_STATE // Expect DOCUMENT-END.
|
||||||
|
yaml_EMIT_FLOW_SEQUENCE_FIRST_ITEM_STATE // Expect the first item of a flow sequence.
|
||||||
|
yaml_EMIT_FLOW_SEQUENCE_ITEM_STATE // Expect an item of a flow sequence.
|
||||||
|
yaml_EMIT_FLOW_MAPPING_FIRST_KEY_STATE // Expect the first key of a flow mapping.
|
||||||
|
yaml_EMIT_FLOW_MAPPING_KEY_STATE // Expect a key of a flow mapping.
|
||||||
|
yaml_EMIT_FLOW_MAPPING_SIMPLE_VALUE_STATE // Expect a value for a simple key of a flow mapping.
|
||||||
|
yaml_EMIT_FLOW_MAPPING_VALUE_STATE // Expect a value of a flow mapping.
|
||||||
|
yaml_EMIT_BLOCK_SEQUENCE_FIRST_ITEM_STATE // Expect the first item of a block sequence.
|
||||||
|
yaml_EMIT_BLOCK_SEQUENCE_ITEM_STATE // Expect an item of a block sequence.
|
||||||
|
yaml_EMIT_BLOCK_MAPPING_FIRST_KEY_STATE // Expect the first key of a block mapping.
|
||||||
|
yaml_EMIT_BLOCK_MAPPING_KEY_STATE // Expect the key of a block mapping.
|
||||||
|
yaml_EMIT_BLOCK_MAPPING_SIMPLE_VALUE_STATE // Expect a value for a simple key of a block mapping.
|
||||||
|
yaml_EMIT_BLOCK_MAPPING_VALUE_STATE // Expect a value of a block mapping.
|
||||||
|
yaml_EMIT_END_STATE // Expect nothing.
|
||||||
|
) |
||||||
|
|
||||||
|
// The emitter structure.
|
||||||
|
//
|
||||||
|
// All members are internal. Manage the structure using the @c yaml_emitter_
|
||||||
|
// family of functions.
|
||||||
|
type yaml_emitter_t struct { |
||||||
|
|
||||||
|
// Error handling
|
||||||
|
|
||||||
|
error yaml_error_type_t // Error type.
|
||||||
|
problem string // Error description.
|
||||||
|
|
||||||
|
// Writer stuff
|
||||||
|
|
||||||
|
write_handler yaml_write_handler_t // Write handler.
|
||||||
|
|
||||||
|
output_buffer *[]byte // String output data.
|
||||||
|
output_file io.Writer // File output data.
|
||||||
|
|
||||||
|
buffer []byte // The working buffer.
|
||||||
|
buffer_pos int // The current position of the buffer.
|
||||||
|
|
||||||
|
raw_buffer []byte // The raw buffer.
|
||||||
|
raw_buffer_pos int // The current position of the buffer.
|
||||||
|
|
||||||
|
encoding yaml_encoding_t // The stream encoding.
|
||||||
|
|
||||||
|
// Emitter stuff
|
||||||
|
|
||||||
|
canonical bool // If the output is in the canonical style?
|
||||||
|
best_indent int // The number of indentation spaces.
|
||||||
|
best_width int // The preferred width of the output lines.
|
||||||
|
unicode bool // Allow unescaped non-ASCII characters?
|
||||||
|
line_break yaml_break_t // The preferred line break.
|
||||||
|
|
||||||
|
state yaml_emitter_state_t // The current emitter state.
|
||||||
|
states []yaml_emitter_state_t // The stack of states.
|
||||||
|
|
||||||
|
events []yaml_event_t // The event queue.
|
||||||
|
events_head int // The head of the event queue.
|
||||||
|
|
||||||
|
indents []int // The stack of indentation levels.
|
||||||
|
|
||||||
|
tag_directives []yaml_tag_directive_t // The list of tag directives.
|
||||||
|
|
||||||
|
indent int // The current indentation level.
|
||||||
|
|
||||||
|
flow_level int // The current flow level.
|
||||||
|
|
||||||
|
root_context bool // Is it the document root context?
|
||||||
|
sequence_context bool // Is it a sequence context?
|
||||||
|
mapping_context bool // Is it a mapping context?
|
||||||
|
simple_key_context bool // Is it a simple mapping key context?
|
||||||
|
|
||||||
|
line int // The current line.
|
||||||
|
column int // The current column.
|
||||||
|
whitespace bool // If the last character was a whitespace?
|
||||||
|
indention bool // If the last character was an indentation character (' ', '-', '?', ':')?
|
||||||
|
open_ended bool // If an explicit document end is required?
|
||||||
|
|
||||||
|
// Anchor analysis.
|
||||||
|
anchor_data struct { |
||||||
|
anchor []byte // The anchor value.
|
||||||
|
alias bool // Is it an alias?
|
||||||
|
} |
||||||
|
|
||||||
|
// Tag analysis.
|
||||||
|
tag_data struct { |
||||||
|
handle []byte // The tag handle.
|
||||||
|
suffix []byte // The tag suffix.
|
||||||
|
} |
||||||
|
|
||||||
|
// Scalar analysis.
|
||||||
|
scalar_data struct { |
||||||
|
value []byte // The scalar value.
|
||||||
|
multiline bool // Does the scalar contain line breaks?
|
||||||
|
flow_plain_allowed bool // Can the scalar be expessed in the flow plain style?
|
||||||
|
block_plain_allowed bool // Can the scalar be expressed in the block plain style?
|
||||||
|
single_quoted_allowed bool // Can the scalar be expressed in the single quoted style?
|
||||||
|
block_allowed bool // Can the scalar be expressed in the literal or folded styles?
|
||||||
|
style yaml_scalar_style_t // The output style.
|
||||||
|
} |
||||||
|
|
||||||
|
// Dumper stuff
|
||||||
|
|
||||||
|
opened bool // If the stream was already opened?
|
||||||
|
closed bool // If the stream was already closed?
|
||||||
|
|
||||||
|
// The information associated with the document nodes.
|
||||||
|
anchors *struct { |
||||||
|
references int // The number of references.
|
||||||
|
anchor int // The anchor id.
|
||||||
|
serialized bool // If the node has been emitted?
|
||||||
|
} |
||||||
|
|
||||||
|
last_anchor_id int // The last assigned anchor id.
|
||||||
|
|
||||||
|
document *yaml_document_t // The currently emitted document.
|
||||||
|
} |
@ -0,0 +1,173 @@ |
|||||||
|
package yaml |
||||||
|
|
||||||
|
const ( |
||||||
|
// The size of the input raw buffer.
|
||||||
|
input_raw_buffer_size = 512 |
||||||
|
|
||||||
|
// The size of the input buffer.
|
||||||
|
// It should be possible to decode the whole raw buffer.
|
||||||
|
input_buffer_size = input_raw_buffer_size * 3 |
||||||
|
|
||||||
|
// The size of the output buffer.
|
||||||
|
output_buffer_size = 128 |
||||||
|
|
||||||
|
// The size of the output raw buffer.
|
||||||
|
// It should be possible to encode the whole output buffer.
|
||||||
|
output_raw_buffer_size = (output_buffer_size*2 + 2) |
||||||
|
|
||||||
|
// The size of other stacks and queues.
|
||||||
|
initial_stack_size = 16 |
||||||
|
initial_queue_size = 16 |
||||||
|
initial_string_size = 16 |
||||||
|
) |
||||||
|
|
||||||
|
// Check if the character at the specified position is an alphabetical
|
||||||
|
// character, a digit, '_', or '-'.
|
||||||
|
func is_alpha(b []byte, i int) bool { |
||||||
|
return b[i] >= '0' && b[i] <= '9' || b[i] >= 'A' && b[i] <= 'Z' || b[i] >= 'a' && b[i] <= 'z' || b[i] == '_' || b[i] == '-' |
||||||
|
} |
||||||
|
|
||||||
|
// Check if the character at the specified position is a digit.
|
||||||
|
func is_digit(b []byte, i int) bool { |
||||||
|
return b[i] >= '0' && b[i] <= '9' |
||||||
|
} |
||||||
|
|
||||||
|
// Get the value of a digit.
|
||||||
|
func as_digit(b []byte, i int) int { |
||||||
|
return int(b[i]) - '0' |
||||||
|
} |
||||||
|
|
||||||
|
// Check if the character at the specified position is a hex-digit.
|
||||||
|
func is_hex(b []byte, i int) bool { |
||||||
|
return b[i] >= '0' && b[i] <= '9' || b[i] >= 'A' && b[i] <= 'F' || b[i] >= 'a' && b[i] <= 'f' |
||||||
|
} |
||||||
|
|
||||||
|
// Get the value of a hex-digit.
|
||||||
|
func as_hex(b []byte, i int) int { |
||||||
|
bi := b[i] |
||||||
|
if bi >= 'A' && bi <= 'F' { |
||||||
|
return int(bi) - 'A' + 10 |
||||||
|
} |
||||||
|
if bi >= 'a' && bi <= 'f' { |
||||||
|
return int(bi) - 'a' + 10 |
||||||
|
} |
||||||
|
return int(bi) - '0' |
||||||
|
} |
||||||
|
|
||||||
|
// Check if the character is ASCII.
|
||||||
|
func is_ascii(b []byte, i int) bool { |
||||||
|
return b[i] <= 0x7F |
||||||
|
} |
||||||
|
|
||||||
|
// Check if the character at the start of the buffer can be printed unescaped.
|
||||||
|
func is_printable(b []byte, i int) bool { |
||||||
|
return ((b[i] == 0x0A) || // . == #x0A
|
||||||
|
(b[i] >= 0x20 && b[i] <= 0x7E) || // #x20 <= . <= #x7E
|
||||||
|
(b[i] == 0xC2 && b[i+1] >= 0xA0) || // #0xA0 <= . <= #xD7FF
|
||||||
|
(b[i] > 0xC2 && b[i] < 0xED) || |
||||||
|
(b[i] == 0xED && b[i+1] < 0xA0) || |
||||||
|
(b[i] == 0xEE) || |
||||||
|
(b[i] == 0xEF && // #xE000 <= . <= #xFFFD
|
||||||
|
!(b[i+1] == 0xBB && b[i+2] == 0xBF) && // && . != #xFEFF
|
||||||
|
!(b[i+1] == 0xBF && (b[i+2] == 0xBE || b[i+2] == 0xBF)))) |
||||||
|
} |
||||||
|
|
||||||
|
// Check if the character at the specified position is NUL.
|
||||||
|
func is_z(b []byte, i int) bool { |
||||||
|
return b[i] == 0x00 |
||||||
|
} |
||||||
|
|
||||||
|
// Check if the beginning of the buffer is a BOM.
|
||||||
|
func is_bom(b []byte, i int) bool { |
||||||
|
return b[0] == 0xEF && b[1] == 0xBB && b[2] == 0xBF |
||||||
|
} |
||||||
|
|
||||||
|
// Check if the character at the specified position is space.
|
||||||
|
func is_space(b []byte, i int) bool { |
||||||
|
return b[i] == ' ' |
||||||
|
} |
||||||
|
|
||||||
|
// Check if the character at the specified position is tab.
|
||||||
|
func is_tab(b []byte, i int) bool { |
||||||
|
return b[i] == '\t' |
||||||
|
} |
||||||
|
|
||||||
|
// Check if the character at the specified position is blank (space or tab).
|
||||||
|
func is_blank(b []byte, i int) bool { |
||||||
|
//return is_space(b, i) || is_tab(b, i)
|
||||||
|
return b[i] == ' ' || b[i] == '\t' |
||||||
|
} |
||||||
|
|
||||||
|
// Check if the character at the specified position is a line break.
|
||||||
|
func is_break(b []byte, i int) bool { |
||||||
|
return (b[i] == '\r' || // CR (#xD)
|
||||||
|
b[i] == '\n' || // LF (#xA)
|
||||||
|
b[i] == 0xC2 && b[i+1] == 0x85 || // NEL (#x85)
|
||||||
|
b[i] == 0xE2 && b[i+1] == 0x80 && b[i+2] == 0xA8 || // LS (#x2028)
|
||||||
|
b[i] == 0xE2 && b[i+1] == 0x80 && b[i+2] == 0xA9) // PS (#x2029)
|
||||||
|
} |
||||||
|
|
||||||
|
func is_crlf(b []byte, i int) bool { |
||||||
|
return b[i] == '\r' && b[i+1] == '\n' |
||||||
|
} |
||||||
|
|
||||||
|
// Check if the character is a line break or NUL.
|
||||||
|
func is_breakz(b []byte, i int) bool { |
||||||
|
//return is_break(b, i) || is_z(b, i)
|
||||||
|
return ( // is_break:
|
||||||
|
b[i] == '\r' || // CR (#xD)
|
||||||
|
b[i] == '\n' || // LF (#xA)
|
||||||
|
b[i] == 0xC2 && b[i+1] == 0x85 || // NEL (#x85)
|
||||||
|
b[i] == 0xE2 && b[i+1] == 0x80 && b[i+2] == 0xA8 || // LS (#x2028)
|
||||||
|
b[i] == 0xE2 && b[i+1] == 0x80 && b[i+2] == 0xA9 || // PS (#x2029)
|
||||||
|
// is_z:
|
||||||
|
b[i] == 0) |
||||||
|
} |
||||||
|
|
||||||
|
// Check if the character is a line break, space, or NUL.
|
||||||
|
func is_spacez(b []byte, i int) bool { |
||||||
|
//return is_space(b, i) || is_breakz(b, i)
|
||||||
|
return ( // is_space:
|
||||||
|
b[i] == ' ' || |
||||||
|
// is_breakz:
|
||||||
|
b[i] == '\r' || // CR (#xD)
|
||||||
|
b[i] == '\n' || // LF (#xA)
|
||||||
|
b[i] == 0xC2 && b[i+1] == 0x85 || // NEL (#x85)
|
||||||
|
b[i] == 0xE2 && b[i+1] == 0x80 && b[i+2] == 0xA8 || // LS (#x2028)
|
||||||
|
b[i] == 0xE2 && b[i+1] == 0x80 && b[i+2] == 0xA9 || // PS (#x2029)
|
||||||
|
b[i] == 0) |
||||||
|
} |
||||||
|
|
||||||
|
// Check if the character is a line break, space, tab, or NUL.
|
||||||
|
func is_blankz(b []byte, i int) bool { |
||||||
|
//return is_blank(b, i) || is_breakz(b, i)
|
||||||
|
return ( // is_blank:
|
||||||
|
b[i] == ' ' || b[i] == '\t' || |
||||||
|
// is_breakz:
|
||||||
|
b[i] == '\r' || // CR (#xD)
|
||||||
|
b[i] == '\n' || // LF (#xA)
|
||||||
|
b[i] == 0xC2 && b[i+1] == 0x85 || // NEL (#x85)
|
||||||
|
b[i] == 0xE2 && b[i+1] == 0x80 && b[i+2] == 0xA8 || // LS (#x2028)
|
||||||
|
b[i] == 0xE2 && b[i+1] == 0x80 && b[i+2] == 0xA9 || // PS (#x2029)
|
||||||
|
b[i] == 0) |
||||||
|
} |
||||||
|
|
||||||
|
// Determine the width of the character.
|
||||||
|
func width(b byte) int { |
||||||
|
// Don't replace these by a switch without first
|
||||||
|
// confirming that it is being inlined.
|
||||||
|
if b&0x80 == 0x00 { |
||||||
|
return 1 |
||||||
|
} |
||||||
|
if b&0xE0 == 0xC0 { |
||||||
|
return 2 |
||||||
|
} |
||||||
|
if b&0xF0 == 0xE0 { |
||||||
|
return 3 |
||||||
|
} |
||||||
|
if b&0xF8 == 0xF0 { |
||||||
|
return 4 |
||||||
|
} |
||||||
|
return 0 |
||||||
|
|
||||||
|
} |
Loading…
Reference in new issue