git_fast_import/
git-fast-import.rs

1use std::{
2    env::args,
3    io::Write,
4    path::{Path, PathBuf},
5    string::ParseError,
6    time::Instant,
7};
8
9use hg_parser::{file_content, FileType, ManifestEntryDetails, MercurialRepository, Revision};
10
11fn main() -> Result<(), Error> {
12    let path: PathBuf = args().nth(1).expect("path not provided").parse()?;
13    export_repo(path)
14}
15
16fn export_repo<P: AsRef<Path>>(path: P) -> Result<(), Error> {
17    let start = Instant::now();
18    let repo = MercurialRepository::open(path)?;
19
20    let stdout = std::io::stdout();
21    let mut writer = stdout.lock();
22
23    for changeset in &repo {
24        let revision = changeset.revision;
25        eprintln!("rev: {:?}", revision);
26
27        let header = &changeset.header;
28        let mut branch = None;
29        let mut closed = false;
30        for (key, value) in &header.extra {
31            if key == b"branch" {
32                branch = Some(value.as_slice());
33            }
34
35            if key == b"close" && value == b"1" {
36                closed = true;
37            }
38        }
39
40        let mut branch: Vec<_> = branch.unwrap_or(b"master").into();
41        for b in branch.iter_mut() {
42            if *b == b' ' {
43                *b = b'-';
44            }
45        }
46
47        let user = String::from_utf8_lossy(&header.user);
48        let desc = String::from_utf8_lossy(&header.comment);
49
50        let time = header.time.timestamp_secs();
51        let timezone = header.time.tz_offset_secs();
52        let tz = format!("{:+03}{:02}", -timezone / 3600, ((-timezone % 3600) / 60));
53
54        write!(writer, "reset refs/heads/")?;
55        writer.write_all(&branch)?;
56        write!(writer, "\ncommit refs/heads/")?;
57        writer.write_all(&branch)?;
58        writeln!(writer, "\nmark :{}", mark(revision))?;
59
60        writeln!(writer, "author {} {} {}", user, time, tz)?;
61        writeln!(writer, "committer {} {} {}", user, time, tz)?;
62        writeln!(writer, "data {}", desc.len() + 1)?;
63        writeln!(writer, "{}\n", desc)?;
64
65        match (header.p1, header.p2) {
66            (Some(p1), Some(p2)) => {
67                writeln!(writer, "from :{}", mark(p1))?;
68                writeln!(writer, "merge :{}", mark(p2))?;
69            }
70            (Some(p), None) | (None, Some(p)) => {
71                writeln!(writer, "from :{}", mark(p))?;
72            }
73            _ => (),
74        }
75
76        for file in changeset.files {
77            match (file.data, file.manifest_entry) {
78                (None, None) => {
79                    write!(writer, "D ")?;
80                    writer.write_all(&file.path)?;
81                    writeln!(writer)?;
82                }
83                (Some(data), Some(manifest_entry)) => {
84                    write!(
85                        writer,
86                        "M {} inline ",
87                        match manifest_entry.details {
88                            ManifestEntryDetails::File(FileType::Symlink) => "120000",
89                            ManifestEntryDetails::File(FileType::Executable) => "100755",
90                            ManifestEntryDetails::Tree
91                            | ManifestEntryDetails::File(FileType::Regular) => "100644",
92                        }
93                    )?;
94                    writer.write_all(&file.path)?;
95                    let data = file_content(&data);
96                    writeln!(writer, "\ndata {}", data.len())?;
97                    writer.write_all(data)?;
98                }
99                _ => panic!("Wrong file data!"),
100            }
101        }
102
103        if closed {
104            write!(writer, "reset refs/tags/archive/")?;
105            writer.write_all(&branch)?;
106            writeln!(writer, "\nfrom :{}\n", mark(revision))?;
107
108            write!(writer, "reset refs/heads/")?;
109            writer.write_all(&branch)?;
110            writeln!(writer, "\nfrom 0000000000000000000000000000000000000000\n")?;
111        }
112    }
113
114    for (rev, tag) in repo.tags().unwrap() {
115        eprintln!("export tag {}", tag.name);
116        writeln!(writer, "reset refs/tags/{}", tag.name).unwrap();
117        writeln!(writer, "from :{}", mark(rev)).unwrap();
118        writeln!(writer).unwrap();
119    }
120
121    eprintln!("Done. Elapsed: {:?}", start.elapsed());
122    Ok(())
123}
124
125fn mark<R: Into<Revision>>(rev: R) -> usize {
126    (rev.into() + 1).0 as usize
127}
128
129#[derive(Debug)]
130#[allow(dead_code)]
131enum Error {
132    MercurialRepoException(hg_parser::ErrorKind),
133    Parse(ParseError),
134    IO(std::io::Error),
135}
136
137impl From<ParseError> for Error {
138    fn from(value: ParseError) -> Self {
139        Error::Parse(value)
140    }
141}
142
143impl From<std::io::Error> for Error {
144    fn from(value: std::io::Error) -> Self {
145        Error::IO(value)
146    }
147}
148
149impl From<hg_parser::ErrorKind> for Error {
150    fn from(value: hg_parser::ErrorKind) -> Self {
151        Error::MercurialRepoException(value)
152    }
153}
git_fast_import/git-fast-import.rs

git_fast_import/
git-fast-import.rs
