diff --git a/src/cp.zig b/src/cp.zig index 48248e8..8795895 100644 --- a/src/cp.zig +++ b/src/cp.zig @@ -14,6 +14,12 @@ fn help() !void { try stdout.print("または: cp [OPTION]... SOURCE... DIRECTORY\n", .{}); try stdout.print("または: cp [OPTION]... -t DIRECTORY SOURCE...\n", .{}); try stdout.print("SOURCE から DEST へのコピー、または複数の SOURCE の DIRECTORY へのコピーを行います。\n\n", .{}); + try stdout.print("-e ファイルが既に存在したら、上書かない\n", .{}); + //try stdout.print("-f コピー先にファイルが存在し、開くことができない場合、\n 削除してから再度試みる (-e オプションも指定した\n 場合はこのオプションは無視される)\n", .{}); + try stdout.print("-i 上書きする前に確認する (前に指定した -n オプション\n を上書きする)\n", .{}); + try stdout.print("-n 出力しない\n", .{}); + //try stdout.print("-p 許可、所有者、とタイムスタンプを保管\n", .{}); + //try stdout.print("-r 再帰的にディレクトリをコピーする\n", .{}); try stdout.print("-h ヘルプを表示\n", .{}); try stdout.print("-v バージョンを表示\n", .{}); @@ -47,6 +53,13 @@ pub fn main() !void { } } + var isnover: bool = false; + var isforce: bool = false; + var ischeck: bool = false; + var isigout: bool = false; + var isprsrv: bool = false; + var isrecur: bool = false; + for (option.items) |i| { if (i == 'h') { try help(); @@ -56,18 +69,90 @@ pub fn main() !void { try version.ver("cp"); return; } + if (i == 'e') isnover = true; + if (i == 'f') isforce = true; // TODO + if (i == 'i') ischeck = true; + if (i == 'n') isigout = true; + if (i == 'p') isprsrv = true; // TODO + if (i == 'r') isrecur = true; // TODO } + if (isnover and ischeck) ischeck = false; + if (isnover and isforce) isnover = false; + if (fname.items.len != 2) { try help(); return; } + const stdof = io.getStdOut().writer(); + var bw = io.bufferedWriter(stdof); + const stdout = bw.writer(); + const in = try fs.cwd().openFile(fname.items[0], .{}); defer in.close(); + const instat = try in.stat(); + + const chk = fs.cwd().openFile(fname.items[1], .{}); + if (chk) |f| { + f.close(); + if (isnover) { + try stdout.print("cp: '{s}' は既に存在しています。コピーしません。\n", .{fname.items[1]}); + try bw.flush(); + return; + } + } else |e| { + switch (e) { + error.FileNotFound => {}, + else => {}, + } + } + const out = try fs.cwd().createFile(fname.items[1], .{ .read = true }); defer out.close(); + if (instat.kind != .file) { + if (isrecur) { + if (ischeck) { + try stdout.print("cp: '{s}' をコピーしますか?\n", .{fname.items[0]}); + try bw.flush(); + if (!(try concpy())) { + if (!isigout) { + try stdout.print("cp: '{s}' をコピーしませんでした。\n", .{fname.items[0]}); + try bw.flush(); + } + return; + } + } + try copyfile(in, out); + if (!isigout) { + try stdout.print("cp: '{s}' は '{s}' にコピーしました。\n", .{ fname.items[0], fname.items[1] }); + } + } else { + try stdout.print("cp: '{s}' をコピーできません: ディレクトリです\n", .{fname.items[0]}); + } + } else { + if (ischeck) { + try stdout.print("cp: '{s}' を上書きしますか?\n", .{fname.items[1]}); + try bw.flush(); + if (!(try concpy())) { + if (!isigout) { + try stdout.print("cp: '{s}' をコピーしませんでした。\n", .{fname.items[0]}); + try bw.flush(); + } + return; + } + } + try copyfile(in, out); + if (!isigout) { + try stdout.print("cp: '{s}' は '{s}' にコピーしました。\n", .{ fname.items[0], fname.items[1] }); + } + } + + try bw.flush(); +} + +fn copyfile(in: fs.File, out: fs.File) !void { var buf: [1024]u8 = undefined; while (true) { const br = try in.read(buf[0..]); @@ -75,3 +160,22 @@ pub fn main() !void { try out.writeAll(buf[0..br]); } } + +fn concpy() !bool { + const stdin = io.getStdIn(); + var buf = io.bufferedReader(stdin.reader()); + var r = buf.reader(); + var cbuf: [10]u8 = undefined; + var input = try r.readUntilDelimiterOrEof(&cbuf, '\n'); + var ly: [1]u8 = [_]u8{'y'}; + var uy: [1]u8 = [_]u8{'Y'}; + if (input) |i| { + if (i.len == 1) { + return (std.mem.eql(u8, i[0..1], &ly) or std.mem.eql(u8, i[0..1], &uy)); + } else { + return false; + } + } else { + return false; + } +}