Vous ne pouvez pas sélectionner plus de 25 sujets Les noms de sujets doivent commencer par une lettre ou un nombre, peuvent contenir des tirets ('-') et peuvent comporter jusqu'à 35 caractères.

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275
  1. package main
  2. import "errors"
  3. import "log"
  4. import "os"
  5. import "fmt"
  6. import "path"
  7. import "net/http"
  8. import "io/ioutil"
  9. import "encoding/json"
  10. import "github.com/go-vgo/robotgo"
  11. import "github.com/BurntSushi/toml"
  12. import "coffee.mort.mediator/screencap"
  13. type Config struct {
  14. BasePath string `toml:"base_path"`
  15. ScrollStep int `toml:"scroll_step"`
  16. }
  17. type EmptyData struct {}
  18. type KeyboardTypeData struct {
  19. Text string `json:"text"`
  20. }
  21. type KeyboardKeyData struct {
  22. Key string `json:"key"`
  23. Modifiers []string `json:"modifiers"`
  24. }
  25. type ScrollData struct {
  26. X int `json:"x"`
  27. Y int `json:"y"`
  28. }
  29. type MouseClickData struct {
  30. Button string `json:"button"`
  31. DoubleClick bool `json:"doubleClick"`
  32. }
  33. type MousePosData struct {
  34. X int `json:"x"`
  35. Y int `json:"y"`
  36. }
  37. type ScreenSizeData struct {
  38. Width int `json:"width"`
  39. Height int `json:"height"`
  40. }
  41. type DirEntryData struct {
  42. Name string `json:"name"`
  43. Type string `json:"type"`
  44. }
  45. type ListDirData struct {
  46. Entries []DirEntryData `json:"entries"`
  47. }
  48. type Error struct {
  49. Error string `json:"error"`
  50. }
  51. func readConfig() (*Config, error) {
  52. confFile, err := os.Open("config.toml")
  53. if err != nil {
  54. return nil, err
  55. }
  56. defer confFile.Close()
  57. var conf Config
  58. _, err = toml.NewDecoder(confFile).Decode(&conf)
  59. if err != nil {
  60. return nil, err
  61. }
  62. return &conf, nil
  63. }
  64. type RW http.ResponseWriter
  65. type Req http.Request
  66. func handler(h func(w RW, req *Req) error) (
  67. func(w http.ResponseWriter, req *http.Request)) {
  68. return func(w http.ResponseWriter, req *http.Request) {
  69. err := h(RW(w), (*Req)(req))
  70. if err != nil {
  71. w.WriteHeader(400)
  72. err = json.NewEncoder(w).Encode(&Error{err.Error()})
  73. if err != nil {
  74. w.Write([]byte("Oh no, failed to encode error struct"))
  75. }
  76. }
  77. }
  78. }
  79. func main() {
  80. conf, err := readConfig()
  81. if err != nil {
  82. log.Fatal(err)
  83. }
  84. log.Printf("Config: %#v", conf)
  85. fs := http.FileServer(http.Dir("./web"))
  86. http.Handle("/", fs)
  87. http.HandleFunc("/api/remote/screen-size", handler(func(w RW, req *Req) error {
  88. if req.Method == "GET" {
  89. var size ScreenSizeData
  90. size.Width, size.Height = robotgo.GetScreenSize()
  91. return json.NewEncoder(w).Encode(&size)
  92. } else {
  93. return errors.New("Invalid method: " + req.Method)
  94. }
  95. }))
  96. http.HandleFunc("/api/remote/mouse-pos", handler(func(w RW, req *Req) error {
  97. if req.Method == "GET" {
  98. var pos MousePosData
  99. pos.X, pos.Y = robotgo.GetMousePos()
  100. return json.NewEncoder(w).Encode(&pos)
  101. } else if req.Method == "PUT" {
  102. var pos MousePosData
  103. err := json.NewDecoder(req.Body).Decode(&pos)
  104. if err != nil {
  105. return err
  106. }
  107. robotgo.MoveMouse(pos.X, pos.Y)
  108. return json.NewEncoder(w).Encode(&EmptyData{})
  109. } else {
  110. return errors.New("Invalid method: " + req.Method)
  111. }
  112. }))
  113. http.HandleFunc("/api/remote/mouse-click", handler(func(w RW, req *Req) error {
  114. if req.Method == "POST" {
  115. var click MouseClickData
  116. err := json.NewDecoder(req.Body).Decode(&click)
  117. if err != nil {
  118. return err
  119. }
  120. robotgo.MouseClick(click.Button, click.DoubleClick)
  121. return json.NewEncoder(w).Encode(&EmptyData{})
  122. } else {
  123. return errors.New("Invalid method: " + req.Method)
  124. }
  125. }))
  126. http.HandleFunc("/api/remote/scroll", handler(func(w RW, req *Req) error {
  127. if req.Method == "POST" {
  128. var scroll ScrollData
  129. err := json.NewDecoder(req.Body).Decode(&scroll)
  130. if err != nil {
  131. return err
  132. }
  133. robotgo.Scroll(scroll.X * conf.ScrollStep, scroll.Y * conf.ScrollStep)
  134. return json.NewEncoder(w).Encode(&EmptyData{})
  135. } else {
  136. return errors.New("Invalid method: "+ req.Method)
  137. }
  138. }))
  139. http.HandleFunc("/api/remote/keyboard-type", handler(func(w RW, req *Req) error {
  140. if req.Method == "POST" {
  141. var text KeyboardTypeData
  142. err := json.NewDecoder(req.Body).Decode(&text)
  143. if err != nil {
  144. return err
  145. }
  146. robotgo.TypeStr(text.Text)
  147. return json.NewEncoder(w).Encode(&EmptyData{})
  148. } else {
  149. return errors.New("Invalid method: " + req.Method)
  150. }
  151. }))
  152. http.HandleFunc("/api/remote/keyboard-key", handler(func(w RW, req *Req) error {
  153. if req.Method == "POST" {
  154. var key KeyboardKeyData
  155. err := json.NewDecoder(req.Body).Decode(&key)
  156. if err != nil {
  157. return err
  158. }
  159. var modifiers []interface{}
  160. for _, modifier := range key.Modifiers {
  161. modifiers = append(modifiers, modifier)
  162. }
  163. robotgo.KeyTap(key.Key, modifiers...)
  164. return json.NewEncoder(w).Encode(&EmptyData{})
  165. } else {
  166. return errors.New("Invalid method: " + req.Method)
  167. }
  168. }))
  169. http.HandleFunc("/api/remote/screencast", handler(func(w RW, req *Req) error {
  170. if req.Method == "GET" {
  171. w.Header().Add("Content-Type", "multipart/x-mixed-replace;boundary=MEDIATOR_FRAME_BOUNDARY")
  172. w.WriteHeader(200)
  173. for {
  174. img := <-screencap.Capture()
  175. var err error
  176. _, err = w.Write([]byte(fmt.Sprintf(
  177. "--MEDIATOR_FRAME_BOUNDARY\r\n" +
  178. "Content-Type: image/jpeg\r\n" +
  179. "Content-Length: %d\r\n" +
  180. "\r\n", img.Length)))
  181. if err != nil {
  182. log.Printf("Write error: %v", err)
  183. return nil
  184. }
  185. _, err = w.Write(img.Data[0:img.Length])
  186. if err != nil {
  187. log.Printf("Write error: %v", err)
  188. return nil
  189. }
  190. _, err = w.Write([]byte("\r\n"))
  191. if err != nil {
  192. log.Printf("Write error: %v", err)
  193. return nil
  194. }
  195. }
  196. } else {
  197. return errors.New("Invalid method: " + req.Method)
  198. }
  199. }));
  200. http.HandleFunc("/api/dir/", handler(func(w RW, req *Req) error {
  201. if req.Method == "GET" {
  202. subPath := req.URL.Path[len("/api/dir/"):]
  203. dirEnts, err := ioutil.ReadDir(path.Join(conf.BasePath, subPath))
  204. if err != nil {
  205. return err
  206. }
  207. list := ListDirData{
  208. Entries: make([]DirEntryData, 0, len(dirEnts)),
  209. }
  210. for _, ent := range dirEnts {
  211. entType := "f"
  212. if ent.IsDir() {
  213. entType = "d"
  214. }
  215. list.Entries = append(list.Entries, DirEntryData{
  216. Name: ent.Name(),
  217. Type: entType,
  218. })
  219. }
  220. return json.NewEncoder(w).Encode(&list)
  221. } else {
  222. return errors.New("Invalid method: " + req.Method)
  223. }
  224. }))
  225. go screencap.Run()
  226. log.Println("Listening on :3000...")
  227. err = http.ListenAndServe(":3000", nil)
  228. if err != nil {
  229. log.Fatal(err)
  230. }
  231. }