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

#![no_std]
extern crate alloc;
use alloc::boxed::Box;
use core::future::Future;
use js_ffi::*;

pub type Handle = JSValue;

pub struct Timer {
    fn_set_timeout: JSInvoker,
    fn_set_interval: JSInvoker,
    fn_request_animation_frame: JSInvoker,
    fn_request_animation_loop: JSInvoker,
    fn_clear_timeout: JSInvoker,
    fn_clear_interval: JSInvoker,
}

impl Default for Timer {
    fn default() -> Self {
        Timer {
            fn_set_timeout: js!(window.setTimeout),
            fn_set_interval: js!(window.setInterval),
            fn_request_animation_frame: js!(window.requestAnimationFrame),
            fn_request_animation_loop: js!(
                (cb)=>{
                    let time = Date.now();
                    function run(){
                        let new_time = Date.now();
                        let delta = new_time-time;
                        time = new_time;
                        window.requestAnimationFrame(run);
                        cb(delta);
                    }
                    window.requestAnimationFrame(run);
                }
            ),
            fn_clear_timeout: js!(window.clearTimeout),
            fn_clear_interval: js!(window.clearInterval),
        }
    }
}

impl Timer {
    pub fn set_timeout(
        &self,
        callback: Box<dyn FnMut() -> () + Send>,
        milliseconds: u32,
    ) -> (Handle, JSFunction) {
        let cb = create_callback_0(callback);
        let handle = self.fn_set_timeout.invoke_2(cb, milliseconds);
        (handle, cb)
    }

    pub fn sleep(&self, milliseconds: u32) -> impl Future {
        let (future, cb) = create_callback_future_0();
        self.fn_set_timeout.invoke_2(cb, milliseconds);
        future
    }

    pub fn set_interval(
        &self,
        callback: Box<dyn FnMut() -> () + Send>,
        milliseconds: u32,
    ) -> (Handle, JSFunction) {
        let cb = create_callback_0(callback);
        let handle = self.fn_set_interval.invoke_2(cb, milliseconds);
        (handle, cb)
    }

    pub fn request_animation_frame(&self, callback: Box<dyn FnMut() -> () + Send>) -> JSFunction {
        let cb = create_callback_0(callback);
        self.fn_request_animation_frame.invoke_1(cb);
        cb
    }

    pub fn request_animation_loop(
        &self,
        callback: Box<dyn FnMut(JSValue) -> () + Send>,
    ) -> JSFunction {
        let cb = create_callback_1(callback);
        self.fn_request_animation_loop.invoke_1(cb);
        cb
    }

    pub fn clear_timeout(&self, handle: Handle) {
        self.fn_clear_timeout.invoke_1(handle);
    }

    pub fn clear_interval(&self, handle: Handle) {
        self.fn_clear_interval.invoke_1(handle);
    }
}