1 use std
::convert
::From
;
6 use std
::borrow
::BorrowMut
;
7 use std
::sync
::{Arc
, Mutex
};
10 use num_traits
::cast
::AsPrimitive
;
11 use percent_encoding
::percent_decode_str
;
12 use actix_web
::{middleware
::Logger
, get
, web
, App
, HttpServer
, HttpRequest
, HttpResponse
, Responder
, ResponseError
, Result
};
13 use actix_web
::body
::BoxBody
;
14 use actix_web
::http
::StatusCode
;
15 use image
::{ImageBuffer
, ColorType
, Rgb
, RgbImage
, write_buffer_with_format
};
16 use image
::ImageOutputFormat
::Png
;
17 use image
::imageops
::{FilterType
, overlay
, resize
};
18 use image
::error
::{LimitError
, LimitErrorKind
};
19 use actix_web
::cookie
::time
::OffsetDateTime
;
22 const TIMEAVATAR_SIZE_U32
: u32 = 6;
23 const TIMEAVATAR_SIZE_I64
: i64 = TIMEAVATAR_SIZE_U32
as i64;
25 macro_rules
! response_image
{
27 HttpResponse
::build(StatusCode
::OK
)
28 .content_type("image/png")
29 .body($cursor
.into
_inner
())
34 struct ImageError(image
::ImageError
);
37 fn dimension_error() -> ImageError
{
38 ImageError(image
::ImageError
::Limits(LimitError
::from_kind(LimitErrorKind
::DimensionError
)))
42 impl Display
for ImageError
{
43 fn fmt(&self, f
: &mut fmt
::Formatter
<'_
>) -> fmt
::Result
{
44 write
!(f
, "{}", &self.0)
48 impl ResponseError
for ImageError
{
49 fn status_code(&self) -> StatusCode
{
50 StatusCode
::INTERNAL_SERVER_ERROR
53 fn error_response(&self) -> HttpResponse
<BoxBody
> {
54 HttpResponse
::InternalServerError().body(format
!("{}\n", &self))
58 impl From
<image
::ImageError
> for ImageError
{
59 fn from(e
: image
::ImageError
) -> Self {
67 DimensionMismatch((u32, u32)),
68 ImageError(ImageError
)
71 impl Display
for CanvasError
{
72 fn fmt(&self, f
: &mut fmt
::Formatter
<'_
>) -> fmt
::Result
{
73 write
!(f
, "Canvas Error: see error_response() for more details")
77 impl ResponseError
for CanvasError
{
78 fn status_code(&self) -> StatusCode
{
80 CanvasError
::NotExists
=> StatusCode
::NOT_FOUND
,
81 CanvasError
::DimensionMismatch(_
) => StatusCode
::BAD_REQUEST
,
82 CanvasError
::ImageError(e
) => e
.status_code()
86 fn error_response(&self) -> HttpResponse
<BoxBody
> {
88 CanvasError
::NotExists
=> HttpResponse
::NotFound().body("Canvas does not exist."),
89 CanvasError
::DimensionMismatch(dim
) =>
90 HttpResponse
::BadRequest().body(format
!("Supplied dimensions don't match the dimensions of existing canvas {:?}.", dim
)),
91 CanvasError
::ImageError(e
) => e
.error_response()
96 impl From
<ImageError
> for CanvasError
{
97 fn from(e
: ImageError
) -> CanvasError
{
98 CanvasError
::ImageError(e
)
102 struct ToRgbIter
<'a
, T
> where T
: AsPrimitive
<u8> {
103 iter
: &'a
mut dyn Iterator
<Item
= T
>
106 impl<'a
, T
: AsPrimitive
<u8>> ToRgbIter
<'a
, T
> {
107 fn new(iter
: &'a
mut (dyn Iterator
<Item
= T
>)) -> ToRgbIter
<T
> {
112 impl<T
: AsPrimitive
<u8> + Zero
> Iterator
for ToRgbIter
<'_
, T
> {
114 fn next(&mut self) -> Option
<Rgb
<u8>> {
115 if let Some(r
) = self.iter
.next() {
116 let g
= self.iter
.next().unwrap
_or
_else
(T
::zero
);
117 let b
= self.iter
.next().unwrap
_or
_else
(T
::zero
);
118 Some(Rgb([r
.as_(), g
.as_(), b
.as_()]))
133 fn new(width
: u32, height
: u32) -> Canvas
{
134 Canvas
{ pen
: 0, img
: ImageBuffer
::new(width
, height
) }
137 /*fn from_image(img: RgbImage) -> Canvas {
138 Canvas { pen: 0, img }
141 fn replace(&self, img: RgbImage, pen: usize) -> Canvas {
147 fn advance_pen(mut self, offset
: usize) -> Canvas
{
148 //Canvas { pen: self.pen + offset, img: self.img }
150 self.pen
%= self.img
.width() as usize * self.img
.height() as usize;
155 struct Canvas0(Option
<Canvas
>);
156 struct Canvas1(Option
<Canvas
>);
157 struct Canvas2(Option
<Canvas
>);
159 #[get("/hello/{name}")]
160 async
fn greet(name
: web
::Path
<String
>, req
: HttpRequest
) -> impl Responder
{
161 println
!("{:?}", req
);
162 format
!("Hello {name}!\n{:?}", req
.headers().get("user-agent"))
165 fn rgb_encode_with_pen(img
: &mut RgbImage
, pen
: usize, data
: &mut dyn Iterator
<Item
= u8>) -> Result
<usize, ImageError
> {
166 let mut pixels
= img
.pixels_mut().skip(pen
);
169 for sp
in ToRgbIter
::new(data
) {
170 let mut dp
= match pixels
.next() {
171 Some(pixel
) => pixel
,
173 pixels
= img
.pixels_mut().skip(0);
174 pixels
.next().ok_or_else(ImageError
::dimension_error
)?
184 fn rgb_encode_to_canvas(mut canvas
: Canvas
, data
: &mut dyn Iterator
<Item
= u8>) -> Result
<Canvas
, CanvasError
> {
185 let counter
= rgb_encode_with_pen(&mut canvas
.img
, canvas
.pen
, data
)?
;
186 Ok(canvas
.advance_pen(counter
))
189 fn rgb_encode(mut img
: RgbImage
, data
: &mut dyn Iterator
<Item
= u8>) -> Result
<RgbImage
, ImageError
> {
190 rgb_encode_with_pen(&mut img
, 0, data
)?
;
194 fn apply_to_canvas
<'a
, F
>(f
: F
, canvas_option
: &'a
mut Option
<Canvas
>, dimensions
: Option
<(u32, u32)>, data
: &str) -> Result
<&'a
mut Canvas
, CanvasError
>
195 where F
: Fn(Canvas
, &mut dyn Iterator
<Item
= u8>) -> Result
<Canvas
, CanvasError
>
197 let canvas
= match canvas_option
.take() {
198 Some(canvas
) => match dimensions
{
199 Some(dimensions
) => {
200 if dimensions
== canvas
.img
.dimensions() {
204 Err(CanvasError
::DimensionMismatch(canvas_option
.insert
(canvas
).img
.dimensions()))
210 None
=> match dimensions
{
211 Some((dim_x
, dim_y
)) =>
212 Ok(Canvas
::new(dim_x
, dim_y
)),
214 Err(CanvasError
::NotExists
)
218 Ok(canvas_option
.insert
(f(canvas
, percent_decode_str(data
).into
_iter
().borrow_mut())?
))
221 fn make_png(dim_x
: u32, dim_y
: u32, scale
: u32, data
: &mut dyn Iterator
<Item
= u8>) -> Result
<Cursor
<Vec
<u8>>, ImageError
> {
222 // Image must not be larger than 1 megapixel
223 if dim_x
* dim_y
* scale
> 1048576 {
224 return Err(ImageError
::dimension_error())
227 let img
: RgbImage
= rgb_encode(ImageBuffer
::new(dim_x
, dim_y
), data
)?
;
228 let cursor
= image_to_cursor(img
, scale
)?
;
232 fn image_to_cursor(img
: RgbImage
, scale
: u32) -> Result
<Cursor
<Vec
<u8>>, ImageError
> {
233 let (dim_x
, dim_y
) = img
.dimensions();
234 let (dim_x
, dim_y
) = (dim_x
* scale
, dim_y
* scale
);
236 // Image must not be larger than 1 megapixel
237 if dim_x
* dim_y
> 1048576 {
238 return Err(ImageError
::dimension_error())
241 let mut cursor
= Cursor
::new(Vec
::new());
242 let img
= resize(&img
, dim_x
, dim_y
, FilterType
::Nearest
);
243 write_buffer_with_format(&mut cursor
, &img
, dim_x
, dim_y
, ColorType
::Rgb8
, Png
)?
;
248 #[get("/gen/0/{data}")]
249 async
fn img_gen0(req
: HttpRequest
) -> Result
<impl Responder
> {
250 let data
= req
.ur
i().path().split('
/'
).nth(3).unwrap
();
251 let cursor
= make_png(32, 32, 16, percent_decode_str(data
).into
_iter
().borrow_mut())?
;
252 Ok(response_image
!(cursor
))
255 #[get("/gen/1/{dim_x}/{dim_y}/{scale}/{data}")]
256 async
fn img_gen1(req
: HttpRequest
, path
: web
::Path
<(u32, u32, u32)>) -> Result
<impl Responder
> {
257 let (dim_x
, dim_y
, scale
) = path
.into
_inner
();
258 let data
= req
.ur
i().path().split('
/'
).nth(6).unwrap
();
259 let cursor
= make_png(dim_x
, dim_y
, scale
, percent_decode_str(data
).into
_iter
().borrow_mut())?
;
260 Ok(response_image
!(cursor
))
263 #[get("/gen/2/{dim_x}/{dim_y}/{scale}")]
264 async
fn img_gen2(req
: HttpRequest
, path
: web
::Path
<(u32, u32, u32)>) -> Result
<impl Responder
> {
265 let (dim_x
, dim_y
, scale
) = path
.into
_inner
();
266 let data
= match req
.headers().get("user-agent") {
267 Some(header
) => header
.to_str().unwrap
_or
(""),
270 let cursor
= make_png(dim_x
, dim_y
, scale
, &mut data
.bytes())?
;
271 Ok(response_image
!(cursor
))
274 #[get("/gen/3/{scale}")]
275 async
fn img_gen3(req
: HttpRequest
, path
: web
::Path
<u32>) -> Result
<impl Responder
> {
276 let scale
= path
.into
_inner
();
277 let data
= match req
.headers().get("user-agent") {
278 Some(header
) => header
.to_str().unwrap
_or
(""),
281 let rgbimg
: RgbImage
= rgb_encode(ImageBuffer
::new(TIMEAVATAR_SIZE_U32
, TIMEAVATAR_SIZE_U32
), &mut data
.bytes())?
;
282 let mut resimg
: RgbImage
= ImageBuffer
::new(60 * TIMEAVATAR_SIZE_U32
, 24 * TIMEAVATAR_SIZE_U32
);
283 let time
= OffsetDateTime
::now_utc().time();
284 let (hour
, minute
): (i64, i64) = (time
.hour().into
(), time
.minute().into
());
285 overlay(&mut resimg
, &rgbimg
, minute
* TIMEAVATAR_SIZE_I64
, hour
* TIMEAVATAR_SIZE_I64
);
286 let cursor
= image_to_cursor(resimg
, scale
)?
;
287 Ok(response_image
!(cursor
))
290 #[get("/pgen/0/{data}")]
291 async
fn img_pgen0(req
: HttpRequest
, canvas0
: web
::Data
<Arc
<Mutex
<Canvas0
>>>) -> Result
<impl Responder
> {
292 let data
= req
.ur
i().path().split('
/'
).nth(3).unwrap
();
294 let img
: RgbImage
= ({
295 let canvas_option
= &mut *canvas0
.lock().unwrap
();
296 Ok(apply_to_canvas( |c
, d
| rgb_encode_to_canvas(c
, d
), &mut canvas_option
.0, None
, data
)?
.img
.clone())
297 } as Result
<RgbImage
, CanvasError
>)?
;
298 let cursor
= image_to_cursor(img
, scale
)?
;
299 Ok(response_image
!(cursor
))
302 #[get("/pgen/1/{dim_x}/{dim_y}/{scale}/{data}")]
303 async
fn img_pgen1(req
: HttpRequest
, path
: web
::Path
<(u32, u32, u32)>, canvas1
: web
::Data
<Arc
<Mutex
<Canvas1
>>>) -> Result
<impl Responder
> {
304 let (dim_x
, dim_y
, scale
) = path
.into
_inner
();
305 let data
= req
.ur
i().path().split('
/'
).nth(6).unwrap
();
306 let img
: RgbImage
= ({
307 let canvas_option
= &mut *canvas1
.lock().unwrap
();
308 Ok(apply_to_canvas( |c
, d
| rgb_encode_to_canvas(c
, d
), &mut canvas_option
.0, Some((dim_x
, dim_y
)), data
)?
.img
.clone())
309 } as Result
<RgbImage
, CanvasError
>)?
;
310 let cursor
= image_to_cursor(img
, scale
)?
;
311 Ok(response_image
!(cursor
))
314 #[get("/pgen/2/{dim_x}/{dim_y}/{scale}")]
315 async
fn img_pgen2(req
: HttpRequest
, path
: web
::Path
<(u32, u32, u32)>, canvas1
: web
::Data
<Arc
<Mutex
<Canvas1
>>>) -> Result
<impl Responder
> {
316 let (dim_x
, dim_y
, scale
) = path
.into
_inner
();
317 let data
= match req
.headers().get("user-agent") {
318 Some(header
) => header
.to_str().unwrap
_or
(""),
321 let img
: RgbImage
= ({
322 let canvas_option
= &mut *canvas1
.lock().unwrap
();
323 Ok(apply_to_canvas( |c
, d
| rgb_encode_to_canvas(c
, d
), &mut canvas_option
.0, Some((dim_x
, dim_y
)), data
)?
.img
.clone())
324 } as Result
<RgbImage
, CanvasError
>)?
;
325 let cursor
= image_to_cursor(img
, scale
)?
;
326 Ok(response_image
!(cursor
))
329 #[get("/pgen/3/{scale}")]
330 async
fn img_pgen3(req
: HttpRequest
, path
: web
::Path
<u32>, canvas2
: web
::Data
<Arc
<Mutex
<Canvas2
>>>) -> Result
<impl Responder
> {
331 let scale
= path
.into
_inner
();
332 let data
= match req
.headers().get("user-agent") {
333 Some(header
) => header
.to_str().unwrap
_or
(""),
336 let time
= OffsetDateTime
::now_utc().time();
337 let (hour
, minute
): (i64, i64) = (time
.hour().into
(), time
.minute().into
());
338 let rgbimg
: RgbImage
= rgb_encode(ImageBuffer
::new(TIMEAVATAR_SIZE_U32
, TIMEAVATAR_SIZE_U32
), &mut data
.bytes())?
;
339 let resimg
: RgbImage
= ({
340 let canvas_option
= &mut *canvas2
.lock().unwrap
();
341 let mut canvas
= match canvas_option
.0.take() {
342 Some(canvas
) => canvas
,
343 None
=> Canvas
::new(60 * TIMEAVATAR_SIZE_U32
, 24 * TIMEAVATAR_SIZE_U32
)
345 overlay(&mut canvas
.img
, &rgbimg
, minute
* TIMEAVATAR_SIZE_I64
, hour
* TIMEAVATAR_SIZE_I64
);
346 Ok(canvas_option
.0.insert
(canvas
).img
.clone())
347 } as Result
<RgbImage
, CanvasError
>)?
;
348 let cursor
= image_to_cursor(resimg
, scale
)?
;
349 Ok(response_image
!(cursor
))
352 #[get("/pdrop/{canvas_id}")]
353 async
fn pdrop(canvas_id
: web
::Path
<u8>,
354 canvas0
: web
::Data
<Arc
<Mutex
<Canvas0
>>>,
355 canvas1
: web
::Data
<Arc
<Mutex
<Canvas1
>>>,
356 canvas2
: web
::Data
<Arc
<Mutex
<Canvas2
>>>) -> Result
<impl Responder
> {
358 let canvas_id
= canvas_id
.into
_inner
();
360 0 => canvas0
.lock().unwrap
().0.take(),
361 1 => canvas1
.lock().unwrap
().0.take(),
362 2 => canvas2
.lock().unwrap
().0.take(),
364 }.ok_or(CanvasError
::NotExists
)?
;
365 Ok(HttpResponse
::build(StatusCode
::OK
)
366 .body(format
!("Canvas {} dropped.", canvas_id
)))
370 #[actix_web::main] // or #[tokio::main]
371 async
fn main() -> std
::io
::Result
<()> {
372 env_logger
::builder()
373 .filter
_mod
ule
("actix_web", LevelFilter
::Info
)
377 let listenaddress
= env
::args().nth(1).unwrap
_or
_else
(|| "127.0.0.1:8080".to_string());
378 let canvas0
= Arc
::new(Mutex
::new(Canvas0(Some(Canvas
::new(32, 32)))));
379 let canvas1
= Arc
::new(Mutex
::new(Canvas1(None
)));
380 let canvas2
= Arc
::new(Mutex
::new(Canvas2(None
)));
382 HttpServer
::new(move || {
384 .wrap(Logger
::default())
395 .app_data(web
::Data
::new(canvas0
.clone()))
396 .app_data(web
::Data
::new(canvas1
.clone()))
397 .app_data(web
::Data
::new(canvas2
.clone()))
399 .bind(listenaddress
)?