| 1 | """Function/variables commmon to all the commands |
| 2 | """ |
| 3 | |
| 4 | __copyright__ = """ |
| 5 | Copyright (C) 2005, Catalin Marinas <catalin.marinas@gmail.com> |
| 6 | |
| 7 | This program is free software; you can redistribute it and/or modify |
| 8 | it under the terms of the GNU General Public License version 2 as |
| 9 | published by the Free Software Foundation. |
| 10 | |
| 11 | This program is distributed in the hope that it will be useful, |
| 12 | but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 13 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| 14 | GNU General Public License for more details. |
| 15 | |
| 16 | You should have received a copy of the GNU General Public License |
| 17 | along with this program; if not, write to the Free Software |
| 18 | Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA |
| 19 | """ |
| 20 | |
| 21 | import sys, os, re |
| 22 | from optparse import OptionParser, make_option |
| 23 | |
| 24 | from stgit.utils import * |
| 25 | from stgit import stack, git |
| 26 | |
| 27 | |
| 28 | # Command exception class |
| 29 | class CmdException(Exception): |
| 30 | pass |
| 31 | |
| 32 | |
| 33 | # Utility functions |
| 34 | def git_id(string): |
| 35 | """Return the GIT id |
| 36 | """ |
| 37 | if not string: |
| 38 | return None |
| 39 | |
| 40 | string_list = string.split('/') |
| 41 | if len(string_list) == 2: |
| 42 | patch_id = string_list[1] |
| 43 | if not patch_id: |
| 44 | patch_id = 'top' |
| 45 | elif len(string_list) == 1: |
| 46 | patch_id = 'top' |
| 47 | else: |
| 48 | patch_id = None |
| 49 | |
| 50 | patch_branch = string_list[0].split('@') |
| 51 | if len(patch_branch) == 1: |
| 52 | series = crt_series |
| 53 | elif len(patch_branch) == 2: |
| 54 | series = stack.Series(patch_branch[1]) |
| 55 | else: |
| 56 | raise CmdException, 'Unknown id: %s' % string |
| 57 | |
| 58 | patch_name = patch_branch[0] |
| 59 | if not patch_name: |
| 60 | patch_name = series.get_current() |
| 61 | if not patch_name: |
| 62 | raise CmdException, 'No patches applied' |
| 63 | |
| 64 | # patch |
| 65 | if patch_name in series.get_applied() \ |
| 66 | or patch_name in series.get_unapplied(): |
| 67 | if patch_id == 'top': |
| 68 | return series.get_patch(patch_name).get_top() |
| 69 | elif patch_id == 'bottom': |
| 70 | return series.get_patch(patch_name).get_bottom() |
| 71 | |
| 72 | # base |
| 73 | if patch_name == 'base' and len(string_list) == 1: |
| 74 | return read_string(series.get_base_file()) |
| 75 | |
| 76 | # anything else failed |
| 77 | return git.rev_parse(string) |
| 78 | |
| 79 | def check_local_changes(): |
| 80 | if git.local_changes(): |
| 81 | raise CmdException, \ |
| 82 | 'local changes in the tree. Use "refresh" to commit them' |
| 83 | |
| 84 | def check_head_top_equal(): |
| 85 | if not crt_series.head_top_equal(): |
| 86 | raise CmdException, \ |
| 87 | 'HEAD and top are not the same. You probably committed\n' \ |
| 88 | ' changes to the tree ouside of StGIT. If you know what you\n' \ |
| 89 | ' are doing, use the "refresh -f" command' |
| 90 | |
| 91 | def check_conflicts(): |
| 92 | if os.path.exists(os.path.join(git.base_dir, 'conflicts')): |
| 93 | raise CmdException, 'Unsolved conflicts. Please resolve them first' |
| 94 | |
| 95 | def print_crt_patch(): |
| 96 | patch = crt_series.get_current() |
| 97 | if patch: |
| 98 | print 'Now at patch "%s"' % patch |
| 99 | else: |
| 100 | print 'No patches applied' |
| 101 | |
| 102 | def resolved(filename, reset = None): |
| 103 | if reset: |
| 104 | reset_file = filename + '.' + reset |
| 105 | if os.path.isfile(reset_file): |
| 106 | if os.path.isfile(filename): |
| 107 | os.remove(filename) |
| 108 | os.rename(reset_file, filename) |
| 109 | |
| 110 | git.update_cache([filename], force = True) |
| 111 | |
| 112 | for ext in ['.local', '.older', '.remote']: |
| 113 | fn = filename + ext |
| 114 | if os.path.isfile(fn): |
| 115 | os.remove(fn) |
| 116 | |
| 117 | def resolved_all(reset = None): |
| 118 | conflicts = git.get_conflicts() |
| 119 | if conflicts: |
| 120 | for filename in conflicts: |
| 121 | resolved(filename, reset) |
| 122 | os.remove(os.path.join(git.base_dir, 'conflicts')) |
| 123 | |
| 124 | def name_email(string): |
| 125 | """Return a tuple consisting of the name and email parsed from a |
| 126 | standard 'name <email>' string |
| 127 | """ |
| 128 | string = re.sub('([^\w\s<>@.])', '\\\\\\1', string) |
| 129 | str_list = re.findall('^(.*)\s*<(.*)>\s*$', string) |
| 130 | if not str_list: |
| 131 | raise CmdException, 'Incorrect "name <email>" string: %s' % string |
| 132 | |
| 133 | return str_list[0] |
| 134 | |
| 135 | def name_email_date(string): |
| 136 | """Return a tuple consisting of the name, email and date parsed |
| 137 | from a 'name <email> date' string |
| 138 | """ |
| 139 | string = re.sub('([^\w\s<>@.])', '\\\\\\1', string) |
| 140 | str_list = re.findall('^(.*)\s*<(.*)>\s*(.*)\s*$', string) |
| 141 | if not str_list: |
| 142 | raise CmdException, 'Incorrect "name <email> date" string: %s' % string |
| 143 | |
| 144 | return str_list[0] |