60 lines
930 B
Go
60 lines
930 B
Go
|
package bash
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"io"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Encoder implements parser.Encoder
|
||
|
|
type Encoder struct {
|
||
|
|
writer io.Writer
|
||
|
|
prefix []byte
|
||
|
|
indent []byte
|
||
|
|
}
|
||
|
|
|
||
|
|
// Encode is the main function of the parser.Encoder interface
|
||
|
|
func (e *Encoder) Encode(v interface{}) error {
|
||
|
|
|
||
|
|
// check 'v'
|
||
|
|
vcast, ok := v.(*File)
|
||
|
|
if !ok {
|
||
|
|
return ErrInvalidReceiver
|
||
|
|
}
|
||
|
|
|
||
|
|
// empty config
|
||
|
|
if len(vcast.Lines) < 1 {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, line := range vcast.Lines {
|
||
|
|
|
||
|
|
// line representation
|
||
|
|
repr := ""
|
||
|
|
|
||
|
|
// ASSIGNMENT
|
||
|
|
if line.Type == ASSIGNMENT {
|
||
|
|
repr = fmt.Sprintf("%s%s=%s;", line.Components[0], line.Components[1], line.Components[2])
|
||
|
|
|
||
|
|
// optional comment
|
||
|
|
if len(line.Components[3]) > 0 {
|
||
|
|
repr += fmt.Sprintf(" #%s", line.Components[3])
|
||
|
|
}
|
||
|
|
|
||
|
|
// ANY
|
||
|
|
} else {
|
||
|
|
repr = strings.Join(line.Components, "")
|
||
|
|
}
|
||
|
|
|
||
|
|
repr += "\n"
|
||
|
|
|
||
|
|
_, err := e.writer.Write([]byte(repr))
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|