Momentum-Apps/minesweeper/minesweeper.c

#include <furi.h>
#include <furi_hal.h>
#include <gui/gui.h>
#include <input/input.h>
#include <stdlib.h>

#include <notification/notification_messages.h>
#include <dialogs/dialogs.h>
#include <m-string.h>

#include "assets.h"

#define PLAYFIELD_WIDTH 16
#define PLAYFIELD_HEIGHT 7
#define TILE_WIDTH 8
#define TILE_HEIGHT 8

#define MINECOUNT 20

typedef enum {
    EventTypeTick,
    EventTypeKey,
} EventType;

typedef struct {
    EventType type;
    InputEvent input;
} PluginEvent;

typedef enum {
  TileType0, // this HAS to be in order, for hint assignment to be ez pz
  TileType1,
  TileType2,
  TileType3,
  TileType4,
  TileType5,
  TileType6,
  TileType7,
  TileType8,
  TileTypeUncleared,
  TileTypeFlag,
  TileTypeMine
} TileType;

typedef enum {
    FieldEmpty,
    FieldMine
} Field;

typedef struct {
  FuriMutex* mutex;
  DialogsApp* dialogs;
  NotificationApp* notifications;
  Field minefield[PLAYFIELD_WIDTH][PLAYFIELD_HEIGHT];
  TileType playfield[PLAYFIELD_WIDTH][PLAYFIELD_HEIGHT];
  int cursor_x;
  int cursor_y;
  int mines_left;
  int fields_cleared;
  int flags_set;
  bool game_started;
  uint32_t game_started_tick;
} Minesweeper;

static void input_callback(InputEvent* input_event, FuriMessageQueue* event_queue) {
    furi_assert(event_queue);

    PluginEvent event = {.type = EventTypeKey, .input = *input_event};
    furi_message_queue_put(event_queue, &event, FuriWaitForever);
}

static void render_callback(Canvas* const canvas, void* ctx) {
    furi_assert(ctx);
    const Minesweeper* minesweeper_state = ctx;
    furi_mutex_acquire(minesweeper_state->mutex, FuriWaitForever);
    FuriString* mineStr;
    FuriString* timeStr;
    mineStr = furi_string_alloc();
    timeStr = furi_string_alloc();

    furi_string_printf(mineStr, "Mines: %d", MINECOUNT - minesweeper_state->flags_set);
    canvas_set_font(canvas, FontSecondary);
    canvas_draw_str_aligned(canvas, 0, 0, AlignLeft, AlignTop, furi_string_get_cstr(mineStr));

    int seconds = 0;
    int minutes = 0;
    if (minesweeper_state->game_started) {
      uint32_t ticks_elapsed = furi_get_tick() - minesweeper_state->game_started_tick;
      seconds = (int) ticks_elapsed / furi_kernel_get_tick_frequency();
      minutes = (int) seconds / 60;
      seconds = seconds % 60;
    }
    furi_string_printf(timeStr, "%01d:%02d", minutes, seconds);
    canvas_draw_str_aligned(canvas, 128, 0, AlignRight, AlignTop, furi_string_get_cstr(timeStr));

    uint8_t* tile_to_draw;

    for (int y = 0; y < PLAYFIELD_HEIGHT; y++) {
      for (int x = 0; x < PLAYFIELD_WIDTH; x++) {
        if ( x == minesweeper_state->cursor_x && y == minesweeper_state->cursor_y) {
          canvas_invert_color(canvas);
        }
        switch (minesweeper_state->playfield[x][y]) {
          case TileType0:
            tile_to_draw = tile_0_bits;
            break;
          case TileType1:
            tile_to_draw = tile_1_bits;
            break;
          case TileType2:
            tile_to_draw = tile_2_bits;
            break;
          case TileType3:
            tile_to_draw = tile_3_bits;
            break;
          case TileType4:
            tile_to_draw = tile_4_bits;
            break;
          case TileType5:
            tile_to_draw = tile_5_bits;
            break;
          case TileType6:
            tile_to_draw = tile_6_bits;
            break;
          case TileType7:
            tile_to_draw = tile_7_bits;
            break;
          case TileType8:
            tile_to_draw = tile_8_bits;
            break;
          case TileTypeFlag:
            tile_to_draw = tile_flag_bits;
            break;
          case TileTypeUncleared:
            tile_to_draw = tile_uncleared_bits;
            break;
          case TileTypeMine:
            tile_to_draw = tile_mine_bits;
            break;
          default:
            // this should never happen
            tile_to_draw = tile_mine_bits;
            break;
        }
        canvas_draw_xbm(
            canvas,
            x*TILE_HEIGHT, // x
            8 + (y * TILE_WIDTH), // y
            TILE_WIDTH,
            TILE_HEIGHT,
            tile_to_draw);
        if ( x == minesweeper_state->cursor_x && y == minesweeper_state->cursor_y) {
          canvas_invert_color(canvas);
        }
      }
    }

    furi_string_free(mineStr);
    furi_string_free(timeStr);
    furi_mutex_release(minesweeper_state->mutex);
}

