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