|
8 | 8 | "log/slog" |
9 | 9 | "net/http" |
10 | 10 | "os" |
| 11 | + "os/user" |
11 | 12 | "path/filepath" |
| 13 | + "strconv" |
12 | 14 |
|
13 | 15 | "http-pull/internal/config" |
14 | 16 | ) |
@@ -110,10 +112,83 @@ func (p *Puller) atomicWrite(r io.Reader) error { |
110 | 112 | return fmt.Errorf("closing temp file: %w", err) |
111 | 113 | } |
112 | 114 |
|
| 115 | + if err := p.applyFilePermissions(tmpPath); err != nil { |
| 116 | + os.Remove(tmpPath) |
| 117 | + return fmt.Errorf("applying file permissions: %w", err) |
| 118 | + } |
| 119 | + |
113 | 120 | if err := os.Rename(tmpPath, p.target.Destination); err != nil { |
114 | 121 | os.Remove(tmpPath) |
115 | 122 | return fmt.Errorf("renaming temp file: %w", err) |
116 | 123 | } |
117 | 124 |
|
118 | 125 | return nil |
119 | 126 | } |
| 127 | + |
| 128 | +func (p *Puller) applyFilePermissions(path string) error { |
| 129 | + fc := p.target.File |
| 130 | + if fc == nil { |
| 131 | + return nil |
| 132 | + } |
| 133 | + |
| 134 | + if fc.Mode != "" { |
| 135 | + parsed, err := strconv.ParseUint(fc.Mode, 8, 32) |
| 136 | + if err != nil { |
| 137 | + return fmt.Errorf("parsing mode %q: %w", fc.Mode, err) |
| 138 | + } |
| 139 | + if err := os.Chmod(path, os.FileMode(parsed)); err != nil { |
| 140 | + return fmt.Errorf("chmod: %w", err) |
| 141 | + } |
| 142 | + p.logger.Debug("applied file mode", "path", path, "mode", fc.Mode) |
| 143 | + } |
| 144 | + |
| 145 | + if fc.Owner != "" || fc.Group != "" { |
| 146 | + uid := -1 |
| 147 | + gid := -1 |
| 148 | + |
| 149 | + if fc.Owner != "" { |
| 150 | + id, err := resolveUID(fc.Owner) |
| 151 | + if err != nil { |
| 152 | + return fmt.Errorf("resolving owner %q: %w", fc.Owner, err) |
| 153 | + } |
| 154 | + uid = id |
| 155 | + } |
| 156 | + |
| 157 | + if fc.Group != "" { |
| 158 | + id, err := resolveGID(fc.Group) |
| 159 | + if err != nil { |
| 160 | + return fmt.Errorf("resolving group %q: %w", fc.Group, err) |
| 161 | + } |
| 162 | + gid = id |
| 163 | + } |
| 164 | + |
| 165 | + if err := os.Chown(path, uid, gid); err != nil { |
| 166 | + return fmt.Errorf("chown: %w", err) |
| 167 | + } |
| 168 | + p.logger.Debug("applied file ownership", "path", path, "uid", uid, "gid", gid) |
| 169 | + } |
| 170 | + |
| 171 | + return nil |
| 172 | +} |
| 173 | + |
| 174 | +func resolveUID(name string) (int, error) { |
| 175 | + if id, err := strconv.Atoi(name); err == nil { |
| 176 | + return id, nil |
| 177 | + } |
| 178 | + u, err := user.Lookup(name) |
| 179 | + if err != nil { |
| 180 | + return 0, err |
| 181 | + } |
| 182 | + return strconv.Atoi(u.Uid) |
| 183 | +} |
| 184 | + |
| 185 | +func resolveGID(name string) (int, error) { |
| 186 | + if id, err := strconv.Atoi(name); err == nil { |
| 187 | + return id, nil |
| 188 | + } |
| 189 | + g, err := user.LookupGroup(name) |
| 190 | + if err != nil { |
| 191 | + return 0, err |
| 192 | + } |
| 193 | + return strconv.Atoi(g.Gid) |
| 194 | +} |
0 commit comments