static void setup_playfield(Minesweeper* minesweeper_state) {
  int mines_left = MINECOUNT;
  for (int y = 0; y < PLAYFIELD_HEIGHT; y++) {
    for (int x = 0; x < PLAYFIELD_WIDTH; x++){
        minesweeper_state->minefield[x][y] = FieldEmpty;
        minesweeper_state->playfield[x][y] = TileTypeUncleared;
    }
  }
  while(mines_left > 0) {
    int rand_x = rand() % PLAYFIELD_WIDTH;
    int rand_y = rand() % PLAYFIELD_HEIGHT;
    // make sure first guess isn't a mine
    if (minesweeper_state->minefield[rand_x][rand_y] == FieldEmpty &&
      (minesweeper_state->cursor_x != rand_x || minesweeper_state->cursor_y != rand_y)) {
       minesweeper_state->minefield[rand_x][rand_y] = FieldMine;
       mines_left--;
    }
  }
  minesweeper_state->mines_left = MINECOUNT;
  minesweeper_state->fields_cleared = 0;
  minesweeper_state->flags_set = 0;
  minesweeper_state->game_started_tick = furi_get_tick();
  minesweeper_state->game_started = false;
}

static void place_flag(Minesweeper* minesweeper_state) {
  if (minesweeper_state->playfield[minesweeper_state->cursor_x][minesweeper_state->cursor_y] == TileTypeUncleared) {
    minesweeper_state->playfield[minesweeper_state->cursor_x][minesweeper_state->cursor_y] = TileTypeFlag;
    minesweeper_state->flags_set++;
  } else if (minesweeper_state->playfield[minesweeper_state->cursor_x][minesweeper_state->cursor_y] == TileTypeFlag) {
    minesweeper_state->playfield[minesweeper_state->cursor_x][minesweeper_state->cursor_y] = TileTypeUncleared;
    minesweeper_state->flags_set--;
  }
}

static bool game_lost(Minesweeper* minesweeper_state) {
  // returns true if the player wants to restart, otherwise false

  DialogMessage* message = dialog_message_alloc();

  dialog_message_set_header(message, "Game Over", 64, 3, AlignCenter, AlignTop);
  dialog_message_set_text(message, "You hit a mine!", 64, 32, AlignCenter, AlignCenter);
  dialog_message_set_buttons(message, NULL, "Play again", NULL);

  // Set cursor to initial position
  minesweeper_state->cursor_x = 0;
  minesweeper_state->cursor_y = 0;

  notification_message(minesweeper_state->notifications, &sequence_single_vibro);

  DialogMessageButton choice = dialog_message_show(minesweeper_state->dialogs, message);
  dialog_message_free(message);

  return choice == DialogMessageButtonCenter;
}

static bool game_won(Minesweeper* minesweeper_state) {

  FuriString* tempStr;
  tempStr = furi_string_alloc();

  int seconds = 0;
  int minutes = 0;
  uint32_t ticks_elapsed = furi_get_tick() - minesweeper_state->game_started_tick;
  seconds = (int) ticks_elapsed / furi_kernel_get_tick_frequency();
  minutes = (int) seconds / 60;
  seconds = seconds % 60;

  DialogMessage* message = dialog_message_alloc();
  const char* header_text = "Game won!";
  furi_string_cat_printf(tempStr, "Minefield cleared in %01d:%02d", minutes, seconds);
  dialog_message_set_header(message, header_text, 64, 3, AlignCenter, AlignTop);
  dialog_message_set_text(message, furi_string_get_cstr(tempStr), 64, 32, AlignCenter, AlignCenter);
  dialog_message_set_buttons(message, NULL, "Play again", NULL);

  DialogMessageButton choice = dialog_message_show(minesweeper_state->dialogs, message);
  dialog_message_free(message);
  furi_string_free(tempStr);
  return choice == DialogMessageButtonCenter;
}

