sam boyer | a9836df | 2016-10-18 04:39:37 | [diff] [blame] | 1 | // Copyright 2016 The Go Authors. All rights reserved. |
| 2 | // Use of this source code is governed by a BSD-style |
| 3 | // license that can be found in the LICENSE file. |
| 4 | |
Miguel Molina | 0acfd38 | 2017-01-27 20:39:07 | [diff] [blame] | 5 | package dep |
sam boyer | a9836df | 2016-10-18 04:39:37 | [diff] [blame] | 6 | |
| 7 | import ( |
Dave Cheney | 9ea8489 | 2017-05-10 04:13:22 | [diff] [blame] | 8 | "bytes" |
sam boyer | a9836df | 2016-10-18 04:39:37 | [diff] [blame] | 9 | "encoding/hex" |
sam boyer | a9836df | 2016-10-18 04:39:37 | [diff] [blame] | 10 | "io" |
Jess Frazelle | 3e7c05f | 2016-12-05 21:30:33 | [diff] [blame] | 11 | "sort" |
sam boyer | a9836df | 2016-10-18 04:39:37 | [diff] [blame] | 12 | |
Dave Cheney | 9ea8489 | 2017-05-10 04:13:22 | [diff] [blame] | 13 | "github.com/golang/dep/internal/gps" |
Carolyn Van Slyck | d3135b8 | 2017-03-20 18:13:48 | [diff] [blame] | 14 | "github.com/pelletier/go-toml" |
Carolyn Van Slyck | cfc2c94 | 2017-03-22 15:58:50 | [diff] [blame] | 15 | "github.com/pkg/errors" |
sam boyer | a9836df | 2016-10-18 04:39:37 | [diff] [blame] | 16 | ) |
| 17 | |
Carolyn Van Slyck | e31a186 | 2017-04-06 15:15:54 | [diff] [blame] | 18 | const LockName = "Gopkg.lock" |
Miguel Molina | 0acfd38 | 2017-01-27 20:39:07 | [diff] [blame] | 19 | |
| 20 | type Lock struct { |
sam boyer | 75ac569 | 2017-05-22 17:41:13 | [diff] [blame] | 21 | SolveMeta SolveMeta |
| 22 | P []gps.LockedProject |
sam boyer | 8be9bb6 | 2017-05-16 03:05:45 | [diff] [blame] | 23 | } |
| 24 | |
sam boyer | 75ac569 | 2017-05-22 17:41:13 | [diff] [blame] | 25 | type SolveMeta struct { |
sam boyer | 05c40eb | 2017-05-27 03:50:29 | [diff] [blame] | 26 | InputsDigest []byte |
sam boyer | 8be9bb6 | 2017-05-16 03:05:45 | [diff] [blame] | 27 | AnalyzerName string |
| 28 | AnalyzerVersion int |
sam boyer | 75ac569 | 2017-05-22 17:41:13 | [diff] [blame] | 29 | SolverName string |
| 30 | SolverVersion int |
sam boyer | a9836df | 2016-10-18 04:39:37 | [diff] [blame] | 31 | } |
| 32 | |
| 33 | type rawLock struct { |
sam boyer | 75ac569 | 2017-05-22 17:41:13 | [diff] [blame] | 34 | SolveMeta solveMeta `toml:"solve-meta"` |
| 35 | Projects []rawLockedProject `toml:"projects"` |
sam boyer | a9836df | 2016-10-18 04:39:37 | [diff] [blame] | 36 | } |
| 37 | |
sam boyer | 75ac569 | 2017-05-22 17:41:13 | [diff] [blame] | 38 | type solveMeta struct { |
sam boyer | 0081879 | 2017-05-23 02:35:18 | [diff] [blame] | 39 | InputsDigest string `toml:"inputs-digest"` |
sam boyer | 75ac569 | 2017-05-22 17:41:13 | [diff] [blame] | 40 | AnalyzerName string `toml:"analyzer-name"` |
| 41 | AnalyzerVersion int `toml:"analyzer-version"` |
| 42 | SolverName string `toml:"solver-name"` |
| 43 | SolverVersion int `toml:"solver-version"` |
sam boyer | a9836df | 2016-10-18 04:39:37 | [diff] [blame] | 44 | } |
| 45 | |
Carolyn Van Slyck | 28c2316 | 2017-03-21 19:53:53 | [diff] [blame] | 46 | type rawLockedProject struct { |
Carolyn Van Slyck | fb7c3fa | 2017-03-30 00:01:16 | [diff] [blame] | 47 | Name string `toml:"name"` |
| 48 | Branch string `toml:"branch,omitempty"` |
| 49 | Revision string `toml:"revision"` |
| 50 | Version string `toml:"version,omitempty"` |
| 51 | Source string `toml:"source,omitempty"` |
| 52 | Packages []string `toml:"packages"` |
sam boyer | a9836df | 2016-10-18 04:39:37 | [diff] [blame] | 53 | } |
| 54 | |
Miguel Molina | 0acfd38 | 2017-01-27 20:39:07 | [diff] [blame] | 55 | func readLock(r io.Reader) (*Lock, error) { |
Carolyn Van Slyck | fb7c3fa | 2017-03-30 00:01:16 | [diff] [blame] | 56 | buf := &bytes.Buffer{} |
| 57 | _, err := buf.ReadFrom(r) |
| 58 | if err != nil { |
| 59 | return nil, errors.Wrap(err, "Unable to read byte stream") |
| 60 | } |
| 61 | |
| 62 | raw := rawLock{} |
| 63 | err = toml.Unmarshal(buf.Bytes(), &raw) |
sam boyer | a9836df | 2016-10-18 04:39:37 | [diff] [blame] | 64 | if err != nil { |
Carolyn Van Slyck | 28c2316 | 2017-03-21 19:53:53 | [diff] [blame] | 65 | return nil, errors.Wrap(err, "Unable to parse the lock as TOML") |
sam boyer | a9836df | 2016-10-18 04:39:37 | [diff] [blame] | 66 | } |
| 67 | |
Carolyn Van Slyck | 28c2316 | 2017-03-21 19:53:53 | [diff] [blame] | 68 | return fromRawLock(raw) |
| 69 | } |
| 70 | |
| 71 | func fromRawLock(raw rawLock) (*Lock, error) { |
| 72 | var err error |
| 73 | l := &Lock{ |
| 74 | P: make([]gps.LockedProject, len(raw.Projects)), |
| 75 | } |
| 76 | |
sam boyer | 05c40eb | 2017-05-27 03:50:29 | [diff] [blame] | 77 | l.SolveMeta.InputsDigest, err = hex.DecodeString(raw.SolveMeta.InputsDigest) |
sam boyer | a9836df | 2016-10-18 04:39:37 | [diff] [blame] | 78 | if err != nil { |
Carolyn Van Slyck | cfc2c94 | 2017-03-22 15:58:50 | [diff] [blame] | 79 | return nil, errors.Errorf("invalid hash digest in lock's memo field") |
sam boyer | a9836df | 2016-10-18 04:39:37 | [diff] [blame] | 80 | } |
sam boyer | a9836df | 2016-10-18 04:39:37 | [diff] [blame] | 81 | |
sam boyer | 75ac569 | 2017-05-22 17:41:13 | [diff] [blame] | 82 | l.SolveMeta.AnalyzerName = raw.SolveMeta.AnalyzerName |
| 83 | l.SolveMeta.AnalyzerVersion = raw.SolveMeta.AnalyzerVersion |
| 84 | l.SolveMeta.SolverName = raw.SolveMeta.SolverName |
| 85 | l.SolveMeta.SolverVersion = raw.SolveMeta.SolverVersion |
sam boyer | 8be9bb6 | 2017-05-16 03:05:45 | [diff] [blame] | 86 | |
Carolyn Van Slyck | 28c2316 | 2017-03-21 19:53:53 | [diff] [blame] | 87 | for i, ld := range raw.Projects { |
sam boyer | a9836df | 2016-10-18 04:39:37 | [diff] [blame] | 88 | r := gps.Revision(ld.Revision) |
| 89 | |
Daniel Martà | dc6530b | 2017-01-24 21:48:14 | [diff] [blame] | 90 | var v gps.Version = r |
sam boyer | a9836df | 2016-10-18 04:39:37 | [diff] [blame] | 91 | if ld.Version != "" { |
| 92 | if ld.Branch != "" { |
Carolyn Van Slyck | cfc2c94 | 2017-03-22 15:58:50 | [diff] [blame] | 93 | return nil, errors.Errorf("lock file specified both a branch (%s) and version (%s) for %s", ld.Branch, ld.Version, ld.Name) |
sam boyer | a9836df | 2016-10-18 04:39:37 | [diff] [blame] | 94 | } |
| 95 | v = gps.NewVersion(ld.Version).Is(r) |
| 96 | } else if ld.Branch != "" { |
| 97 | v = gps.NewBranch(ld.Branch).Is(r) |
| 98 | } else if r == "" { |
Carolyn Van Slyck | cfc2c94 | 2017-03-22 15:58:50 | [diff] [blame] | 99 | return nil, errors.Errorf("lock file has entry for %s, but specifies no branch or version", ld.Name) |
sam boyer | a9836df | 2016-10-18 04:39:37 | [diff] [blame] | 100 | } |
| 101 | |
| 102 | id := gps.ProjectIdentifier{ |
| 103 | ProjectRoot: gps.ProjectRoot(ld.Name), |
Carolyn Van Slyck | 6ae7723 | 2017-03-07 19:47:13 | [diff] [blame] | 104 | Source: ld.Source, |
sam boyer | a9836df | 2016-10-18 04:39:37 | [diff] [blame] | 105 | } |
sam boyer | f569990 | 2016-10-18 05:18:18 | [diff] [blame] | 106 | l.P[i] = gps.NewLockedProject(id, v, ld.Packages) |
sam boyer | a9836df | 2016-10-18 04:39:37 | [diff] [blame] | 107 | } |
sam boyer | 8be9bb6 | 2017-05-16 03:05:45 | [diff] [blame] | 108 | |
sam boyer | a9836df | 2016-10-18 04:39:37 | [diff] [blame] | 109 | return l, nil |
| 110 | } |
| 111 | |
Miguel Molina | 0acfd38 | 2017-01-27 20:39:07 | [diff] [blame] | 112 | func (l *Lock) InputHash() []byte { |
sam boyer | 05c40eb | 2017-05-27 03:50:29 | [diff] [blame] | 113 | return l.SolveMeta.InputsDigest |
sam boyer | a9836df | 2016-10-18 04:39:37 | [diff] [blame] | 114 | } |
| 115 | |
Miguel Molina | 0acfd38 | 2017-01-27 20:39:07 | [diff] [blame] | 116 | func (l *Lock) Projects() []gps.LockedProject { |
sam boyer | a9836df | 2016-10-18 04:39:37 | [diff] [blame] | 117 | return l.P |
| 118 | } |
sam boyer | 881f310 | 2016-11-30 05:17:59 | [diff] [blame] | 119 | |
Carolyn Van Slyck | d3135b8 | 2017-03-20 18:13:48 | [diff] [blame] | 120 | // toRaw converts the manifest into a representation suitable to write to the lock file |
| 121 | func (l *Lock) toRaw() rawLock { |
sam boyer | 881f310 | 2016-11-30 05:17:59 | [diff] [blame] | 122 | raw := rawLock{ |
sam boyer | 75ac569 | 2017-05-22 17:41:13 | [diff] [blame] | 123 | SolveMeta: solveMeta{ |
sam boyer | 05c40eb | 2017-05-27 03:50:29 | [diff] [blame] | 124 | InputsDigest: hex.EncodeToString(l.SolveMeta.InputsDigest), |
sam boyer | 75ac569 | 2017-05-22 17:41:13 | [diff] [blame] | 125 | AnalyzerName: l.SolveMeta.AnalyzerName, |
| 126 | AnalyzerVersion: l.SolveMeta.AnalyzerVersion, |
| 127 | SolverName: l.SolveMeta.SolverName, |
| 128 | SolverVersion: l.SolveMeta.SolverVersion, |
sam boyer | 8be9bb6 | 2017-05-16 03:05:45 | [diff] [blame] | 129 | }, |
Carolyn Van Slyck | 28c2316 | 2017-03-21 19:53:53 | [diff] [blame] | 130 | Projects: make([]rawLockedProject, len(l.P)), |
sam boyer | 881f310 | 2016-11-30 05:17:59 | [diff] [blame] | 131 | } |
| 132 | |
Miguel Molina | 0acfd38 | 2017-01-27 20:39:07 | [diff] [blame] | 133 | sort.Sort(SortedLockedProjects(l.P)) |
sam boyer | 2c832bd | 2016-12-03 17:33:40 | [diff] [blame] | 134 | |
sam boyer | 881f310 | 2016-11-30 05:17:59 | [diff] [blame] | 135 | for k, lp := range l.P { |
| 136 | id := lp.Ident() |
Carolyn Van Slyck | 28c2316 | 2017-03-21 19:53:53 | [diff] [blame] | 137 | ld := rawLockedProject{ |
Carolyn Van Slyck | 6ae7723 | 2017-03-07 19:47:13 | [diff] [blame] | 138 | Name: string(id.ProjectRoot), |
| 139 | Source: id.Source, |
| 140 | Packages: lp.Packages(), |
sam boyer | 881f310 | 2016-11-30 05:17:59 | [diff] [blame] | 141 | } |
| 142 | |
sam boyer | 166b616 | 2016-11-30 16:50:57 | [diff] [blame] | 143 | v := lp.Version() |
Carolyn Van Slyck | 34bbce0 | 2017-04-11 15:38:50 | [diff] [blame] | 144 | ld.Revision, ld.Branch, ld.Version = gps.VersionComponentStrings(v) |
sam boyer | 166b616 | 2016-11-30 16:50:57 | [diff] [blame] | 145 | |
Carolyn Van Slyck | 28c2316 | 2017-03-21 19:53:53 | [diff] [blame] | 146 | raw.Projects[k] = ld |
sam boyer | 881f310 | 2016-11-30 05:17:59 | [diff] [blame] | 147 | } |
| 148 | |
sam boyer | d1c74a8 | 2016-11-30 17:50:41 | [diff] [blame] | 149 | // TODO sort output - #15 |
sam boyer | 20bf102 | 2016-12-01 00:37:28 | [diff] [blame] | 150 | |
Carolyn Van Slyck | d3135b8 | 2017-03-20 18:13:48 | [diff] [blame] | 151 | return raw |
| 152 | } |
| 153 | |
Carolyn Van Slyck | 9065775 | 2017-04-02 20:20:21 | [diff] [blame] | 154 | func (l *Lock) MarshalTOML() ([]byte, error) { |
Carolyn Van Slyck | d3135b8 | 2017-03-20 18:13:48 | [diff] [blame] | 155 | raw := l.toRaw() |
Carolyn Van Slyck | fb7c3fa | 2017-03-30 00:01:16 | [diff] [blame] | 156 | result, err := toml.Marshal(raw) |
Carolyn Van Slyck | 9065775 | 2017-04-02 20:20:21 | [diff] [blame] | 157 | return result, errors.Wrap(err, "Unable to marshal lock to TOML string") |
Carolyn Van Slyck | d3135b8 | 2017-03-20 18:13:48 | [diff] [blame] | 158 | } |
| 159 | |
sam boyer | 2081136 | 2017-05-16 03:12:06 | [diff] [blame] | 160 | // LockFromSolution converts a gps.Solution to dep's representation of a lock. |
sam boyer | 7b1fbdb | 2016-12-01 05:25:48 | [diff] [blame] | 161 | // |
| 162 | // Data is defensively copied wherever necessary to ensure the resulting *lock |
| 163 | // shares no memory with the original lock. |
sam boyer | 2081136 | 2017-05-16 03:12:06 | [diff] [blame] | 164 | func LockFromSolution(in gps.Solution) *Lock { |
sam boyer | 7b1fbdb | 2016-12-01 05:25:48 | [diff] [blame] | 165 | h, p := in.InputHash(), in.Projects() |
| 166 | |
Miguel Molina | 0acfd38 | 2017-01-27 20:39:07 | [diff] [blame] | 167 | l := &Lock{ |
sam boyer | 75ac569 | 2017-05-22 17:41:13 | [diff] [blame] | 168 | SolveMeta: SolveMeta{ |
sam boyer | 05c40eb | 2017-05-27 03:50:29 | [diff] [blame] | 169 | InputsDigest: make([]byte, len(h)), |
sam boyer | 8be9bb6 | 2017-05-16 03:05:45 | [diff] [blame] | 170 | AnalyzerName: in.AnalyzerName(), |
| 171 | AnalyzerVersion: in.AnalyzerVersion(), |
sam boyer | 75ac569 | 2017-05-22 17:41:13 | [diff] [blame] | 172 | SolverName: in.SolverName(), |
| 173 | SolverVersion: in.SolverVersion(), |
sam boyer | 8be9bb6 | 2017-05-16 03:05:45 | [diff] [blame] | 174 | }, |
| 175 | P: make([]gps.LockedProject, len(p)), |
sam boyer | 7b1fbdb | 2016-12-01 05:25:48 | [diff] [blame] | 176 | } |
| 177 | |
sam boyer | 05c40eb | 2017-05-27 03:50:29 | [diff] [blame] | 178 | copy(l.SolveMeta.InputsDigest, h) |
sam boyer | 7b1fbdb | 2016-12-01 05:25:48 | [diff] [blame] | 179 | copy(l.P, p) |
| 180 | return l |
| 181 | } |
sam boyer | 2c832bd | 2016-12-03 17:33:40 | [diff] [blame] | 182 | |
Miguel Molina | 0acfd38 | 2017-01-27 20:39:07 | [diff] [blame] | 183 | type SortedLockedProjects []gps.LockedProject |
sam boyer | 2c832bd | 2016-12-03 17:33:40 | [diff] [blame] | 184 | |
Miguel Molina | 0acfd38 | 2017-01-27 20:39:07 | [diff] [blame] | 185 | func (s SortedLockedProjects) Len() int { return len(s) } |
| 186 | func (s SortedLockedProjects) Swap(i, j int) { s[i], s[j] = s[j], s[i] } |
| 187 | func (s SortedLockedProjects) Less(i, j int) bool { |
sam boyer | 2c832bd | 2016-12-03 17:33:40 | [diff] [blame] | 188 | l, r := s[i].Ident(), s[j].Ident() |
| 189 | |
| 190 | if l.ProjectRoot < r.ProjectRoot { |
| 191 | return true |
| 192 | } |
| 193 | if r.ProjectRoot < l.ProjectRoot { |
| 194 | return false |
| 195 | } |
| 196 | |
sam boyer | 5b2ff11 | 2017-01-04 02:19:24 | [diff] [blame] | 197 | return l.Source < r.Source |
sam boyer | 2c832bd | 2016-12-03 17:33:40 | [diff] [blame] | 198 | } |