File: autoCom.ml

package info (click to toggle)
herdtools7 7.58-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 19,732 kB
  • sloc: ml: 128,583; ansic: 3,827; makefile: 670; python: 407; sh: 212; awk: 14
file content (93 lines) | stat: -rw-r--r-- 3,065 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
(****************************************************************************)
(*                           the diy toolsuite                              *)
(*                                                                          *)
(* Jade Alglave, University College London, UK.                             *)
(* Luc Maranget, INRIA Paris-Rocquencourt, France.                          *)
(*                                                                          *)
(* Copyright 2015-present Institut National de Recherche en Informatique et *)
(* en Automatique and the authors. All rights reserved.                     *)
(*                                                                          *)
(* This software is governed by the CeCILL-B license under French law and   *)
(* abiding by the rules of distribution of free software. You can use,      *)
(* modify and/ or redistribute the software under the terms of the CeCILL-B *)
(* license as circulated by CEA, CNRS and INRIA at the following URL        *)
(* "http://www.cecill.info". We also give a copy in LICENSE.txt.            *)
(****************************************************************************)
open Printf

type mach = Local | Distant of string

module type Config = sig
  val addpath : string list
  val target : mach
end

module type S =  sig
  val dist_sh : string -> unit
  val dist_sh_save : string -> string -> unit
  val dist_upload : string -> string -> unit
  val dist_download : string -> string -> unit
end


module Make(C:Config) = struct

  let path = match C.addpath with
  | [] -> ""
  | ds ->
      "PATH=" ^
      List.fold_right
        (fun d k -> sprintf "%s:%s" d k)
        ds "$PATH && "

  module LocalCom = struct

    let dist_sh com = MySys.exec_stdout (sprintf "sh -c '%s'" com)

    let dist_sh_save com name =
      MySys.exec_stdout (sprintf "sh -c '%s' > %s" com name)

    let dist_upload f t = MySys.exec_stdout (sprintf "cp %s %s" f t)

    let dist_download _ _ = assert false
  end


  module DistCom (C:sig val addr : string end) =
    struct
      let ssh = "ssh -q -T -x -n"

      let dist_sh com =
        let com = sprintf "%s %s '%s%s'" ssh C.addr path com in
        MySys.exec_stdout com

      let dist_sh_save com name =
        let com = sprintf "%s %s '%s%s' > %s" ssh C.addr path com name in
        MySys.exec_stdout com


      let dist_upload f t =
        let com = sprintf "scp -q %s %s:%s > /dev/null" f C.addr t in
        MySys.exec_stdout com

      let dist_download f t =
        let com = sprintf "scp -q %s:%s %s > /dev/null" C.addr f t in
        MySys.exec_stdout com
    end


  let upload,download,sh,sh_save =
    match C.target with
    | Local ->
        let open LocalCom in
        dist_upload,dist_download,dist_sh,dist_sh_save
    | Distant addr ->
        let module M = DistCom(struct let addr = addr end) in
        let open M in
        dist_upload,dist_download,dist_sh,dist_sh_save

  let dist_upload = upload
  let dist_download = download
  let dist_sh = sh
  let dist_sh_save = sh_save
end