// returns false if the move loses the game - otherwise true
static bool play_move(Minesweeper* minesweeper_state, int cursor_x, int cursor_y) {
  if (minesweeper_state->playfield[cursor_x][cursor_y] == TileTypeFlag) {
    // we're on a flagged field, do nothing
    return true;
  }
  if (minesweeper_state->minefield[cursor_x][cursor_y] == FieldMine) {
    // player loses - draw mine
    minesweeper_state->playfield[cursor_x][cursor_y] = TileTypeMine;
    return false;
  }

  if (minesweeper_state->playfield[cursor_x][cursor_y] >= TileType1 && minesweeper_state->playfield[cursor_x][cursor_y] <= TileType8) {
    // click on a cleared cell with a number
    // count the flags around
    int flags = 0;
    for (int y = cursor_y-1; y <= cursor_y+1; y++) {
      for (int x = cursor_x-1; x <= cursor_x+1; x++) {
        if ( x == cursor_x && y == cursor_y ) {
          // we're on the cell the user selected, so ignore.
          continue;
        }
        // make sure we don't go OOB
        if ( x >= 0 && x < PLAYFIELD_WIDTH && y >= 0 && y < PLAYFIELD_HEIGHT) {
          if (minesweeper_state->playfield[x][y] == TileTypeFlag) {
              flags ++;
          }
        }
      }
    }
    int mines = minesweeper_state->playfield[cursor_x][cursor_y];   // ¯\_(ツ)_/¯
    if (flags == mines) {
      // auto uncover all non-flags around (to win faster ;)
      for (int auto_y = cursor_y-1; auto_y <= cursor_y+1; auto_y++) {
        for (int auto_x = cursor_x-1; auto_x <= cursor_x+1; auto_x++) {
          if ( auto_x == cursor_x && auto_y == cursor_y ) {
            continue;
          }
          if ( auto_x >= 0 && auto_x < PLAYFIELD_WIDTH && auto_y >= 0 && auto_y < PLAYFIELD_HEIGHT) {
            if (minesweeper_state->playfield[auto_x][auto_y] == TileTypeUncleared) {
              if(!play_move(minesweeper_state, auto_x, auto_y)) {
                // flags were wrong, we got a mine!
                return false;
              }
            }
          }
        }
      }
      // we're done without hitting a mine - so return
      return true;
    }
  }

  // calculate number of surrounding mines.
  int hint = 0;
  for (int y = cursor_y-1; y <= cursor_y+1; y++) {
    for (int x = cursor_x-1; x <= cursor_x+1; x++) {
      if ( x == cursor_x && y == cursor_y ) {
        // we're on the cell the user selected, so ignore.
        continue;
      }
      // make sure we don't go OOB
      if ( x >= 0 && x < PLAYFIELD_WIDTH && y >= 0 && y < PLAYFIELD_HEIGHT) {
        if(minesweeper_state->minefield[x][y] == FieldMine) {
            hint ++;
        }
      }
    }
  }
  // 〜(￣▽￣〜) don't judge me (〜￣▽￣)〜
  minesweeper_state->playfield[cursor_x][cursor_y] = hint;
  minesweeper_state->fields_cleared++;
  FURI_LOG_D("Minesweeper", "Setting %d,%d to %d", cursor_x, cursor_y, hint);
  if (hint == 0) {
    // the field is "empty"
    // auto open surrounding fields.
    for (int auto_y = cursor_y-1; auto_y <= cursor_y+1; auto_y++) {
      for (int auto_x = cursor_x-1; auto_x <= cursor_x+1; auto_x++) {
        if ( auto_x == cursor_x && auto_y == cursor_y ) {
          continue;
        }
        if ( auto_x >= 0 && auto_x < PLAYFIELD_WIDTH && auto_y >= 0 && auto_y < PLAYFIELD_HEIGHT) {
          if (minesweeper_state->playfield[auto_x][auto_y] == TileTypeUncleared) {
            play_move(minesweeper_state, auto_x, auto_y);
          }
        }
      }
    }
  }
  return true;
}

static void minesweeper_state_init(Minesweeper* const minesweeper_state) {
    minesweeper_state->cursor_x = minesweeper_state->cursor_y = 0;
    minesweeper_state->game_started = false;
    for (int y = 0; y < PLAYFIELD_HEIGHT; y++) {
      for (int x = 0; x < PLAYFIELD_WIDTH; x++){
          minesweeper_state->playfield[x][y] = TileTypeUncleared;
      }
    }
}

