| 1 | #ifndef WORKTREE_H |
| 2 | #define WORKTREE_H |
| 3 | |
| 4 | #include "refs.h" |
| 5 | |
| 6 | struct strbuf; |
| 7 | |
| 8 | struct worktree { |
| 9 | /* The repository this worktree belongs to. */ |
| 10 | struct repository *repo; |
| 11 | char *path; |
| 12 | char *id; |
| 13 | char *head_ref; /* NULL if HEAD is broken or detached */ |
| 14 | char *lock_reason; /* private - use worktree_lock_reason */ |
| 15 | char *prune_reason; /* private - use worktree_prune_reason */ |
| 16 | struct object_id head_oid; |
| 17 | int is_detached; |
| 18 | int is_bare; |
| 19 | int is_current; /* does `path` match `repo->worktree` */ |
| 20 | int lock_reason_valid; /* private */ |
| 21 | int prune_reason_valid; /* private */ |
| 22 | }; |
| 23 | |
| 24 | /* |
| 25 | * Get the worktrees. The primary worktree will always be the first returned, |
| 26 | * and linked worktrees will follow in no particular order. |
| 27 | * |
| 28 | * The caller is responsible for freeing the memory from the returned |
| 29 | * worktrees by calling free_worktrees(). |
| 30 | */ |
| 31 | struct worktree **get_worktrees(void); |
| 32 | |
| 33 | /* |
| 34 | * Like `get_worktrees`, but does not read HEAD. Skip reading HEAD allows to |
| 35 | * get the worktree without worrying about failures pertaining to parsing |
| 36 | * the HEAD ref. This is useful in contexts where it is assumed that the |
| 37 | * refdb may not be in a consistent state. |
| 38 | */ |
| 39 | struct worktree **get_worktrees_without_reading_head(void); |
| 40 | |
| 41 | /* |
| 42 | * Construct a struct worktree corresponding to repo->gitdir and |
| 43 | * repo->worktree. |
| 44 | */ |
| 45 | struct worktree *get_current_worktree(struct repository *repo); |
| 46 | |
| 47 | /* |
| 48 | * Returns 1 if linked worktrees exist, 0 otherwise. |
| 49 | */ |
| 50 | int submodule_uses_worktrees(const char *path); |
| 51 | |
| 52 | /* |
| 53 | * Return git dir of the worktree. Note that the path may be relative. |
| 54 | */ |
| 55 | char *get_worktree_git_dir(const struct worktree *wt); |
| 56 | |
| 57 | /* |
| 58 | * Search for the worktree identified unambiguously by `arg` -- typically |
| 59 | * supplied by the user via the command-line -- which may be a pathname or some |
| 60 | * shorthand uniquely identifying a worktree, thus making it convenient for the |
| 61 | * user to specify a worktree with minimal typing. For instance, if the last |
| 62 | * component (say, "foo") of a worktree's pathname is unique among worktrees |
| 63 | * (say, "work/foo" and "work/bar"), it can be used to identify the worktree |
| 64 | * unambiguously. |
| 65 | * |
| 66 | * `prefix` should be the `prefix` handed to top-level Git commands along with |
| 67 | * `argc` and `argv`. |
| 68 | * |
| 69 | * Return the worktree identified by `arg`, or NULL if not found. |
| 70 | */ |
| 71 | struct worktree *find_worktree(struct worktree **list, |
| 72 | const char *prefix, |
| 73 | const char *arg); |
| 74 | |
| 75 | /* |
| 76 | * Look up the worktree corresponding to `id`, or NULL of no such worktree |
| 77 | * exists. |
| 78 | */ |
| 79 | struct worktree *get_linked_worktree(const char *id, |
| 80 | int skip_reading_head); |
| 81 | |
| 82 | /* |
| 83 | * Return the worktree corresponding to `path`, or NULL if no such worktree |
| 84 | * exists. |
| 85 | */ |
| 86 | struct worktree *find_worktree_by_path(struct worktree **, const char *path); |
| 87 | |
| 88 | /* |
| 89 | * Return true if the given worktree is the main one. |
| 90 | */ |
| 91 | int is_main_worktree(const struct worktree *wt); |
| 92 | |
| 93 | /* |
| 94 | * Return the reason string if the given worktree is locked or NULL |
| 95 | * otherwise. |
| 96 | */ |
| 97 | const char *worktree_lock_reason(struct worktree *wt); |
| 98 | |
| 99 | /* |
| 100 | * Return the reason string if the given worktree should be pruned, otherwise |
| 101 | * NULL if it should not be pruned. `expire` defines a grace period to prune |
| 102 | * the worktree when its path does not exist. |
| 103 | */ |
| 104 | const char *worktree_prune_reason(struct worktree *wt, timestamp_t expire); |
| 105 | |
| 106 | /* |
| 107 | * Return true if worktree entry should be pruned, along with the reason for |
| 108 | * pruning. Otherwise, return false and the worktree's path in `wtpath`, or |
| 109 | * NULL if it cannot be determined. Caller is responsible for freeing |
| 110 | * returned path. |
| 111 | * |
| 112 | * `expire` defines a grace period to prune the worktree when its path |
| 113 | * does not exist. |
| 114 | */ |
| 115 | int should_prune_worktree(const char *id, |
| 116 | struct strbuf *reason, |
| 117 | char **wtpath, |
| 118 | timestamp_t expire); |
| 119 | |
| 120 | #define WT_VALIDATE_WORKTREE_MISSING_OK (1 << 0) |
| 121 | |
| 122 | /* |
| 123 | * Return zero if the worktree is in good condition. Error message is |
| 124 | * returned if "errmsg" is not NULL. |
| 125 | */ |
| 126 | int validate_worktree(const struct worktree *wt, |
| 127 | struct strbuf *errmsg, |
| 128 | unsigned flags); |
| 129 | |
| 130 | /* |
| 131 | * Update worktrees/xxx/gitdir with the new path. |
| 132 | */ |
| 133 | void update_worktree_location(struct worktree *wt, const char *path_, |
| 134 | int use_relative_paths); |
| 135 | |
| 136 | typedef void (* worktree_repair_fn)(int iserr, const char *path, |
| 137 | const char *msg, void *cb_data); |
| 138 | |
| 139 | /* |
| 140 | * Visit each registered linked worktree and repair corruptions. For each |
| 141 | * repair made or error encountered while attempting a repair, the callback |
| 142 | * function, if non-NULL, is called with the path of the worktree and a |
| 143 | * description of the repair or error, along with the callback user-data. |
| 144 | */ |
| 145 | void repair_worktrees(worktree_repair_fn, void *cb_data, int use_relative_paths); |
| 146 | |
| 147 | /* |
| 148 | * Repair the linked worktrees after the gitdir has been moved. |
| 149 | */ |
| 150 | void repair_worktrees_after_gitdir_move(const char *old_path); |
| 151 | |
| 152 | /* |
| 153 | * Repair the linked worktree after the gitdir has been moved. |
| 154 | */ |
| 155 | void repair_worktree_after_gitdir_move(struct worktree *wt, const char *old_path); |
| 156 | |
| 157 | /* |
| 158 | * Repair administrative files corresponding to the worktree at the given path. |
| 159 | * The worktree's .git file pointing at the repository must be intact for the |
| 160 | * repair to succeed. Useful for re-associating an orphaned worktree with the |
| 161 | * repository if the worktree has been moved manually (without using "git |
| 162 | * worktree move"). For each repair made or error encountered while attempting |
| 163 | * a repair, the callback function, if non-NULL, is called with the path of the |
| 164 | * worktree and a description of the repair or error, along with the callback |
| 165 | * user-data. |
| 166 | */ |
| 167 | void repair_worktree_at_path(const char *, worktree_repair_fn, |
| 168 | void *cb_data, int use_relative_paths); |
| 169 | |
| 170 | /* |
| 171 | * Free up the memory for a worktree. |
| 172 | */ |
| 173 | void free_worktree(struct worktree *); |
| 174 | |
| 175 | /* |
| 176 | * Free up the memory for worktree(s) |
| 177 | */ |
| 178 | void free_worktrees(struct worktree **); |
| 179 | |
| 180 | /* |
| 181 | * Check if a per-worktree symref points to a ref in the main worktree |
| 182 | * or any linked worktree, and return the worktree that holds the ref, |
| 183 | * or NULL otherwise. |
| 184 | */ |
| 185 | const struct worktree *find_shared_symref(struct worktree **worktrees, |
| 186 | const char *symref, |
| 187 | const char *target); |
| 188 | |
| 189 | /* |
| 190 | * Returns true if a symref points to a ref in a worktree. |
| 191 | */ |
| 192 | int is_shared_symref(const struct worktree *wt, |
| 193 | const char *symref, const char *target); |
| 194 | |
| 195 | /* |
| 196 | * Similar to head_ref() for all HEADs _except_ one from the current |
| 197 | * worktree, which is covered by head_ref(). |
| 198 | */ |
| 199 | int other_head_refs(refs_for_each_cb fn, void *cb_data); |
| 200 | |
| 201 | int is_worktree_being_rebased(const struct worktree *wt, const char *target); |
| 202 | int is_worktree_being_bisected(const struct worktree *wt, const char *target); |
| 203 | |
| 204 | /* |
| 205 | * Return a refname suitable for access from the current ref store. |
| 206 | */ |
| 207 | void strbuf_worktree_ref(const struct worktree *wt, |
| 208 | struct strbuf *sb, |
| 209 | const char *refname); |
| 210 | |
| 211 | /** |
| 212 | * Enable worktree config for the first time. This will make the following |
| 213 | * adjustments: |
| 214 | * |
| 215 | * 1. Add extensions.worktreeConfig=true in the common config file. |
| 216 | * |
| 217 | * 2. If the common config file has a core.worktree value, then that value |
| 218 | * is moved to the main worktree's config.worktree file. |
| 219 | * |
| 220 | * 3. If the common config file has a core.bare enabled, then that value |
| 221 | * is moved to the main worktree's config.worktree file. |
| 222 | * |
| 223 | * If extensions.worktreeConfig is already true, then this method |
| 224 | * terminates early without any of the above steps. The existing config |
| 225 | * arrangement is assumed to be intentional. |
| 226 | * |
| 227 | * Returns 0 on success. Reports an error message and returns non-zero |
| 228 | * if any of these steps fail. |
| 229 | */ |
| 230 | int init_worktree_config(struct repository *r); |
| 231 | |
| 232 | /** |
| 233 | * Write the .git file and gitdir file that links the worktree to the repository. |
| 234 | * |
| 235 | * The `dotgit` parameter is the path to the worktree's .git file, and `gitdir` |
| 236 | * is the path to the repository's `gitdir` file. |
| 237 | * |
| 238 | * Example |
| 239 | * dotgit: "/path/to/foo/.git" |
| 240 | * gitdir: "/path/to/repo/worktrees/foo/gitdir" |
| 241 | */ |
| 242 | void write_worktree_linking_files(const char *dotgit, const char *gitdir, |
| 243 | int use_relative_paths); |
| 244 | |
| 245 | #endif |