int32_t minesweeper_app(void* p) {
  UNUSED(p);

  FuriMessageQueue* event_queue = furi_message_queue_alloc(8, sizeof(PluginEvent));

  Minesweeper* minesweeper_state = malloc(sizeof(Minesweeper));
  // setup
  minesweeper_state_init(minesweeper_state);

  minesweeper_state->mutex = furi_mutex_alloc(FuriMutexTypeNormal);
  if(!minesweeper_state->mutex) {
      FURI_LOG_E("Minesweeper", "cannot create mutex\r\n");
      free(minesweeper_state);
      return 255;
  }
  // BEGIN IMPLEMENTATION

  minesweeper_state->dialogs = furi_record_open(RECORD_DIALOGS);
  minesweeper_state->notifications = furi_record_open(RECORD_NOTIFICATION);

  DialogMessage* message = dialog_message_alloc();

  dialog_message_set_header(message, "Minesweeper", 64, 3, AlignCenter, AlignTop);
  dialog_message_set_text(
      message,
      "Hold OK pressed to toggle flags.\ngithub.com/panki27",
      64,
      32,
      AlignCenter,
      AlignCenter);
  dialog_message_set_buttons(message, NULL, "Play", NULL);

  dialog_message_show(minesweeper_state->dialogs, message);
  dialog_message_free(message);

  // Set system callbacks
  ViewPort* view_port = view_port_alloc();
  view_port_draw_callback_set(view_port, render_callback, minesweeper_state);
  view_port_input_callback_set(view_port, input_callback, event_queue);

  // Open GUI and register view_port
  Gui* gui = furi_record_open(RECORD_GUI);
  gui_add_view_port(gui, view_port, GuiLayerFullscreen);

  PluginEvent event;
  for (bool processing = true; processing;) {
    FuriStatus event_status = furi_message_queue_get(event_queue, &event, 100);
    if(event_status == FuriStatusOk) {
      // press events
      if(event.type == EventTypeKey) {
        if(event.input.type == InputTypeShort) {
          switch(event.input.key) {
            case InputKeyUp:
              furi_mutex_acquire(minesweeper_state->mutex, FuriWaitForever);
              minesweeper_state->cursor_y--;
              if(minesweeper_state->cursor_y < 0) {
                 minesweeper_state->cursor_y = PLAYFIELD_HEIGHT - 1;
              }
              furi_mutex_release(minesweeper_state->mutex);
              break;
            case InputKeyDown:
              furi_mutex_acquire(minesweeper_state->mutex, FuriWaitForever);
              minesweeper_state->cursor_y++;
              if(minesweeper_state->cursor_y >= PLAYFIELD_HEIGHT) {
                 minesweeper_state->cursor_y = 0;
              }
              furi_mutex_release(minesweeper_state->mutex);
              break;
            case InputKeyRight:
              furi_mutex_acquire(minesweeper_state->mutex, FuriWaitForever);
              minesweeper_state->cursor_x++;
              if(minesweeper_state->cursor_x >= PLAYFIELD_WIDTH) {
                 minesweeper_state->cursor_x = 0;
              }
              furi_mutex_release(minesweeper_state->mutex);
              break;
            case InputKeyLeft:
              furi_mutex_acquire(minesweeper_state->mutex, FuriWaitForever);
              minesweeper_state->cursor_x--;
              if(minesweeper_state->cursor_x < 0) {
                 minesweeper_state->cursor_x = PLAYFIELD_WIDTH-1;
              }
              furi_mutex_release(minesweeper_state->mutex);
              break;
            case InputKeyOk:
              if (!minesweeper_state->game_started) {
                setup_playfield(minesweeper_state);
                minesweeper_state->game_started = true;
              }
              if (!play_move(minesweeper_state, minesweeper_state->cursor_x, minesweeper_state->cursor_y)) {
                // ooops. looks like we hit a mine!
                if (game_lost(minesweeper_state)) {
                  // player wants to restart.
                  setup_playfield(minesweeper_state);
                } else {
                  // player wants to exit :(
                  processing = false;
                }
              } else {
                // check win condition.
                if (minesweeper_state->fields_cleared == (PLAYFIELD_HEIGHT*PLAYFIELD_WIDTH) - MINECOUNT){
                  if (game_won(minesweeper_state)) {
                    //player wants to restart
                    setup_playfield(minesweeper_state);
                  } else {
                    processing = false;
                  }
                }
              }
              break;
            case InputKeyBack:
              // Exit the plugin
              processing = false;
              break;
            default:
              break;
          }
        } else if (event.input.type == InputTypeLong) {
          // hold events
          FURI_LOG_D("Minesweeper", "Got a long press!");
          switch(event.input.key) {
            case InputKeyUp:
            case InputKeyDown:
            case InputKeyRight:
            case InputKeyLeft:
              break;
            case InputKeyOk:
              FURI_LOG_D("Minesweeper", "Toggling flag");
              furi_mutex_acquire(minesweeper_state->mutex, FuriWaitForever);
              place_flag(minesweeper_state);
              furi_mutex_release(minesweeper_state->mutex);
              break;
            case InputKeyBack:
              processing = false;
              break;
            default:
              break;
          }
        }
      }
    }
    view_port_update(view_port);
  }
  view_port_enabled_set(view_port, false);
  gui_remove_view_port(gui, view_port);
  furi_record_close(RECORD_GUI);
  furi_record_close(RECORD_DIALOGS);
  furi_record_close(RECORD_NOTIFICATION);
  view_port_free(view_port);
  furi_message_queue_free(event_queue);
  furi_mutex_free(minesweeper_state->mutex);
  free(minesweeper_state);

  return 0;
}